[Feat] Support training on MPS (#331) (2994195b) · Commits · Florian Schiffel / ICV-mmengine_basecode · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

I

ICV-mmengine_basecode
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Environments
- Terraform modules
- Incidents
- Service Desk

Snippets Groups Projects

2994195b

Unverified Commit 2994195b authored 2 years ago by Alex Yang Committed by GitHub 2 years ago

Downloads
- Patches
- Plain Diff

[Feat] Support training on MPS (#331)

* [Feat] Support mps

* fix docstring

parent e877862d

No related branches found

No related tags found

Loading

Changes 3

Hide whitespace changes

Inline Side-by-side

Showing

mmengine/device/__init__.py 2 additions, 2 deletions

mmengine/device/__init__.py
mmengine/device/utils.py 11 additions, 1 deletion

mmengine/device/utils.py
tests/test_device/test_device.py 4 additions, 1 deletion

tests/test_device/test_device.py

with 17 additions and 4 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment

Datenschutz und Impressum