Update documentation for dataloader_args

TeamEpochGithub · May 16, 2024 · a571cdc · a571cdc
1 parent eb1c207
commit a571cdc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/epochalyst/pipeline/model/training/torch_trainer.py b/epochalyst/pipeline/model/training/torch_trainer.py
@@ -43,6 +43,7 @@ class TorchTrainer(TrainingBlock):
     - `model_name` (str): Name of the model
     - `n_folds` (float): Number of folds for cross validation (0 for train full,
     - `fold` (int): Fold number
+    - `dataloader_args (dict): Arguments for the dataloader`
 
     Methods
     -------