Commit
·
8fdfa17
1
Parent(s):
ca78f11
Create results folder if not present
Browse files- 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
|
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 |
-
|
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 |
|