apply suggestions from code review

This commit is contained in:
Benjamin Trom 2024-04-25 10:17:08 +02:00
parent d6c225a112
commit 05a63ef44e

View file

@ -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