Fixed json file saving

RenieWell · Aug 29, 2016 · 7728ca1 · 7728ca1
1 parent c2f412b
commit 7728ca1
Showing 1 changed file with 1 addition and 12 deletions.
diff --git a/kitti_train.py b/kitti_train.py
@@ -76,16 +76,5 @@
 
 if save_model:
     json_string = model.to_json()
-    with open(f, "w") as f:
+    with open(json_file, "w") as f:
         f.write(json_string)
-
-
-# will remove this
-import matplotlib
-matplotlib.use('Agg')
-import matplotlib.pyplot as plt
-plt.plot(history.history['loss'])
-plt.plot(history.history['val_loss'])
-plt.xlabel('Epoch')
-plt.ylabel('Loss')
-plt.savefig('error_curve.png')