jays009 commited on
Commit
7d3c2b5
·
verified ·
1 Parent(s): 9bd1dc9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -1
config.json CHANGED
@@ -1 +1,7 @@
1
- {"architectures": ["ResNet"], "num_labels": 2, "input_size": 224, "image_mean": [0.485, 0.456, 0.406], "image_std": [0.229, 0.224, 0.225]}
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["ResNet50"],
3
+ "model_type": "resnet",
4
+ "input_ids": "image",
5
+ "task": "image-classification",
6
+ "framework": "pytorch"
7
+ }