Skip to content

Commit

Permalink
Fix saved model name output
Browse files Browse the repository at this point in the history
Corrected the output showing the name of the model being saved
  • Loading branch information
joydeepml authored Jul 1, 2018
1 parent bece037 commit 2f91f74
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def train_and_eval(_base_model_file, _train_image_folder, _test_image_folder, _r

if not testing:
trained_model.save(_new_model_file)
print("Stored trained model at %s" % tl_model_file)
print("Stored trained model at %s" % _new_model_file)

# evaluate test images
with open(_results_file, 'w') as output_file:
Expand Down

0 comments on commit 2f91f74

Please sign in to comment.