{
    "lr": 0.001,
    "mlp_lr": 0.005,
    "wd": 5e-05,
    "mlp_wd": 0.0,
    "epochs": 80,
    "tr_batch_size": 1024,
    "hidden_channels": 256,
    "num_heads": 16,
    "num_layers": 2,
    "num_neighbors": [32],
    "disjoint_train_ratio": 0.3,
    "mlp_dropout": 0.1,
    "grad_clip_norm": 5.0,
    "eval_threshold": 0.5,
    "neg_sampling_ratio_l1": 6,
    "pos_weight_l1": 6,
    "neg_sampling_ratio_l2": 70,
    "pos_weight_l2": 70,
    "neg_sampling_ratio_l3": 100,
    "pos_weight_l3": 100,
    "neg_sampling_ratio_l4": 100,
    "pos_weight_l4": 100,
    "test_on_cpu": true,
    "checkpoint_interval": 20,
    "val_eval_num_proteins": 500
}