默认的模型并行参数为2,如果需要修改,可以使用change_mp.py,并调整generate_text.sh中的MPSIZE。change_mp.py的使用示例如下: python change_mp.py /path/to/CPM MPSIZE 这里的/path/to/CPM为模型路径,MPSIZE为一个整数,可以为1或者2的倍数,结果会生成一个新的模型,存储路径为/path/to/CPM_MPSIZE。
data_utils fp16 model mpu scripts .gitignore LICENSE README.md arguments.py change_mp.py configure_data.py example.txt generate_samples.py requirements.txt utils.py zero-shot-cls.pyBreadcrumbs CPM-1-Generate / example.txt Latest commit
mpu scripts .gitignore LICENSE README.md arguments.py change_mp.py configure_data.py example.txt generate_samples.py requirements.txt utils.py zero-shot-cls.py Breadcrumbs CPM-1-Generate / Latest commit zzy14 first commit Nov 13, 2020 ...
默认的模型并行参数为2,如果需要修改,可以使用change_mp.py,并调整generate_text.sh中的MPSIZE。change_mp.py的使用示例如下: python change_mp.py /path/to/CPM MPSIZE 这里的/path/to/CPM为模型路径,MPSIZE为一个整数,可以为1或者2的倍数,结果会生成一个新的模型,存储路径为/path/to/CPM_MPSIZE。