Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

apply suggestions from code review

Browse source
This commit is contained in:
Benjamin Trom 2024-04-25 10:17:08 +02:00
parent d6c225a112
commit 05a63ef44e
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/training_utils/data_loader.py
View file

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