sankaku_rating_1l_s0/meta.json

{
    "encoder_model": "wdtagger:SmilingWolf/wd-swinv2-tagger-v3",
    "flops": 525824.0,
    "init_params": {
        "in_dims": 1024,
        "layers": [
            0.5
        ],
        "out_dims": 3
    },
    "model_type": "mlp",
    "params": 526339.0,
    "problem": {
        "labels": [
            "r15",
            "r18",
            "safe"
        ],
        "type": "classification"
    },
    "train": {
        "batch_size": 16,
        "key_metric": "AUC",
        "learning_rate": 0.001,
        "loss": "focal",
        "max_epochs": 100,
        "processes": 8,
        "seed": 0,
        "weight_decay": 0.001
    }
}