From 05a63ef44ef75cc7a5d271d0c395fc1c47e35d00 Mon Sep 17 00:00:00 2001 From: Benjamin Trom Date: Thu, 25 Apr 2024 10:17:08 +0200 Subject: [PATCH] apply suggestions from code review --- src/refiners/training_utils/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/refiners/training_utils/data_loader.py b/src/refiners/training_utils/data_loader.py index 9a3ac74..f6b75aa 100644 --- a/src/refiners/training_utils/data_loader.py +++ b/src/refiners/training_utils/data_loader.py @@ -6,7 +6,7 @@ from torch.utils.data import DataLoader, Dataset BatchT = TypeVar("BatchT") -class DataloaderConfig(BaseModel): +class DataLoaderConfig(BaseModel): batch_size: PositiveInt = 1 num_workers: int = 0 pin_memory: bool = False