Dragonscypher commited on
Commit
22529ad
·
verified ·
1 Parent(s): 6e24260

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -2
config.json CHANGED
@@ -1,4 +1,9 @@
1
  {
2
- "architecture": "EfficientNet_B0",
3
- "framework": "PyTorch"
 
 
 
 
 
4
  }
 
1
  {
2
+ "architectures": ["EfficientNetForImageClassification"],
3
+ "model_type": "efficientnet",
4
+ "depth_coefficient": 1.0,
5
+ "width_coefficient": 1.0,
6
+ "dropout_rate": 0.2,
7
+ "num_classes": 1000,
8
+ "image_size": 224
9
  }