db7d3b8
1
2
3
4
5
6
from transformers import AutoModel model = AutoModel.from_pretrained( "RGBD-SOD/custom-resnet50d", trust_remote_code=True ) print(model)