Fix get_latest_run to search 'search_dir' recursivly

utils/utils.py

@@ -36,10 +36,10 @@ def init_seeds(seed=0):
     np.random.seed(seed)
     torch_utils.init_seeds(seed=seed)
 
-def get_latest_run(search_dir = './runs/'):
+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')
+    last_list =  glob.glob(f'{search_dir}/**/last.pt', recursive=True)
     latest = max(last_list, key = os.path.getctime)
     return latest