Skip to content

Commit 66c9c3b

Browse files
committed
Lower WIC batch size default to 160.
1 parent 3e06129 commit 66c9c3b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scoutbot/wic/dataloader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import torchvision
88
import utool as ut
99

10-
BATCH_SIZE = int(os.getenv('WIC_BATCH_SIZE', 256))
10+
BATCH_SIZE = int(os.getenv('WIC_BATCH_SIZE', 160))
1111
INPUT_SIZE = 224
1212

1313

0 commit comments

Comments
 (0)