Commit
·
f783fb4
1
Parent(s):
16f7114
Fix timestamp format in training results folder creation to prevent error on windows
Browse files- pet_seg_core/train.py +1 -1
pet_seg_core/train.py
CHANGED
@@ -9,7 +9,7 @@ from pet_seg_core.data import train_dataloader, val_dataloader
|
|
9 |
from pet_seg_core.model import UNet
|
10 |
|
11 |
def train():
|
12 |
-
curr_time = datetime.now().strftime('%Y-%m-%d_%H
|
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:
|
|
|
9 |
from pet_seg_core.model import UNet
|
10 |
|
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:
|