lebiraja commited on
Commit
7556457
·
verified ·
1 Parent(s): b489cd0

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +29 -0
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "efficientnet",
3
+ "architectures": ["EfficientNetB4"],
4
+ "num_classes": 45,
5
+ "image_size": 384,
6
+ "num_channels": 3,
7
+ "task": "multi-label-image-classification",
8
+ "framework": "pytorch",
9
+ "training": {
10
+ "optimizer": "AdamW",
11
+ "scheduler": "CosineAnnealingLR",
12
+ "loss_function": "BCEWithLogitsLoss",
13
+ "batch_size": 5,
14
+ "epochs": 50,
15
+ "learning_rate": 1e-4,
16
+ "weight_decay": 1e-2
17
+ },
18
+ "performance": {
19
+ "best_auc": 0.8204,
20
+ "train_loss": 0.2118,
21
+ "val_loss": 0.2578,
22
+ "macro_f1": 0.1517,
23
+ "micro_f1": 0.4450
24
+ },
25
+ "normalization": {
26
+ "mean": [0.485, 0.456, 0.406],
27
+ "std": [0.229, 0.224, 0.225]
28
+ }
29
+ }