model.save("migration-smoke.keras")
loaded = keras.models.load_model("migration-smoke.keras")