[STN] use queues for mnist-addition

7d3a24eb · Yuxin Wu · 8c9969f8 · 7d3a24eb
Commit 7d3a24eb authored Nov 18, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/SpatialTransformer/mnist-addition.py examples/SpatialTransformer/mnist-addition.py +1 -1

No files found.
--- a/examples/SpatialTransformer/mnist-addition.py
+++ b/examples/SpatialTransformer/mnist-addition.py
@@ -157,7 +157,7 @@ def get_config():

    return TrainConfig(
        model=Model(),
-        dataflow=dataset_train,
+        data=QueueInput(dataset_train),
        callbacks=[
            ModelSaver(),
            InferenceRunner(dataset_test,