mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
apply suggestions from code review
This commit is contained in:
parent
d6c225a112
commit
05a63ef44e
|
@ -6,7 +6,7 @@ from torch.utils.data import DataLoader, Dataset
|
||||||
BatchT = TypeVar("BatchT")
|
BatchT = TypeVar("BatchT")
|
||||||
|
|
||||||
|
|
||||||
class DataloaderConfig(BaseModel):
|
class DataLoaderConfig(BaseModel):
|
||||||
batch_size: PositiveInt = 1
|
batch_size: PositiveInt = 1
|
||||||
num_workers: int = 0
|
num_workers: int = 0
|
||||||
pin_memory: bool = False
|
pin_memory: bool = False
|
||||||
|
|
Loading…
Reference in a new issue