lightning/tests/callbacks
Carlos Mocholí 4a8422c2dc
Fix ModelPruning(make_pruning_permanent=True) buffers getting removed when saved during training (#6073)
Co-authored-by: chaton <thomas@grid.ai>
2021-03-03 13:29:58 +01:00
..
__init__.py Progress bar callback (#1450) 2020-04-23 20:46:18 -04:00
test_callback_hook_outputs.py Refactor simplify tests (#5861) 2021-02-08 11:52:02 +01:00
test_callbacks.py Call `optimizer.zero_grad()` before backward inside closure in AutoOpt (#6147) 2021-03-01 14:36:46 +01:00
test_early_stopping.py fix duplicate console logging bug v2 (#6275) 2021-03-02 15:17:55 +05:30
test_finetuning_callback.py [accelerator][FeatBugFix] Improve manual optimization API (#5771) 2021-02-16 16:00:35 -05:00
test_gpu_stats_monitor.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_lambda_function.py Refactor simplify tests (#5861) 2021-02-08 11:52:02 +01:00
test_lr_monitor.py Prune EvalModelTemplate from callbacks and utilities (#6018) 2021-02-16 19:59:57 +00:00
test_progress_bar.py Feature/5275 clean progress bar print (#5470) 2021-02-22 09:40:18 +00:00
test_pruning.py Fix ModelPruning(make_pruning_permanent=True) buffers getting removed when saved during training (#6073) 2021-03-03 13:29:58 +01:00
test_quantization.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_stochastic_weight_avg.py Refactor: runif for spec 6/6 (#6307) 2021-03-02 18:57:13 +00:00