diff --git a/training/nvidia/cpm-pytorch/README.md b/training/nvidia/cpm-pytorch/README.md index 97c860dcc..b6dcafa26 100644 --- a/training/nvidia/cpm-pytorch/README.md +++ b/training/nvidia/cpm-pytorch/README.md @@ -42,7 +42,9 @@ * 性能指标 -| 配置 | precision | fix_hp | e2e_time | p_whole | p_train | p_core | acc | mem | -| ------------------ | --------- | ------------- | -------- | ------- | ------- | ------ | ----- | --------- | -| A100单机8卡(1x8) | fp16 | / | 1641 | 587 | 835 | 1059 | 0.92 | 12.9/40.0 | -| A100单机8卡(1x8) | fp16 | bs=256,lr=0.8 | 5469 | 771 | 1090 | 1292 | 0.918 | 23.1/40.0 | \ No newline at end of file +| 配置 | precision | fix_hp | e2e_time | p_whole | p_train | p_core | acc | mem | +| ------------------- | --------- | ---------------- | -------- | ------- | ------- | ------ | ----- | --------- | +| A100单机8卡(1x8) | fp16 | / | 1641 | 587 | 835 | 1059 | 0.92 | 12.9/40.0 | +| A100单机8卡(1x8) | fp16 | bs=128,lr=0.002 | 5469 | 771 | 1090 | 1292 | 0.918 | 23.1/40.0 | +| A100单机单卡(1x1) | fp16 | bs=192,lr=0.0005 | | 78.4 | 111.9 | 127.2 | | 34.8/40.0 | +| A100两机8卡(2x8) | fp16 | bs=192,lr=0.0005 | | 1583 | 2221 | 2583.8 | | 29.9/40.0 | \ No newline at end of file