runs/exp increment bug fix #344

This commit is contained in:
Glenn Jocher committed 2020-07-15 16:07:40 -07:00
1 parent f796708b50
commit 98f4a5af87
1 file changed
+2 -2
+2 -2
View File
@@ -902,10 +902,10 @@ def output_to_target(output, width, height):
def increment_dir(dir, comment=''):
# Increments a directory runs/exp1 --> runs/exp2_comment
n = 0 # number
dir = str(Path(dir)) # os-agnostic
d = sorted(glob.glob(dir + '*')) # directories
if len(d):
d = d[-1].replace(dir, '')
n = int(d[:d.find('_')] if '_' in d else d) + 1 # increment
n = max([int(x[len(dir):x.find('_') if '_' in x else None]) for x in d]) + 1 # increment
return dir + str(n) + ('_' + comment if comment else '')