evaluate_forward_func as last argument

f-dangel · Sep 13, 2019 · ff6529d · ff6529d
1 parent 49bb694
commit ff6529d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/deepobs/pytorch/testproblems/testproblem.py b/deepobs/pytorch/testproblems/testproblem.py
@@ -95,11 +95,10 @@ def _get_next_batch(self):
         return next(self._iterator)
 
     def get_batch_loss_and_accuracy(self,
-                                    return_forward_func = False,
-                                    evaluate_forward_func = True,
-                                    reduction = 'mean',
-                                    add_regularization_if_available = True):
-
+                                    return_forward_func=False,
+                                    reduction='mean',
+                                    add_regularization_if_available=True,
+                                    evaluate_forward_func=True):
         """Gets a new batch and calculates the loss and accuracy (if available)
         on that batch. This is a default implementation for image classification.
         Testproblems with different calculation routines (e.g. RNNs) overwrite this method accordingly.