pytorch-image-models
pytorch-image-models copied to clipboard
Support third-party backend for Timm
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]
cc @rwightman