diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py index 723cc1be80cf8282274b8b46e2a70f2e4f08b79f..7fa0b257ac009ea7f7c95dba97140ec6ad19acc8 100644 --- a/climax/main/lr_range_test.py +++ b/climax/main/lr_range_test.py @@ -66,7 +66,7 @@ if __name__ == '__main__': state_file = state_file.replace('.pt', '_lr_test.pt') # path to model state - state_file = MODEL_PATH.joinpath(PREDICTAND, state_file) + state_file = MODEL_PATH.joinpath('lr-range', PREDICTAND, state_file) # initialize logging log_file = MODEL_PATH.joinpath(PREDICTAND,