pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

Support third-party backend for Timm

Open Tyx-main opened this issue 1 year ago • 1 comments

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 avatar Feb 27 '24 08:02 Tyx-main

cc @rwightman

Tyx-main avatar Feb 28 '24 01:02 Tyx-main