Skip to content

Commit 641af7f

Browse files
committed
Fix benchmark to use standard epochs argument.
1 parent 716ff6b commit 641af7f

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

Diff for: Pilot3/P3B7/default_model.txt

+2-3
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,8 @@ kernel3 = 5
1010
embed_dim = 300
1111
n_filters = 100
1212
batch_size = 50
13-
num_epochs = 10
13+
epochs = 10
1414
rng_seed = 13
1515
weight_decay = 1e-3
1616
use_synthetic_data = 'True'
17-
epochs = 10
18-
device = 'cpu'
17+
device = 'cpu'

Diff for: Pilot3/P3B7/p3b7.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
'weight_decay',
3434
'rng_seed',
3535
'batch_size',
36-
'num_epochs',
36+
'epochs',
3737
]
3838

3939

0 commit comments

Comments
 (0)