sreenathsree1578 commited on
Commit
a3b2747
·
verified ·
1 Parent(s): 4afd59d

Upload 3 files

Browse files
best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbf01c14458a7397954b6d450da80e3d6357e38abc91ac38fec8365d4c724ed1
3
+ size 16341369
model_config(1).json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backbone": "efficientnet_b0",
3
+ "num_classes": 2,
4
+ "dropout": 0.3,
5
+ "lr": 0.0001,
6
+ "epochs": 20,
7
+ "face_threshold": 0.5
8
+ }
preprocessor_config(2).json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "size": [
3
+ 224,
4
+ 224
5
+ ],
6
+ "normalize_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "normalize_std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "batch_size": 32,
17
+ "num_workers": 2
18
+ }