diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py index f15e5c29cb4650e8714c4d71beebc437e4cf4c03..723cc1be80cf8282274b8b46e2a70f2e4f08b79f 100644 --- a/climax/main/lr_range_test.py +++ b/climax/main/lr_range_test.py @@ -34,7 +34,7 @@ LOGGER = logging.getLogger(__name__) # network training configuration TRAIN_CONFIG = { 'checkpoint_state': {}, - 'epochs': 20, + 'epochs': 25, 'save': True, 'save_loaders': False, 'early_stop': False,