Skip to content
This repository was archived by the owner on Jun 22, 2022. It is now read-only.

Commit 79224f2

Browse files
authored
Update neptune.yaml
1 parent 8864784 commit 79224f2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

configs/neptune.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@ parameters:
6868

6969
# Training schedule
7070
epochs_nr: 100
71-
batch_size_train: 8
72-
batch_size_inference: 8
71+
batch_size_train: 1
72+
batch_size_inference: 1
7373
lr: 0.00001
7474
momentum: 0.9
7575
gamma: 1.0

0 commit comments

Comments
 (0)