style

intel-analytics · Nov 2, 2020 · b527063 · b527063
1 parent a9618ea
commit b527063
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyzoo/zoo/orca/learn/pytorch/training_operator.py b/pyzoo/zoo/orca/learn/pytorch/training_operator.py
@@ -354,8 +354,9 @@ def validate_batch(self, batch, batch_info):
             # Can't directly call torch.squeeze() in case batch size is 1.
             for i in reversed(range(1, len(_target.size()))):
                 _target = torch.squeeze(_target, i)
-        error_msg = "Currently in validate, only accuracy for classification with zero-based label is supported " \
-                    "by default. You can override validate_batch in TrainingOperator for other validation metrics"
+        error_msg = "Currently in validate, only accuracy for classification with zero-based " \
+                    "label is supported by default. You can override validate_batch in " \
+                    "TrainingOperator for other validation metrics"
         assert len(_target.size()) == 1, error_msg
 
         if self.use_gpu: