From 24d004a89bc2be3565cb76980ac9666f39a4aff8 Mon Sep 17 00:00:00 2001 From: chenyu Date: Sat, 23 Mar 2024 17:16:38 -0400 Subject: [PATCH] hotfix check ckpts before writing achieved model (#3901) this killed tinybox green run --- examples/mlperf/model_train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/mlperf/model_train.py b/examples/mlperf/model_train.py index f11f6ddb2b..25fea1e6d8 100644 --- a/examples/mlperf/model_train.py +++ b/examples/mlperf/model_train.py @@ -200,6 +200,7 @@ def train_resnet(): # save model if achieved target if not achieved and total_top_1 >= target: + if not os.path.exists("./ckpts"): os.mkdir("./ckpts") fn = f"./ckpts/resnet50.safe" safe_save(get_state_dict(model), fn) print(f" *** Model saved to {fn} ***")