Carlos Mocholí dbea5bb710 Add typing to `ModelPruning` callback (#7529) ... Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>		2021-05-19 22:01:42 +02:00
..
__init__.py	…
base.py	…
early_stopping.py	…
finetuning.py	refactor optimizer loop logic for manual and automatic optimization (#7526)	2021-05-17 14:42:01 +02:00
gpu_stats_monitor.py	…
gradient_accumulation_scheduler.py	…
lambda_function.py	…
lr_monitor.py	…
model_checkpoint.py	Support TPU Pod Training (n/n) (#7296)	2021-05-17 11:33:44 +00:00
prediction_writer.py	…
progress.py	…
pruning.py	Add typing to `ModelPruning` callback (#7529)	2021-05-19 22:01:42 +02:00
quantization.py	…
stochastic_weight_avg.py	…
timer.py	…