Orca PyTorchEstimatorLeNet MNIST Example (intel-analytics#2766)

* orca pytorch mnist lenet example * add backend * lenet5 * add readme * add evaluate * fix readme * fix readme * fix evaluate
yangw1234 · Sep 23, 2021 · fa3056a · fa3056a
1 parent 150a25d
commit fa3056a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/orca/src/bigdl/orca/learn/pytorch/estimator.py b/python/orca/src/bigdl/orca/learn/pytorch/estimator.py
@@ -229,7 +229,7 @@ def evaluate(self, data, validation_methods=None, batch_size=32):
             val_feature_set = FeatureSet.sample_rdd(data.rdd.flatMap(to_sample))
             return self.estimator.evaluate(val_feature_set, validation_methods, batch_size)
         elif isinstance(data, DataLoader) or callable(data):
-            val_feature_set = FeatureSet.pytorch_dataloader(data, "", "")
+            val_feature_set = FeatureSet.pytorch_dataloader(data)
             return self.estimator.evaluate_minibatch(val_feature_set, validation_methods)
         else:
             raise ValueError("Data should be a SparkXShards, a DataLoader or a callable "