Skip to content

Commit ad1b6e8

Browse files
authored
Merge pull request #196 from xyhuang/cherry-pick-187
Add a CosmoFlow RCP for BS512 (#187)
2 parents be51f7d + b200031 commit ad1b6e8

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

mlperf_logging/rcp_checker/hpc_1.0.0/rcps_cosmoflow.json

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,22 @@
3131
"opt_weight_decay": 0.0
3232
}
3333
},
34+
"cosmoflow_ref_512": {
35+
"Benchmark": "cosmoflow",
36+
"BS": 512,
37+
"Epochs to converge": [ 22, 23, 23, 24, 24, 21, 23, 23, 23, 23, 24, 23, 23, 21, 23, 23, 21, 23, 23, 23 ],
38+
"Hyperparams": {
39+
"global_batch_size": 512,
40+
"opt_name": "SGD",
41+
"opt_base_learning_rate": 0.006,
42+
"opt_learning_rate_warmup_epochs": 6,
43+
"opt_learning_rate_warmup_factor": 4.0,
44+
"opt_learning_rate_decay_boundary_epochs": [ 19, 21, 22, 23 ],
45+
"opt_learning_rate_decay_factor": [0.5, 0.25, 0.125, 0.0625],
46+
"dropout": 0.0,
47+
"opt_weight_decay": 0.01
48+
}
49+
},
3450
"cosmoflow_ref_1024": {
3551
"Benchmark": "cosmoflow",
3652
"BS": 1024,

0 commit comments

Comments
 (0)