#22 minor changes

suragnair · Jan 19, 2018 · 610fa37 · 610fa37
1 parent 668a0e5
commit 610fa37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/othello/pytorch/NNet.py b/othello/pytorch/NNet.py
@@ -147,6 +147,6 @@ def load_checkpoint(self, folder='checkpoint', filename='checkpoint.pth.tar'):
         # https://github.com/pytorch/examples/blob/master/imagenet/main.py#L98
         filepath = os.path.join(folder, filename)
         if not os.path.exists(filepath):
-            raise("No model in path {}".format(checkpoint))
+            raise("No model in path {}".format(filepath))
         checkpoint = torch.load(filepath)
         self.nnet.load_state_dict(checkpoint['state_dict'])
diff --git a/pit.py b/pit.py
@@ -21,7 +21,7 @@
 
 # nnet players
 n1 = NNet(g)
-n1.load_checkpoint('./pretrained_models/','6x100x25_best.pth.tar')
+n1.load_checkpoint('./pretrained_models/othello/pytorch/','6x100x25_best.pth.tar')
 args1 = dotdict({'numMCTSSims': 50, 'cpuct':1.0})
 mcts1 = MCTS(g, n1, args1)
 n1p = lambda x: np.argmax(mcts1.getActionProb(x, temp=0))