foldingdiff/hyperparam_jsons/bert_angle_diffusion_configs.json

{
    "timesteps": [
        250,
        500,
        750,
        1000
    ],
    "variance_schedule": [
        "linear",
        "cosine",
        "quadratic"
    ],
    "batch_size": [
        128,
        64
    ],
    "lr": [
        1e-3,
        1e-4
    ],
    "gradient_clip": [
        0.0,
        0.5
    ]
}