File size: 307 Bytes
04ee3bd 7144fce 04ee3bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | {
"model": "mlp",
"feature": "bert",
"in_dim": 768,
"hidden": 128,
"layers": 2,
"heads": 4,
"dropout": 0.5,
"edge_drop": 0.2,
"admm_p": 2.5,
"admm_lambda": 1.0,
"admm_iterations": 2,
"feature_l1": 0.001,
"epochs": 200,
"batch_size": 128,
"lr": 0.001,
"weight_decay": 0.0001
} |