初始化项目,由ModelHub XC社区提供模型

Model: openbmb/BitCPM-CANN-1B-unquantized
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-04 14:32:59 +08:00
commit a94919dde4
28 changed files with 181544 additions and 0 deletions

29
example/ds_config.json Normal file
View File

@@ -0,0 +1,29 @@
{
"bf16": {
"enabled": true
},
"zero_optimization": {
"stage": 3,
"offload_optimizer": {
"device": "cpu",
"pin_memory": true
},
"offload_param": {
"device": "none"
},
"overlap_comm": true,
"contiguous_gradients": true,
"sub_group_size": 1e9,
"reduce_bucket_size": 2e8,
"stage3_prefetch_bucket_size": 2e8,
"stage3_param_persistence_threshold": 1e5,
"stage3_max_live_parameters": 2e9,
"stage3_max_reuse_distance": 2e9,
"stage3_gather_16bit_weights_on_model_save": true
},
"gradient_accumulation_steps": "auto",
"gradient_clipping": "auto",
"train_batch_size": "auto",
"train_micro_batch_size_per_gpu": "auto",
"wall_clock_breakdown": false
}