fix keras example

c270a1ed · Yuxin Wu · 6f3ed357 · c270a1ed
Commit c270a1ed authored Oct 11, 2017 by Yuxin Wu
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/mnist-keras.py examples/mnist-keras.py +1 -1

No files found.
--- a/examples/mnist-keras.py
+++ b/examples/mnist-keras.py
@@ -101,7 +101,7 @@ def get_config():
    dataset_train, dataset_test = get_data()

    return TrainConfig(
-        model=KerasModel(get_keras_model()),
+        model=Model(),
        dataflow=dataset_train,
        callbacks=[
            KerasCallback(True),   # for Keras training