lightning/pytorch_lightning
ORippler 456cc87954
Fuse_modules in a qat-respecting way (#12891)
* Fuse_modules in a qat-respecting way

* Add compatibility for PyTorch <1.11

In older pytorch versions, `fuse_modules` used the `Module.training`
flag to determine wheter fusion should be QAT-compliant or not, refer
https://github.com/pytorch/pytorch/releases/tag/v1.11.0

* Add CHANGELOG for pull #12891

* Fix conditional import of fuse_modules_qat

`torch.ao.quantization.fuse_modules_qat` was actually added in
torch 1.11.

* Update CHANGELOG.md

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
2022-05-02 12:37:08 +00:00
..
accelerators Add cpu device parser to validate cpu devices (#12160) 2022-03-28 20:33:04 +05:30
callbacks Fuse_modules in a qat-respecting way (#12891) 2022-05-02 12:37:08 +00:00
core Fix typo in predict_step docs (#12911) 2022-04-28 12:26:57 +00:00
demos docs refactor 3/n (#12795) 2022-04-19 14:15:47 -04:00
distributed Renamed the `DDPSpawnPlugin` to `DDPSpawnStrategy` (#11145) 2021-12-21 23:06:14 +00:00
lite Support auto_select_gpus with accelerator and devices api (#12608) 2022-04-12 21:28:54 +00:00
loggers Remove deprecated `TestTubeLogger` (#12859) 2022-04-24 20:05:48 +02:00
loops Print ragged dict of metrics in `EvaluationLoop._print_results` properly (#12857) 2022-04-28 16:05:24 +00:00
overrides Fix import error when `distributed` module not available (#12794) 2022-04-20 17:28:39 +00:00
plugins ShardedGradScaler should only be set for FP16 (#12915) 2022-04-28 17:44:31 +02:00
profiler Drop PyTorch 1.7 support (#12432) 2022-03-27 21:31:20 +00:00
strategies Override `optimizer_zero_grad` when using the `IPUStrategy` (#12913) 2022-05-01 22:47:21 +00:00
trainer Fix `trainer.logger` deprecation message (#12671) 2022-04-27 16:11:34 +02:00
tuner Support auto_select_gpus with accelerator and devices api (#12608) 2022-04-12 21:28:54 +00:00
utilities Fuse_modules in a qat-respecting way (#12891) 2022-05-02 12:37:08 +00:00
__about__.py Update version after the 1.6.0 release (#12512) 2022-03-30 08:10:59 +02:00
__init__.py Add DETAIL logs for batch use cases (#11008) 2022-01-12 01:22:48 +01:00
py.typed
setup_tools.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00