CLIP训练报错,疑似transformers版本问题
收藏回复举报
CLIP训练报错,疑似transformers版本问题
t('forum.solved') 已解决
发表于2024-09-02 21:08:50
0 查看

### CLIP训练参考链接

https://gitee.com/ascend/ModelZoo-PyTorch/blob/master/PyTorch/built-in/others/CLIP_for_PyTorch

由于权重拉下来后需要执行python3 save_clip_roberta.py 合并,这个过程是依赖transformers版本的,但文档并没注明调试使用哪个版本

问题:启动训练后报错

bash test/train_clip_full_1p.sh --data_path=/data1/model_from_hf/coco2017/ --model_path=./clip-roberta/ 

cke_227.png

我要发帖子