File size: 826 Bytes
a1f8545
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
{
    "architecture": {
        "base_model": "MobileNetV2",
        "input_shape": [
            224,
            224,
            3
        ]
    },
    "custom_layers": [
        {
            "type": "GlobalAveragePooling2D"
        },
        {
            "type": "Dense",
            "units": 128,
            "activation": "relu"
        },
        {
            "type": "Dropout",
            "rate": 0.5
        },
        {
            "type": "Dense",
            "units": 1,
            "activation": "sigmoid"
        }
    ],
    "training": {
        "optimizer": "Adam",
        "learning_rate": 0.001,
        "loss": "binary_crossentropy",
        "metrics": [
            "accuracy"
        ]
    },
    "fine_tuning": {
        "unfreeze_from_layer": 100,
        "fine_tune_learning_rate": 0.0001
    }
}