-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
feat(mlu): Support MLU Backend #1608
Conversation
@zgplvyou , please help fix the lint problem |
@kennymckormick it seems like the test_rawframe_decode requires libturbojpeg library, but we didn't installed it in our system. That's how i guess. |
LGTM, is it OK to merge this PR now? |
LGTM |
Hi @zgplvyou !First of all, we want to express our gratitude for your significant PR in the mmaction2 project. Your contribution is highly appreciated, and we are grateful for your efforts in helping improve this open-source project during your personal time. We believe that many developers will benefit from your PR. We would also like to invite you to join our Special Interest Group (SIG) private channel on Discord, where you can share your experiences, ideas, and build connections with like-minded peers. To join the SIG channel, simply message moderator— OpenMMLab on Discord or briefly share your open-source contributions in the #introductions channel and we will assist you. Look forward to seeing you there! Join us :https://discord.gg/raweFPmdzG If you have WeChat,welcome to join our community on WeChat. You can add our assistant :openmmlabwx. Please add "mmsig + Github ID" as a remark when adding friends:) |
Support Cambricon MLU device for MMaction2.
Motivation
Support Cambricon MLU device for MMaction2.
Modification
mmaction/apis/train.py : train on the default device.
tools/test.py : test on the default device.
mmaction/utils/distribution_env.py : choose the default device; build ddp and dp model.
BC-breaking (Optional)
No BC-breaking
Use cases (Optional)
you can use the native script of gpu devices to train the model on mlu devices without any change.