Orca change default backend of PyTorch estimator to bigdl (intel-anal…

…ytics#2966) * change type * Change default backend of PyTorch estimator to bigdl * Update PythonInferenceModel.scala * fix * update docs
sgwhat · Oct 16, 2020 · 0bc5c9b · 0bc5c9b
1 parent f85b270
commit 0bc5c9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/orca/example/learn/horovod/pytorch_estimator.py b/python/orca/example/learn/horovod/pytorch_estimator.py
@@ -87,7 +87,7 @@ def train_example(workers_per_node):
             "lr": 1e-2,  # used in optimizer_creator
             "hidden_size": 1,  # used in model_creator
             "batch_size": 4,  # used in data_creator
-        })
+        }, backend="horovod")
 
     # train 5 epochs
     stats = estimator.fit(train_data_creator, epochs=5)