- 目标:在mbart预训练模型上微调RO-EN语言对
- 环境安装:
大坑:fairseq官网一定不要切换到main分支clone(如果你的cuda版本够新的话可以)
我的cuda:11.2,如果是10多看看自己cuda集合,可以切换到比较新的cuda版本--版本切换
pytorch:1.10.0
然后 进行fairseq install
安装过程中会出现其他报错:搜索对应解决方案都能解决/ 比如numpy版本问题等
- 代码执行:
大坑:fairseq官网一定不要切换到main分支clone(如果你的cuda版本够新的话可以)
我的cuda:11.2,如果是10多看看自己cuda集合,可以切换到比较新的cuda版本--版本切换
pytorch:1.10.0
然后 进行fairseq install
安装过程中会出现其他报错:搜索对应解决方案都能解决/ 比如numpy版本问题等