Skip to content

Commit 2d2eb30

Browse files
authored
Fix test05 seeds missing error for v5.0 submission checker (#1976)
1 parent 8948934 commit 2d2eb30

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

‎tools/submission/submission_checker.py‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -827,7 +827,8 @@ def __init__(
827827
self.version = version
828828
self.models = self.base["models"]
829829
self.seeds = self.base["seeds"]
830-
self.test05_seeds = self.base["test05_seeds"]
830+
if self.base.get("test05_seeds"):
831+
self.test05_seeds = self.base["test05_seeds"]
831832
self.accuracy_target = self.base["accuracy-target"]
832833
self.accuracy_delta_perc = self.base["accuracy-delta-perc"]
833834
self.accuracy_upper_limit = self.base.get("accuracy-upper-limit", {})
@@ -968,7 +969,7 @@ def get_args():
968969
parser.add_argument("--input", required=True, help="submission directory")
969970
parser.add_argument(
970971
"--version",
971-
default="v4.1",
972+
default="v5.0",
972973
choices=list(MODEL_CONFIG.keys()),
973974
help="mlperf version",
974975
)

0 commit comments

Comments
 (0)