Skip to content

Commit f5aa223

Browse files
committed
fixed output dir bug
1 parent cd2796f commit f5aa223

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

train/src/model_configuration.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,8 @@ def configure_trainer(train):
184184
output_dir = "./train_output"
185185
if os.path.exists(output_dir):
186186
sly.fs.clean_dir(output_dir)
187+
else:
188+
os.mkdir(output_dir)
187189
cfg.OUTPUT_DIR = output_dir
188190

189191
# set training parameters

0 commit comments

Comments
 (0)