Merge pull request datamllab#284 from Clarivy/master

Fix RuntimeError when running example from docs\toy-examples.md
hellomyheart · May 15, 2023 · fe65713 · fe65713
2 parents ee60aff + e1834a9
commit fe65713
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/rlcard/agents/dqn_agent.py b/rlcard/agents/dqn_agent.py
@@ -25,6 +25,7 @@
 SOFTWARE.
 '''
 
+import os
 import random
 import numpy as np
 import torch
@@ -319,7 +320,7 @@ def save_checkpoint(self, path, filename='checkpoint_dqn.pt'):
         Args:
             path (str): the path to save the model
         '''
-        torch.save(self.checkpoint_attributes(), path + '/' + filename)
+        torch.save(self.checkpoint_attributes(), os.path.join(path, filename))
 
 class Estimator(object):
     '''

diff --git a/rlcard/agents/nfsp_agent.py b/rlcard/agents/nfsp_agent.py
@@ -19,6 +19,7 @@
 See the paper https://arxiv.org/abs/1603.01121 for more details.
 '''
 
+import os
 import random
 import collections
 import enum
@@ -370,7 +371,7 @@ def save_checkpoint(self, path, filename='checkpoint_nfsp.pt'):
         Args:
             path (str): the path to save the model
         '''
-        torch.save(self.checkpoint_attributes(), path + '/' + filename)
+        torch.save(self.checkpoint_attributes(), os.path.join(path, filename))
 
 
 class AveragePolicyNetwork(nn.Module):