lightning/tests/tests_pytorch/trainer/optimization
Justus Schock f54abc506f
Merge pull request #13123 from Lightning-AI/mps_accelerator
MPS Accelerator
2022-06-24 08:15:48 -04:00
..
__init__.py
test_backward_calls.py
test_manual_optimization.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_multiple_optimizers.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_optimizers.py