chaton 7b00894130 [feat] Add StochasticWeightAveragingCallback (#5640) ... * add swa callback * switch back to 1.6.0 * remove optimizer_step * move super * update * forgot update_parameters * update on comments * works for ddp * resolve flake8 * remove set_model * resolve flake8 * resolve cpu * resolve flake8 * resolve flake8 * update * update on comments		2021-02-11 00:05:59 +00:00
..
__init__.py	[feat] Add StochasticWeightAveragingCallback (#5640)	2021-02-11 00:05:59 +00:00
base.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
early_stopping.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
finetuning.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
gpu_stats_monitor.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
gradient_accumulation_scheduler.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
lambda_function.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
lr_monitor.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
model_checkpoint.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
progress.py	formatting 3/n: PL modules (#5716)	2021-02-08 14:28:38 -05:00
pruning.py	Fix Pruning callback and add a few features (#5825)	2021-02-10 15:03:23 +00:00
swa.py	[feat] Add StochasticWeightAveragingCallback (#5640)	2021-02-11 00:05:59 +00:00