Check if model already exists.

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