Update

resnet_model/configuration_resnet.py

@@ -20,7 +20,7 @@ class ResnetConfig(PretrainedConfig):
     Defining a model_type for your configuration (here model_type="resnet") is not mandatory,
     unless you want to register your model with the auto classes (see last section)."""
 
-    model_type = "resnet"
+    model_type = "rgbdsod-resnet"
 
     def __init__(
         self,

run.py

@@ -2,11 +2,19 @@ import timm
 
 from resnet_model.configuration_resnet import ResnetConfig
 from resnet_model.modeling_resnet import ResnetModel, ResnetModelForImageClassification
+from transformers import AutoConfig, AutoModel, AutoModelForImageClassification
 
 ResnetConfig.register_for_auto_class()
 ResnetModel.register_for_auto_class("AutoModel")
 ResnetModelForImageClassification.register_for_auto_class("AutoModel")
 
+
+# AutoConfig.register("rgbdsod-resnet", ResnetConfig)
+# AutoModel.register(ResnetConfig, ResnetModel)
+# AutoModelForImageClassification.register(
+#     ResnetConfig, ResnetModelForImageClassification
+# )
+
 resnet50d_config = ResnetConfig(
     block_type="bottleneck", stem_width=32, stem_type="deep", avg_down=True
 )

test.py

@@ -0,0 +1,6 @@
+from transformers import AutoModel
+
+model = AutoModel.from_pretrained(
+    "RGBD-SOD/custom-resnet50d", trust_remote_code=True
+)
+print(model)