Orca change default backend of PyTorch estimator to bigdl (#2966)

* change type * Change default backend of PyTorch estimator to bigdl * Update PythonInferenceModel.scala * fix * update docs
intel-analytics · Oct 16, 2020 · b1021f9 · b1021f9
1 parent a2fd232
commit b1021f9
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)