fix typo

hepesu · Jan 3, 2018 · 184b2c9 · 184b2c9
1 parent bb69a6e
commit 184b2c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -221,4 +221,4 @@ def residual_block(input_tensor, filters, shortcut_conv=False):
     callbacks=[checkpointer]
 )
 
-model.save_weights('model_final.h5')
+model.save('model_final.h5')