Skip to content

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

[STN] use queues for mnist-addition

parent 8c9969f8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/SpatialTransformer/mnist-addition.py
View file @ 7d3a24eb
...@@ -157,7 +157,7 @@ def get_config(): ...@@ -157,7 +157,7 @@ def get_config():
return TrainConfig( return TrainConfig(
model=Model(), model=Model(),
dataflow=dataset_train, data=QueueInput(dataset_train),
callbacks=[ callbacks=[
ModelSaver(), ModelSaver(),
InferenceRunner(dataset_test, InferenceRunner(dataset_test,
......
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