From edcd5cf95bfb28c14b2b6f3609dd59a6b9aa04d8 Mon Sep 17 00:00:00 2001
From: Yina Chen <33650826+cyita@users.noreply.github.com>
Date: Tue, 24 Nov 2020 14:42:11 +0800
Subject: [PATCH] Orca pytorch load checkpoint bugfix (#3115)

* Orca pytorch load checkpoint bugfix

* fix
---
 python/orca/src/bigdl/orca/learn/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/orca/src/bigdl/orca/learn/utils.py b/python/orca/src/bigdl/orca/learn/utils.py
index 203731dbcb8..e6e29c41143 100644
--- 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: