+folder for saved weights

rmeertens · Jul 4, 2017 · e63a99e · e63a99e
1 parent a56069b
commit e63a99e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/BrainDQN_Nature.py b/BrainDQN_Nature.py
@@ -43,7 +43,7 @@ def __init__(self,actions):
 		self.saver = tf.train.Saver()
 		self.session = tf.InteractiveSession()
 		self.session.run(tf.global_variables_initializer())
-		checkpoint = tf.train.get_checkpoint_state(".")
+		checkpoint = tf.train.get_checkpoint_state("./savedweights")
 		if checkpoint and checkpoint.model_checkpoint_path:
 				self.saver.restore(self.session, checkpoint.model_checkpoint_path)
 				print "Successfully loaded:", checkpoint.model_checkpoint_path
@@ -128,7 +128,7 @@ def trainQNetwork(self):
 
 		# save network every 100000 iteration
 		if self.timeStep % 10000 == 0:
-			self.saver.save(self.session, 'network' + '-dqn', global_step = self.timeStep)
+			self.saver.save(self.session, './savedweights/network' + '-dqn', global_step = self.timeStep)
 
 		if self.timeStep % UPDATE_TIME == 0:
 			self.copyTargetQNetwork()

diff --git a/savedweights/info.txt b/savedweights/info.txt
@@ -0,0 +1 @@
+In this folder all saved weights will appear