-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Description
Proposed refactor
Follow up for #11001: Generalize internal checks for precision plugin type, training type, accelerator type
Motivation
Code simplification
Pitch
After #11001, _device_type is not needed anymore
@property
def _device_type(self) -> _AcceleratorType:
return self._accelerator_connector.
Instead in tests and where we need check device type, use
isinstance(trainer.accelerator, XAccelerator)
Additional context
If you enjoy Lightning, check out our other projects! ⚡
-
Metrics: Machine learning metrics for distributed, scalable PyTorch applications.
-
Lite: enables pure PyTorch users to scale their existing code on any kind of device while retaining full control over their own loops and optimization logic.
-
Flash: The fastest way to get a Lightning baseline! A collection of tasks for fast prototyping, baselining, fine-tuning, and solving problems with deep learning.
-
Bolts: Pretrained SOTA Deep Learning models, callbacks, and more for research and production with PyTorch Lightning and PyTorch.
-
Lightning Transformers: Flexible interface for high-performance research using SOTA Transformers leveraging Pytorch Lightning, Transformers, and Hydra.