Pownie/backgammon
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects Releases Wiki Activity

works now

Browse Source
This commit is contained in:
Alexander Munch-Hansen 2018-03-06 11:53:42 +01:00
parent 09973b6cde
commit 5845edf084
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
network.py
View File

@ -93,8 +93,9 @@ class Network:
self.saver.save(self.session, self.checkpoint_path + 'model.ckpt') self.saver.save(self.session, self.checkpoint_path + 'model.ckpt')
def restore_model(self): def restore_model(self):
latest_checkpoint = tf.train.latest_checkpoint(self.checkpoint_path) if os.path.isfile(self.checkpoint_path):
self.saver.restore(self.session, latest_checkpoint) latest_checkpoint = tf.train.latest_checkpoint(self.checkpoint_path)
self.saver.restore(self.session, latest_checkpoint)
# Have a circular dependency, #fuck, need to rewrite something # Have a circular dependency, #fuck, need to rewrite something
def train(self, x, v_next): def train(self, x, v_next):