Generate dataset for video prediction

2026-09-09 11:13:47 +08:00 · 2017-12-16 20:03:11 -07:00
parent 56304e63e6
commit 214f7693c2
8 changed files with 154 additions and 18 deletions
@@ -112,4 +112,4 @@ class DQNAgent:

    def save(self, file_name):
        with open(file_name, 'wb') as f:
-            pickle.dump(self.learning_network.state_dict(), f)
+            torch.save(self.learning_network.state_dict(), f)