fix bin/save_model
This commit is contained in:
parent
7b0e60f069
commit
153776d83e
|
@ -2,10 +2,10 @@
|
|||
require 'date'
|
||||
|
||||
models_dir = 'models'
|
||||
model_name = ARGV[0]
|
||||
model_name = (ARGV[0] or "default")
|
||||
model_path = File.join(models_dir, model_name)
|
||||
|
||||
episode_count = (File.read File.join(model_path, 'model.episodes')).to_i
|
||||
episode_count = (File.read File.join(model_path, 'episodes_trained')).to_i
|
||||
|
||||
puts "Found model #{model_name} with episodes #{episode_count} trained!"
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user