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