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 |
|