Orca PyTorch Estimator input torch loss instance (#3003)

* loss instance * update
intel-analytics · Nov 2, 2020 · b50afe5 · b50afe5
1 parent 44a55f7
commit b50afe5
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
@@ -80,7 +80,7 @@ def train_example(workers_per_node):
     estimator = Estimator.from_torch(
         model=model_creator,
         optimizer=optimizer_creator,
-        loss=nn.MSELoss,
+        loss=nn.MSELoss(),
         scheduler_creator=scheduler_creator,
         workers_per_node=workers_per_node,
         config={