| { | |
| "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" | |
| } | |
| } | |