Upgrade DDPG

This commit is contained in:
Shangtong Zhang
2018-05-17 17:17:12 -06:00
parent ab1067d8a8
commit 08038f9533
7 changed files with 109 additions and 40 deletions
-3
View File
@@ -120,6 +120,3 @@ class Batcher:
indices = np.arange(self.num_entries)
np.random.shuffle(indices)
self.data = [d[indices] for d in self.data]
# def torch_max(tensor, dim):
# return torch.max(tensor, dim=dim, keepdim=True)[0]