| { | |
| "model_type": "BaseCNN", | |
| "model_config": { | |
| "layers": [ | |
| 32, | |
| 64, | |
| 128, | |
| 256 | |
| ], | |
| "kernel": 3, | |
| "padding": 1, | |
| "stride": 1, | |
| "batchnorm": true, | |
| "bn_pre_activ": true, | |
| "activation": "ReLU", | |
| "dropout": 0.4, | |
| "pool": true, | |
| "fc_layers": [ | |
| 256, | |
| 128 | |
| ], | |
| "nr_classes": 3, | |
| "in_channels": 3 | |
| }, | |
| "num_labels": 3, | |
| "id2label": { | |
| "0": "Type 1", | |
| "1": "Type 2", | |
| "2": "Type 3" | |
| }, | |
| "label2id": { | |
| "Type 1": 0, | |
| "Type 2": 1, | |
| "Type 3": 2 | |
| } | |
| } |