diff --git a/climax/main/downscale_infer.py b/climax/main/downscale_infer.py index 6f3348d7b6fa65e498405fbe73f547ccf5dc9083..eedd85aec282d5e85a526042abf9f3eaaf596a27 100644 --- a/climax/main/downscale_infer.py +++ b/climax/main/downscale_infer.py @@ -69,8 +69,8 @@ if __name__ == '__main__': sys.exit() # initialize logging - log_file = MODEL_PATH.joinpath(PREDICTAND, - state_file.name.replace('.pt', '_log.txt')) + log_file = state_file.parent.joinpath( + state_file.name.replace(state_file.suffix, '_log.txt')) dictConfig(log_conf(log_file)) # predict reference period diff --git a/climax/main/downscale_train.py b/climax/main/downscale_train.py index 2ae11a250f011ec1ddd560144dd366ee786da5c1..9dc09aba53dff5758cfa34af12973858e491353f 100644 --- a/climax/main/downscale_train.py +++ b/climax/main/downscale_train.py @@ -54,8 +54,8 @@ if __name__ == '__main__': state_file = MODEL_PATH.joinpath(PREDICTAND, state_file) # initialize logging - log_file = MODEL_PATH.joinpath(PREDICTAND, - state_file.name.replace('.pt', '_log.txt')) + log_file = state_file.parent.joinpath( + state_file.name.replace(state_file.suffix, '_log.txt')) if log_file.exists(): log_file.unlink() dictConfig(log_conf(log_file))