Example code does not run anymore

[pwuethri]

I tried to run my distillation code but ran into this problem:

`2019-03-12 12:57:14.387380 JST | Optimizing... Traceback (most recent call last): File "run_ppo.py", line 160, in optim_pol=optim_pol, optim_vf=optim_vf, epoch=args.epoch_per_iter, batch_size=args.batch_size, max_grad_norm=args.max_grad_norm) File "/home/pierre/anaconda3/lib/python3.7/site-packages/machina-0.2.0-py3.7.egg/machina/algos/ppo_clip.py", line 126, in train for batch in iterator: File "/home/pierre/anaconda3/lib/python3.7/site-packages/machina-0.2.0-py3.7.egg/machina/traj/traj.py", line 182, in iterate indices = self._get_indices(indices, shuffle) File "/home/pierre/anaconda3/lib/python3.7/site-packages/machina-0.2.0-py3.7.egg/machina/traj/traj.py", line 128, in _get_indices self.num_step, device=get_device(), dtype=torch.long) TypeError: arange() received an invalid combination of arguments - got (numpy.int64, dtype=torch.dtype, device=torch.device), but expected one of:

• (Number end, Tensor out, torch.dtype dtype, torch.layout layout, torch.device device, bool requires_grad)
• (Number start, Number end, Number step, Tensor out, torch.dtype dtype, torch.layout layout, torch.device device, bool `requires_grad)``

I also ran run_ppo.py and got the same issue. I think that something got changed in the trajectories file which affects all scripts

[rarilurelo]

Please update your pytorch pip install --upgrade torch