Skip to content

Commit ff428ea

Browse files
committed
fix: move batch_size to dataloader configurations
1 parent 5b43c67 commit ff428ea

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

config.yaml

+2-1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,8 @@ dataloader:
3232
num_classes: 1000
3333
# Path to dataset directory (currently only ImageNet supported)
3434
path: /localhome/fair/Dataset/imagenet
35+
# Size of mini-batch
36+
batch_size: 64
3537
# Number of data loading workers
3638
workers: 32
3739
# Seeds random generators in a deterministic way (i.e., set all the seeds 0).
@@ -107,7 +109,6 @@ quan:
107109
# If this field is true, all the bellowing options will be ignored
108110
eval: false
109111

110-
batch_size: 64
111112
epochs: 90
112113

113114
optimizer:

0 commit comments

Comments
 (0)