upgrade

compo-singleone-v2-dev-acc.py

@@ -92,6 +92,9 @@ else:
 
 #######################################################
 
+def get_model_simplified_name_from_dirname(dirname):
+    return dirname.replace("model_","").replace("_864x","").replace("_864","").replace("_new","")
+
 def get_padded_checkpoint_no_from_filename(checkpoint_filename):
     match = re.search(r'ckpt-(\d+)', checkpoint_filename)
     if match:
@@ -167,7 +170,7 @@ def setup(opts):
     # checkpoint3_dir = os.path.join(path, model3_name, 'checkpoint_long')
     print("-----------------------------------------")
     print("modelname is : " + model_name)
-    found_model=model_name
+    found_model=get_model_simplified_name_from_dirname(model_name)
     
     #print("model2name is : " + model2_name)
     # print("model3name is : " + model3_name)