Skip to content

Conversation

@Tyx-main
Copy link

@Tyx-main Tyx-main commented Feb 27, 2024

For Timm supporting third-party backend-NPU, here is a PR opened for compatible.

Note:
We can specify a config.yaml as the value of the ‘config’ variable to activate a third-party backend:

dist_backend: hccl 
device_modules: [torch_npu]

@Tyx-main
Copy link
Author

cc @rwightman

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants