diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py index 7fa0b257ac009ea7f7c95dba97140ec6ad19acc8..ea3148b60fa4920f47b7f3d1cdb06402c5743d88 100644 --- a/climax/main/lr_range_test.py +++ b/climax/main/lr_range_test.py @@ -69,7 +69,7 @@ if __name__ == '__main__': state_file = MODEL_PATH.joinpath('lr-range', PREDICTAND, state_file) # initialize logging - log_file = MODEL_PATH.joinpath(PREDICTAND, + log_file = MODEL_PATH.joinpath('lr-range', PREDICTAND, state_file.name.replace('.pt', '_log.txt')) if log_file.exists(): log_file.unlink()