Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Closed
four4fish opened this issue Dec 9, 2021 · 2 comments
Closed

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

four4fish opened this issue Dec 9, 2021 · 2 comments
Assignees
Milestone

Comments

@four4fish
Copy link
Contributor

four4fish commented Dec 9, 2021

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

@awaelchli
Copy link
Contributor

awaelchli commented Dec 9, 2021

Hey @four4fish
Thanks for looking at this.
This is not something we need to deprecate because it's already declared as internal, not exposed to user.
It can simply be removed together with the last part of #10821

@four4fish four4fish changed the title Deprecate trainer._device_type in favor of check Accelerator class Remove trainer._device_type in favor of check Accelerator class Dec 9, 2021
@four4fish four4fish added this to the 1.5.x milestone Dec 9, 2021
@four4fish
Copy link
Contributor Author

Solved by #11992

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants