Spaces:

xiang-wuu
/

yolov5

Runtime error

astoken commited on Jun 17, 2020

Commit

d342917

1 Parent(s): 5f2eeba

Fix get_latest_run() to search inside 'weights' subfolders

Files changed (1) hide show

utils/utils.py CHANGED Viewed

@@ -39,7 +39,7 @@ def init_seeds(seed=0):
 def get_latest_run(search_dir = './runs/'):
     # get path to most recent 'last.pt' in run dirs
     # assumes most recently saved 'last.pt' is the desired weights to --resume from
-    last_list =  glob.glob('runs/*/last.pt')
     latest = max(last_list, key = os.path.getctime)
     return latest

 def get_latest_run(search_dir = './runs/'):
     # get path to most recent 'last.pt' in run dirs
     # assumes most recently saved 'last.pt' is the desired weights to --resume from
+    last_list =  glob.glob('runs/*/weights/last.pt')
     latest = max(last_list, key = os.path.getctime)
     return latest