Dusan Drevicky 1b06edf2f2 Add the `on_before_optimizer_step` hook (#8048) ... Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>		2021-07-09 13:30:52 +02:00
..
__init__.py	Add XLAStatsMonitor Callback (#8235)	2021-07-05 17:09:46 +05:30
base.py	Add the `on_before_optimizer_step` hook (#8048)	2021-07-09 13:30:52 +02:00
early_stopping.py	fix best score on wrong device in EarlyStopping callback (#8295)	2021-07-06 10:59:33 +02:00
finetuning.py	Fix module dict in base finetuning (#8170)	2021-06-28 10:55:32 +00:00
gpu_stats_monitor.py	Remove on epoch guard from the should stop validation check (#7701)	2021-05-25 15:59:42 +01:00
gradient_accumulation_scheduler.py	Add on_epoch_start to run at the beginning of every loop irrespective of train/val/test (#6498)	2021-03-25 14:20:49 +01:00
lambda_function.py	Add the `on_before_optimizer_step` hook (#8048)	2021-07-09 13:30:52 +02:00
lr_monitor.py	Refactor unnecessary `else` / `elif` when `if` block has a `return` statement (#8156)	2021-06-28 15:27:41 +05:30
model_checkpoint.py	Add auto_insert_metric_name to ModelCheckpoint docstring. (#8310)	2021-07-07 23:15:21 +00:00
prediction_writer.py	Loop Refactor 5/N - Prediction Loop (#7700)	2021-06-23 10:17:04 +01:00
progress.py	rename old `Trainer.train_loop` -> `Trainer.fit_loop` (#8025)	2021-06-22 11:49:32 +02:00
pruning.py	Remove dead code (#7910)	2021-06-10 11:38:33 +01:00
quantization.py	`TrainerState` refactor [5/5] (#7173)	2021-05-04 12:50:56 +02:00
stochastic_weight_avg.py	Bugfix/swa iterable dset (#8172)	2021-06-28 21:18:25 +00:00
timer.py	fix info message when max training time reached (#7780)	2021-05-31 14:50:16 +02:00
xla_stats_monitor.py	Add XLAStatsMonitor Callback (#8235)	2021-07-05 17:09:46 +05:30