534f6b7975dcd334f6d68eef14598d75c79e7921,tests/test_model_saving.py,,test_sequential_model_saving_2,#,51

Before Change

    // test that new updates are the same with both models
    x = np.random.random((1, 3))
    y = np.random.random((1, 3, 3))
    model.train_on_batch(x, y)
    new_model.train_on_batch(x, y)
    out = model.predict(x)
    out2 = new_model.predict(x)

After Change

    fname = "tmp_" + str(np.random.randint(10000)) + ".h5"
    save_model(model, fname)

    model = load_model(fname,
                       custom_objects={"custom_opt": custom_opt,
                                       "custom_loss": custom_loss})
    os.remove(fname)

    out2 = model.predict(x)
    assert_allclose(out, out2, atol=1e-05)

Instances