From 153776d83e5989dad7ead7ff010b114302a08482 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christoffer=20M=C3=BCller=20Madsen?=
 <christoffer@guava.space>
Date: Wed, 14 Mar 2018 11:06:55 +0100
Subject: [PATCH] fix bin/save_model

---
 bin/save_model | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bin/save_model b/bin/save_model
index 37362cc..10c6b81 100755
--- a/bin/save_model
+++ b/bin/save_model
@@ -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!"