add script to save models as tar archives
This commit is contained in:
parent
b07f075627
commit
49966cd207
16
bin/save_model
Executable file
16
bin/save_model
Executable file
|
@ -0,0 +1,16 @@
|
||||||
|
#!/usr/bin/env ruby
|
||||||
|
require 'date'
|
||||||
|
|
||||||
|
models_dir = 'models'
|
||||||
|
model_name = ARGV[0]
|
||||||
|
model_path = File.join(models_dir, model_name)
|
||||||
|
|
||||||
|
episode_count = (File.read File.join(model_path, 'model.episodes')).to_i
|
||||||
|
|
||||||
|
puts "Found model #{model_name} with episodes #{episode_count} trained!"
|
||||||
|
|
||||||
|
file_name = "model-#{model_name}-#{episode_count}-#{Time.now.strftime('%Y%m%d-%H%M%S')}.tar.gz"
|
||||||
|
save_path = File.join(models_dir, 'saves', file_name)
|
||||||
|
puts "Saving to #{save_path}"
|
||||||
|
|
||||||
|
`tar -cvzf #{save_path} -C #{models_dir} #{model_name}`
|
Loading…
Reference in New Issue
Block a user