soumyaprabhamaiti commited on
Commit
8fdfa17
·
1 Parent(s): ca78f11

Create results folder if not present

Browse files
Files changed (1) hide show
  1. pet_seg_core/train.py +2 -2
pet_seg_core/train.py CHANGED
@@ -2,7 +2,7 @@ import lightning as pl
2
  from lightning.pytorch.callbacks import ModelCheckpoint
3
  from lightning.pytorch.loggers import CSVLogger
4
  from datetime import datetime
5
- import os
6
 
7
  from pet_seg_core.config import PetSegTrainConfig
8
  from pet_seg_core.data import train_dataloader, val_dataloader
@@ -11,7 +11,7 @@ from pet_seg_core.model import UNet
11
  def train():
12
  curr_time = datetime.now().strftime('%Y-%m-%d_%H:%M:%S.%f')
13
  results_folder = f"results/{curr_time}"
14
- os.mkdir(results_folder)
15
  with open(f"{results_folder}/description.txt", "w") as f:
16
  f.write(PetSegTrainConfig.DESCRIPTION_TEXT)
17
 
 
2
  from lightning.pytorch.callbacks import ModelCheckpoint
3
  from lightning.pytorch.loggers import CSVLogger
4
  from datetime import datetime
5
+ from pathlib import Path
6
 
7
  from pet_seg_core.config import PetSegTrainConfig
8
  from pet_seg_core.data import train_dataloader, val_dataloader
 
11
  def train():
12
  curr_time = datetime.now().strftime('%Y-%m-%d_%H:%M:%S.%f')
13
  results_folder = f"results/{curr_time}"
14
+ Path(results_folder).mkdir(parents=True, exist_ok=True)
15
  with open(f"{results_folder}/description.txt", "w") as f:
16
  f.write(PetSegTrainConfig.DESCRIPTION_TEXT)
17