This commit is contained in:
parent
3a45712277
commit
b817640e8c
|
|
@ -15,8 +15,6 @@
|
||||||
"reduce_bucket_size": 50000000,
|
"reduce_bucket_size": 50000000,
|
||||||
"stage3_prefetch_bucket_size": 50000000,
|
"stage3_prefetch_bucket_size": 50000000,
|
||||||
"stage3_param_persistence_threshold": 100000,
|
"stage3_param_persistence_threshold": 100000,
|
||||||
|
|
||||||
/* 关键:保存最终权重时做 16bit 聚合,方便直接 from_pretrained 加载 */
|
|
||||||
"stage3_gather_16bit_weights_on_model_save": true
|
"stage3_gather_16bit_weights_on_model_save": true
|
||||||
},
|
},
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue