From 3d68e21d4667c5913503be26f43bb57e49ede4f8 Mon Sep 17 00:00:00 2001
From: BulatVakhitov <vakhitov.bd@gmail.com>
Date: Tue, 16 Jul 2024 12:52:29 +0000
Subject: [PATCH] fix docstrings

---
 batchflow/models/torch/base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py
index 7e952aabd..94b3d4b82 100755
--- 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
         --------