mirror of
https://github.com/aqlaboratory/openfold.git
synced 2026-06-04 12:44:26 +08:00
25 lines
424 B
JSON
25 lines
424 B
JSON
{
|
|
"fp16": {
|
|
"enabled": false,
|
|
"min_loss_scale": 1
|
|
},
|
|
"amp": {
|
|
"enabled": false,
|
|
"opt_level": "O2"
|
|
},
|
|
"bfloat16": {
|
|
"enabled": false
|
|
},
|
|
"zero_optimization": {
|
|
"stage": 2,
|
|
"cpu_offload": true,
|
|
"contiguous_gradients": true
|
|
},
|
|
"activation_checkpointing": {
|
|
"partition_activations": true,
|
|
"cpu_checkpointing": true,
|
|
"profile": false
|
|
},
|
|
"gradient_clipping": 0.1
|
|
}
|