foldingdiff/hyperparam_jsons/small_model_params.json

{
    "timesteps": [
        500,
        1000
    ],
    "variance_schedule": [
        "linear",
        "cosine",
        "quadratic"
    ],
    "num_hidden_layers": [
        6,
        8,
        10
    ],
    "lr": [
        0.001,
        0.0001
    ],
    "batch_size": [
        64,
        128,
        256
    ]
}