diff --git a/climax/main/downscale_train.py b/climax/main/downscale_train.py
index a07649645fe10e4c746643f900f629dca06d5b43..9bf803df9a00c015c0d548f39f0e5afe61bd0039 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