works now

2018-03-06 11:53:42 +01:00 · 2018-03-06 11:53:42 +01:00 · 5845edf084
commit 5845edf084
parent 09973b6cde
1 changed files with 3 additions and 2 deletions
--- a/network.py
+++ b/network.py
@ -93,6 +93,7 @@ class Network:
        self.saver.save(self.session, self.checkpoint_path + 'model.ckpt')
    
    def restore_model(self):
+        if os.path.isfile(self.checkpoint_path):
            latest_checkpoint = tf.train.latest_checkpoint(self.checkpoint_path)
            self.saver.restore(self.session, latest_checkpoint)