From a571cdc73aeda7494a6b118d485ec4f3de17c39c Mon Sep 17 00:00:00 2001 From: Schobbejak Date: Thu, 16 May 2024 10:07:00 +0200 Subject: [PATCH] Update documentation for dataloader_args --- epochalyst/pipeline/model/training/torch_trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/epochalyst/pipeline/model/training/torch_trainer.py b/epochalyst/pipeline/model/training/torch_trainer.py index 1dc0bfb..adeb326 100644 --- 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 -------