Carlos Mocholí 6ce77a102b Set minimum PyTorch version to 1.6 (#8288) ... Co-authored-by: Jirka <jirka.borovec@seznam.cz>		2021-07-13 17:12:49 +00: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	`every_n_val_epochs` -> `every_n_epochs` (#8383)	2021-07-13 01:20:20 +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 `ModelCheckpoint(save_on_train_epoch_end)` (#8389)	2021-07-13 14:47:59 +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	Set minimum PyTorch version to 1.6 (#8288)	2021-07-13 17:12:49 +00: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