MKChens

使用BMTrain，在 cuda12上微调模型会报错 Traceback (most recent call last): File "/home/worker/chenmingkun/github/CPM_Bee/src/finetune_cpm_bee.py", line 427, in main() File "/home/worker/chenmingkun/github/CPM_Bee/src/finetune_cpm_bee.py", line 422, in main tokenizer, model, optimizer, lr_scheduler, optim_manager = setup_model_and_optimizer(args) File "/home/worker/chenmingkun/github/CPM_Bee/src/finetune_cpm_bee.py", line 73,...

问题解决方法： 1、将pytorch包内的 site-packages/torch/version.py 内的version 修改为12.1 2、将pytorch包内的 site-packages/torch/nn/modules/module.py 第2112行 中的「, remove_duplicate=remove_duplicate」删除

试了几次也遇到了这个问题