lightning/pytorch_lightning/callbacks
Rohit Gupta 6701526478
Add typing to callbacks (#10001)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-10-20 10:22:03 +00:00
..
progress use ModuleNotFoundError instead of ImportError (#9867) 2021-10-11 05:57:37 +00:00
__init__.py [2/4] Add DeviceStatsMonitor callback (#9712) 2021-10-13 18:29:36 +00:00
base.py update docs (#9903) 2021-10-12 15:37:08 +05:30
device_stats_monitor.py [2/4] Add DeviceStatsMonitor callback (#9712) 2021-10-13 18:29:36 +00:00
early_stopping.py Add typing to callbacks (#10001) 2021-10-20 10:22:03 +00:00
finetuning.py rename callback FineTune arg `round` (#9711) 2021-10-06 09:39:36 +01:00
gpu_stats_monitor.py Add typing to callbacks (#10001) 2021-10-20 10:22:03 +00:00
gradient_accumulation_scheduler.py Add typing to callbacks (#10001) 2021-10-20 10:22:03 +00:00
lambda_function.py Add on_exception callback hook (#9183) 2021-09-01 10:49:00 +02:00
lr_monitor.py Add typing to callbacks (#10001) 2021-10-20 10:22:03 +00:00
model_checkpoint.py Remove `should_rank_save_checkpoint` property from Trainer (#9433) 2021-10-13 11:36:24 +00:00
model_summary.py fix mypy typing for model summary (#9447) 2021-09-13 11:41:27 +02:00
prediction_writer.py [bugfix] Always return batch indices to prevent duplicated logic for the users (#9432) 2021-09-14 14:40:19 +00:00
pruning.py Add typing to callbacks (#10001) 2021-10-20 10:22:03 +00:00
quantization.py Fix `_compare_version` and add `_TORCH_GREATER_EQUAL_DEV_1_10` (#9960) 2021-10-19 14:42:46 +00:00
rich_model_summary.py use ModuleNotFoundError instead of ImportError (#9867) 2021-10-11 05:57:37 +00:00
stochastic_weight_avg.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
timer.py Call TrainingTypePlugin collective functions directly instead of going through the Accelerator (#9677) 2021-09-27 14:52:57 +02:00
xla_stats_monitor.py Deprecate `GPUStatsMonitor` and `XLAStatsMonitor` in favor of `DeviceStatsMonitor` (#9924) 2021-10-14 15:52:45 +00:00