File size: 635 Bytes

de8a1e2

{
    "encoder_model": "wdtagger:SmilingWolf/wd-swinv2-tagger-v3",
    "flops": 525312.0,
    "init_params": {
        "in_dims": 1024,
        "layers": [
            0.5
        ],
        "out_dims": 2
    },
    "model_type": "mlp",
    "params": 525826.0,
    "problem": {
        "labels": [
            "monochrome",
            "normal"
        ],
        "type": "classification"
    },
    "train": {
        "batch_size": 16,
        "key_metric": "AUC",
        "learning_rate": 0.001,
        "loss": "focal",
        "max_epochs": 100,
        "processes": 8,
        "seed": 114514,
        "weight_decay": 0.001
    }
}