Skip to content

S
seminar-breakout
Commit c270a1ed authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix keras example

parent 6f3ed357
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/mnist-keras.py
View file @ c270a1ed
...@@ -101,7 +101,7 @@ def get_config(): ...@@ -101,7 +101,7 @@ def get_config():
dataset_train, dataset_test = get_data() dataset_train, dataset_test = get_data()
return TrainConfig( return TrainConfig(
model=KerasModel(get_keras_model()), model=Model(),
dataflow=dataset_train, dataflow=dataset_train,
callbacks=[ callbacks=[
KerasCallback(True), # for Keras training KerasCallback(True), # for Keras training
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment