File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -164,7 +164,7 @@ def main_test_layers(model='relu'):
164164 # Add ops to save and restore all the variables, including variables for training.
165165 # ref: https://www.tensorflow.org/versions/r0.8/how_tos/variables/index.html
166166 saver = tf .train .Saver ()
167- save_path = saver .save (sess , "model.ckpt" )
167+ save_path = saver .save (sess , "./ model.ckpt" )
168168 print ("Model saved in file: %s" % save_path )
169169
170170
@@ -243,7 +243,7 @@ def main_test_denoise_AE(model='relu'):
243243 # ref: https://www.tensorflow.org/versions/r0.8/how_tos/variables/index.html
244244 saver = tf .train .Saver ()
245245 # you may want to save the model
246- save_path = saver .save (sess , "model.ckpt" )
246+ save_path = saver .save (sess , "./ model.ckpt" )
247247 print ("Model saved in file: %s" % save_path )
248248 sess .close ()
249249
@@ -405,7 +405,7 @@ def main_test_stacked_denoise_AE(model='relu'):
405405 # ref: https://www.tensorflow.org/versions/r0.8/how_tos/variables/index.html
406406 saver = tf .train .Saver ()
407407 # you may want to save the model
408- save_path = saver .save (sess , "model.ckpt" )
408+ save_path = saver .save (sess , "./ model.ckpt" )
409409 print ("Model saved in file: %s" % save_path )
410410 sess .close ()
411411
You can’t perform that action at this time.
0 commit comments