lightning/pytorch_lightning/core
chaton be255de306
Bugfix/all gather (#5221)
* resolve bug

* add tests

* add tests

* resolve flake8

* update

* update

* remove globals

* typo

* Update pytorch_lightning/utilities/distributed.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* update

* update

* add suport int, float

* update

* resolve pep8

* Update pytorch_lightning/core/lightning.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* Update tests/utilities/test_all_gather_grad.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* update doc

* add bool and np.ndarray

* resolve conflicts

* resolve conflicts

* resolve pep8

* add changelog

* Update pytorch_lightning/core/lightning.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-109.ec2.internal>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-01-09 07:37:44 -05:00
..
__init__.py CI: add flake8 (#4239) 2020-10-19 21:20:17 +01:00
datamodule.py logger docs and api docs (#3950) 2020-11-13 20:35:54 +05:30
decorators.py logger docs and api docs (#3950) 2020-11-13 20:35:54 +05:30
grads.py PyTorch 1.7 Stable support (#3821) 2020-10-30 15:42:14 +00:00
hooks.py Allow any input in to_onnx and to_torchscript (#4378) 2020-12-12 18:17:03 +08:00
lightning.py Bugfix/all gather (#5221) 2021-01-09 07:37:44 -05:00
memory.py Implemented ModelSummary total params values (#4521) 2020-11-22 07:07:52 +01:00
optimizer.py fix/enable - check F401 (#5201) 2020-12-21 10:15:04 +01:00
saving.py refactor - check E501 (#5200) 2020-12-21 14:23:09 +05:30
step_result.py [bug-fix] Metric reduction with Logging (#5150) 2021-01-05 09:58:37 +01:00