Merge pull request #1227 from Brunozml:bug_pt_cfr_load

PiperOrigin-RevId: 633182333 Change-Id: I0908da40afba849d11b6fa10ff15ba0b2f2e2efd
google-deepmind · May 13, 2024 · 7d3a355 · 7d3a355
2 parents c8354d7 + 5336932
commit 7d3a355
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/open_spiel/python/pytorch/dqn.py b/open_spiel/python/pytorch/dqn.py
@@ -428,7 +428,7 @@ def load(self, data_path, optimizer_data_path=None):
         relative or absolute but the filename should be included. For example:
         optimizer.pt or /path/to/optimizer.pt
     """
-    torch.load(self._q_network, data_path)
-    torch.load(self._target_q_network, data_path)
+    self._q_network = torch.load(data_path)
+    self._target_q_network = torch.load(data_path)
     if optimizer_data_path is not None:
-      torch.load(self._optimizer, optimizer_data_path)
+      self._optimizer = torch.load(optimizer_data_path)