fix docstrings

analysiscenter · Jul 16, 2024 · 3d68e21 · 3d68e21
1 parent 38d68aa
commit 3d68e21
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py
@@ -1042,13 +1042,13 @@ def train(self, inputs, targets, outputs=None, mode='train', lock=True, profile=
 
         Returns
         -------
-        Calculated values of requested tensors from `outputs` in the same order.
+        Calculated values of requested tensors from `outputs`.
 
         Examples
         --------
         .. code-block:: python
 
-            model.train(B('images'), B('labels'), fetches='loss')
+            model.train(B('images'), B('labels'), outputs='loss')
         """
         if self.disable_training:
             raise RuntimeError('Training model after ONNX conversion is not allowed!')
@@ -1343,7 +1343,7 @@ def predict(self, inputs, targets=None, outputs=None, lock=True, microbatch_size
 
         Returns
         -------
-        Calculated values of tensors in `outputs` in the same order.
+        Calculated values of tensors in `outputs`.
 
         Examples
         --------