From d411f4ed753f9a577aa5320702c52ab6df3f4b9a Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 22 Oct 2021 16:24:35 +0200
Subject: [PATCH] Check if model already exists.
---
climax/main/downscale_train.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/climax/main/downscale_train.py b/climax/main/downscale_train.py
index a076496..9bf803d 100644
--- a/climax/main/downscale_train.py
+++ b/climax/main/downscale_train.py
@@ -18,7 +18,6 @@ from torch.utils.data import DataLoader
# locals
from pysegcnn.core.utils import search_files
from pysegcnn.core.trainer import NetworkTrainer, LogConfig
-from pysegcnn.core.models import Network
from pysegcnn.core.logging import log_conf
from climax.core.dataset import ERA5Dataset, NetCDFDataset
from climax.core.loss import MSELoss, L1Loss
@@ -64,7 +63,7 @@ if __name__ == '__main__':
# check if model exists
if state_file.exists() and not OVERWRITE:
# load pretrained network
- net, _ = Network.load_pretrained_model(state_file, NET)
+ LogConfig.init_log('{} already exists.'.format(state_file))
sys.exit()
# initialize ERA5 predictor dataset
--
GitLab