fix: path error when saving checkpoint

hellomyheart · May 13, 2023 · e1834a9 · e1834a9
1 parent 362fd12
commit e1834a9
Showing 1 changed file with 2 additions and 1 deletion.
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):