Orca pytorch load checkpoint bugfix (intel-analytics#3115)

* Orca pytorch load checkpoint bugfix * fix
yangw1234 · Sep 26, 2021 · edcd5cf · edcd5cf
1 parent 66bd22a
commit edcd5cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/orca/src/bigdl/orca/learn/utils.py b/python/orca/src/bigdl/orca/learn/utils.py
@@ -26,7 +26,7 @@ def find_latest_checkpoint(model_dir, model_type="bigdl"):
     if model_type == "bigdl":
         optim_regex = ".*\.[0-9]+$"
     elif model_type == "pytorch":
-        optim_regex = "TorchModel[0-9a-z]{8}\.[0-9]+$"
+        optim_regex = "TorchModel[0-9a-z]*\.[0-9]+$"
     elif model_type == "tf":
         optim_regex = "TFParkTraining\.[0-9]+$"
     else: