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