Skip to content

Commit f52007a

Browse files
committed
fix: bump each epoch length to 1000
1 parent 1ba7db4 commit f52007a

File tree

5 files changed

+10
-10
lines changed

5 files changed

+10
-10
lines changed

src/templates/template-common/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@ train_batch_size: 32
44
eval_batch_size: 32
55
num_workers: 4
66
max_epochs: 20
7-
train_epoch_length: null
8-
eval_epoch_length: null
7+
train_epoch_length: 1000
8+
eval_epoch_length: 1000
99
use_amp: false
1010
debug: false
1111

src/templates/template-text-classification/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@ train_batch_size: 32
44
eval_batch_size: 32
55
num_workers: 4
66
max_epochs: 20
7-
train_epoch_length: null
8-
eval_epoch_length: null
7+
train_epoch_length: 1000
8+
eval_epoch_length: 1000
99
use_amp: false
1010
debug: false
1111
model: bert-base-uncased

src/templates/template-vision-classification/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@ train_batch_size: 32
44
eval_batch_size: 32
55
num_workers: 4
66
max_epochs: 20
7-
train_epoch_length: null
8-
eval_epoch_length: null
7+
train_epoch_length: 1000
8+
eval_epoch_length: 1000
99
lr: 0.0001
1010
use_amp: false
1111
debug: false

src/templates/template-vision-dcgan/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@ train_batch_size: 32
44
eval_batch_size: 32
55
num_workers: 4
66
max_epochs: 20
7-
train_epoch_length: null
8-
eval_epoch_length: null
7+
train_epoch_length: 1000
8+
eval_epoch_length: 1000
99
lr: 0.0001
1010
use_amp: false
1111
debug: false

src/templates/template-vision-segmentation/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@ train_batch_size: 32
44
eval_batch_size: 32
55
num_workers: 4
66
max_epochs: 20
7-
train_epoch_length: null
8-
eval_epoch_length: null
7+
train_epoch_length: 1000
8+
eval_epoch_length: 1000
99
lr: 0.007
1010
use_amp: false
1111
debug: false

0 commit comments

Comments
 (0)