Save architecture and weights separately

Author: Colin Grambow

reacdiff/train/train.py

@@ -34,15 +34,20 @@ def train(crnn, data, save_dir,
         crnn.rnn.summary()
         crnn.model.summary()
 
-    model_name = 'model.{epoch:03d}.h5'
+    model_name = 'model.json'
+    weights_name = 'weights.h5'
     model_path = os.path.join(save_dir, model_name)
+    weights_path = os.path.join(save_dir, weights_name)
+    with open(model_path, 'w') as f:
+        f.write(crnn.model.to_json())
 
     def lr_schedule(epoch, _):
         return lr_start * ((lr_end / lr_start) ** (epoch / epochs))
 
-    checkpoint = keras.callbacks.ModelCheckpoint(filepath=model_path,
+    checkpoint = keras.callbacks.ModelCheckpoint(filepath=weights_path,
                                                  verbose=1,
-                                                 save_best_only=True)
+                                                 save_best_only=True,
+                                                 save_weights_only=True)
     early_stopping = keras.callbacks.EarlyStopping(patience=patience,
                                                    verbose=1,
                                                    restore_best_weights=True)