Example Pytorch Agenet