mzou1223 commited on
Commit
6a6c80b
·
verified ·
1 Parent(s): b1b3f7b

Add Model configuration

Browse files
Files changed (1) hide show
  1. model_config.json +22 -0
model_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "efficientnet_b0",
3
+ "num_classes": 3,
4
+ "class_names": [
5
+ "bad",
6
+ "borderline",
7
+ "good"
8
+ ],
9
+ "img_size": 224,
10
+ "test_accuracy": 60.97560975609756,
11
+ "best_val_accuracy": 65.0,
12
+ "mean": [
13
+ 0.485,
14
+ 0.456,
15
+ 0.406
16
+ ],
17
+ "std": [
18
+ 0.229,
19
+ 0.224,
20
+ 0.225
21
+ ]
22
+ }