Skip to content

Remove trainer._device_type in favor of check Accelerator class #11002

@four4fish

Description

@four4fish

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.

cc @justusschock @awaelchli @akihironitta

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions