Hugging Face's logo

young-man
/
custom-resnet50d

Feature Extraction
Transformers
PyTorch
resnet
custom_code
Model card Files Community
young-man commited on
Commit
0a1f279
•
1 Parent(s): 53f54f5

Upload model

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,10 +1,10 @@
1
  {
2
  "architectures": [
3
- "ResnetModelForImageClassification"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_resnet.ResnetConfig",
7
- "AutoModelForImageClassification": "modeling_resnet.ResnetModelForImageClassification"
8
  },
9
  "avg_down": true,
10
  "base_width": 64,
 
1
  {
2
  "architectures": [
3
+ "ResnetModel"
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_resnet.ResnetConfig",
7
+ "AutoModel": "modeling_resnet.ResnetModel"
8
  },
9
  "avg_down": true,
10
  "base_width": 64,