fix: create folder for snapshots

vejvarm · Nov 27, 2023 · 5917195 · 5917195
1 parent 4cff1a7
commit 5917195
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -115,6 +115,7 @@ def main():
     LOGGER.info(f'Epochs: {args.epochs}')
     LOGGER.info(f'Batch size: {args.batch_size}')
 
+    LOGDIR.joinpath("tb").mkdir(parents=True, exist_ok=True)
     tb_writer = SummaryWriter(LOGDIR.joinpath("tb"))
 
     # run epochs