diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py index c5626d16084b4d5bc80dc851fc291b978f51c2e2..e789ac9ce85449894e0a758f9918de9c17045a95 100644 --- a/climax/main/lr_range_test.py +++ b/climax/main/lr_range_test.py @@ -36,7 +36,7 @@ LOGGER = logging.getLogger(__name__) # network training configuration TRAIN_CONFIG = { 'checkpoint_state': {}, - 'epochs': 50, + 'epochs': 75, 'save': True, 'save_loaders': False, 'early_stop': False,