lightning/tests/callbacks
chaton c9622bafe0
[feat] Add Trainer(stochastic_weight_avg=True/False) (#6038)
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-02-17 23:10:05 +00:00
..
__init__.py
test_callback_hook_outputs.py Refactor simplify tests (#5861) 2021-02-08 11:52:02 +01:00
test_callbacks.py [BugFix] Add on_epoch_end hook after on_test/validation_epoch_end hook (#5986) 2021-02-17 18:21:49 +01:00
test_early_stopping.py Prune EvalModelTemplate from callbacks and utilities (#6018) 2021-02-16 19:59:57 +00:00
test_finetuning_callback.py [accelerator][FeatBugFix] Improve manual optimization API (#5771) 2021-02-16 16:00:35 -05:00
test_gpu_stats_monitor.py Prune EvalModelTemplate from callbacks and utilities (#6018) 2021-02-16 19:59:57 +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 Prevent flickering progress bar (#6009) 2021-02-17 19:01:51 +00:00
test_pruning.py Fix Pruning callback and add a few features (#5825) 2021-02-10 15:03:23 +00:00
test_quantization.py Quantisation (#5706) 2021-02-11 07:04:57 -05:00
test_swa.py [feat] Add Trainer(stochastic_weight_avg=True/False) (#6038) 2021-02-17 23:10:05 +00:00