File size: 389 Bytes
74b941a cd555aa 74b941a cd555aa 74b941a |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers import AutoConfig
from modeling.modeling_resnet import ResNetCustomForImageClassification
cfg = AutoConfig.from_pretrained("microsoft/resnet-18")
ResNetCustomForImageClassification.register_for_auto_class("AutoModelForImageClassification")
model = ResNetCustomForImageClassification(cfg)
model.save_pretrained("/home/fxmarty/hf_internship/tiny-testing-remote-code")
|