Merge pull request #188 from ranamihir/patch-1

Fix data type for torch v0.4.1
ikostrikov · Apr 16, 2019 · b4133ec · b4133ec
2 parents 4f04391 + 6a89459
commit b4133ec
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/a2c_ppo_acktr/algo/gail.py b/a2c_ppo_acktr/algo/gail.py
@@ -119,9 +119,11 @@ def __init__(self, file_name, num_trajectories=4, subsample_frequency=20):
  idx = perm[:num_trajectories]
 
  self.trajectories = {}
-
+
+ # See https://github.com/pytorch/pytorch/issues/14886
+ # .long() for fixing bug in torch v0.4.1
  start_idx = torch.randint(
- 0, subsample_frequency, size=(num_trajectories, ))
+ 0, subsample_frequency, size=(num_trajectories, )).long()
 
  for k, v in all_trajectories.items():
  data = v[idx]
@@ -162,4 +164,4 @@ def __getitem__(self, i):
  traj_idx, i = self.get_idx[i]
 
  return self.trajectories['states'][traj_idx][i], self.trajectories[
- 'actions'][traj_idx][i]
+ 'actions'][traj_idx][i]