Skip to content

Commit 796f18c

Browse files
committed
sample held out workloads
1 parent f463c1d commit 796f18c

2 files changed

Lines changed: 7 additions & 2 deletions

File tree

scoring/generate_held_out_workloads.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99

1010
flags.DEFINE_integer('held_out_workloads_seed',
1111
None,
12-
'Random seed for scoring.')
12+
'Random seed for scoring.'
13+
'AlgoPerf v0.5 seed: 3438810845')
1314
flags.DEFINE_string('output_filename',
1415
'held_out_workloads.json',
1516
'Path to file to record sampled held_out workloads.')
@@ -19,7 +20,10 @@
1920
'librispeech': [
2021
'librispeech_conformer_attention_temperature',
2122
'librispeech_conformer_layernorm',
22-
'librispeech_conformer_gelu'
23+
# 'librispeech_conformer_gelu', # Removed due to bug in target setting procedure
24+
'librispeech_deepspeech_no_resnet',
25+
'librispeech_deepspeech_norm_and_spec_aug',
26+
'librispeech_deepspeech_tanh'
2327
],
2428
'imagenet': [
2529
'imagenet_resnet_silu',
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
["librispeech_conformer_layernorm", "imagenet_resnet_large_bn_init", "ogbg_model_size", "wmt_glu_tanh", "fastmri_tanh", "criteo1tb_embed_init"]

0 commit comments

Comments
 (0)