my torch vision is 2.1.0
rename_privateuse1_backend
is defined in the torch.utils
namespace so using torch.utils.rename_privateuse1_backend(backend_name)
should work.
thanks so much! this function solved my question
1 Like