CLIP训练报错,疑似数据集加载问题
收藏回复举报
CLIP训练报错,疑似数据集加载问题
t('forum.solved') 已解决
发表于2024-09-03 10:57:14
0 查看

### CLIP训练参考链接

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

按照文档配置已拉取权重,安装transformers,但是训练报数据加载问题

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

cke_3280.png

数据集是完整的COCO2017,是跑过yolo训练的

cke_9755.png

我要发帖子