Skip to content

Instantly share code, notes, and snippets.

@thierryherrmann

thierryherrmann/reload_continue_train.py

Created August 2, 2020 23:54
Show Gist options
  • Save thierryherrmann/9c2f376f4e9c1b20c9ebf786bb71f2cb to your computer and use it in GitHub Desktop.
Save thierryherrmann/9c2f376f4e9c1b20c9ebf786bb71f2cb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
reload_continue_train.py
del new_module
new_module_2 = tf.keras.models.load_model(model_dir)
loss_hist = train_module(new_module_2, train_dataset, valid_dataset)
plot_loss(loss_hist)
save_module(new_module_2, model_dir)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment