Reinforcement Learning Pytorch Tutorial Example