mirror of
https://github.com/microsoft/foldingdiff.git
synced 2026-06-04 13:30:33 +08:00
29 lines
742 B
JSON
29 lines
742 B
JSON
{
|
|
"shift_angles_zero_twopi": false,
|
|
"noise_prior": "gaussian",
|
|
"timesteps": 250,
|
|
"variance_schedule": "linear",
|
|
"variance_scale": 1.0,
|
|
"time_encoding": "gaussian_fourier",
|
|
"implementation": "huggingface_encoder",
|
|
"position_embedding_type": "absolute",
|
|
"num_hidden_layers": 12,
|
|
"hidden_size": 384,
|
|
"intermediate_size": 768,
|
|
"num_heads": 12,
|
|
"dropout_p": 0.1,
|
|
"decoder": "mlp",
|
|
"gradient_clip": 1.0,
|
|
"lr": 5e-5,
|
|
"loss": "radian_l1_smooth",
|
|
"l2_norm": 0.0,
|
|
"l1_norm": 0.0,
|
|
"circle_reg": 0.0,
|
|
"lr_scheduler": "",
|
|
"min_epochs": 10000,
|
|
"max_epochs": 10000,
|
|
"early_stop_patience": 0,
|
|
"use_swa": false,
|
|
"batch_size": 64,
|
|
"multithread": true
|
|
} |