File size: 1,366 Bytes
0d366b5 dc08063 0d366b5 5556236 dc08063 5556236 0d366b5 87ffc3d dc08063 87ffc3d dc08063 0e7e341 87ffc3d 0d366b5 |
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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
{
"architectures": [
"EfficientNetB0"
],
"model_type": "EfficientNetB0",
"num_labels": 7,
"hidden_size": 300,
"num_attention_heads": 1,
"num_hidden_layers": 2,
"intermediate_size": 1200,
"hidden_act": "relu",
"hidden_dropout_prob": 0.2,
"attention_probs_dropout_prob": 0.2,
"initializer_range": 0.02,
"layer_norm_eps": 1e-12,
"image_size": 128,
"num_channels": 3,
"rescale": 0.00392156863,
"shear_range": 0.2,
"zoom_range": 0.2,
"horizontal_flip": true,
"batch_size": 64,
"learning_rate": 0.001,
"loss": "categorical_crossentropy",
"metrics": ["accuracy"],
"callbacks": [
{
"type": "EarlyStopping",
"params": {
"monitor": "val_loss",
"mode": "min",
"patience": 5,
"verbose": 1
}
},
{
"type": "ModelCheckpoint",
"params": {
"filepath": "./best_model_EB0_7_classification.h5",
"monitor": "val_loss",
"mode": "min",
"verbose": 1,
"save_best_only": true
}
}
],
"label2id": {
"celeba": 0,
"celeba_hq": 1,
"deepfakes": 2,
"face2face": 3,
"faceswap": 4,
"neuraltextures": 5,
"youtube": 6
},
"id2label": {
"0": "celeba",
"1": "celeba_hq",
"2": "deepfakes",
"3": "face2face",
"4": "faceswap",
"5": "neuraltextures",
"6": "youtube"
}
}
|