def test_loading_without_autocompile(session_tf, filename, model): gp.Saver().save(filename, model) with session_context() as session: context = gp.SaverContext(autocompile=False) loaded = gp.Saver().load(filename, context=context) assert loaded.is_built(session_tf.graph) == gp.Build.NO assert loaded.is_built(session.graph) == gp.Build.NO assert not any(loaded.trainable_tensors)
def test_loading_into_specific_session(session_tf, filename, model): x_new = Data.x_new() predict_origin = model.predict_f(x_new) gp.Saver().save(filename, model) with session_context() as session: context = gp.SaverContext(session=session) loaded = gp.Saver().load(filename, context=context) predict_loaded = loaded.predict_f(x_new, session=session) assert_allclose(predict_origin, predict_loaded)
def encode_decode(value): ctx = gp.SaverContext() e = gp.saver.CoderDispatcher(ctx).encode(value) return gp.saver.CoderDispatcher(ctx).decode(e)