afscomercial commited on
Commit
4ca50f1
1 Parent(s): 6ff9ad4

add config

Browse files
best_resnet50_model.pth → best_resnet50_model.bin RENAMED
File without changes
config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "resnet",
3
+ "architectures": ["ResNet"],
4
+ "num_labels": 4,
5
+ "image_size": 224,
6
+ "input_size": [3, 224, 224]
7
+ }