Transformers How to use FreddeFrallan/custom-resnet50d2 with Transformers:
# Load model directly
from transformers import sitionalInvariantTransformation
model = sitionalInvariantTransformation.from_pretrained("FreddeFrallan/custom-resnet50d2", dtype="auto") 
