LMDrive icon indicating copy to clipboard operation
LMDrive copied to clipboard
verified publisher icon opendilab

新采集数据集微调视觉编码器

Open CoderXuans opened this issue 11 months ago • 2 comments

@deepcs233 你好，我看到视觉编码器的train.sh脚本中有 --pretrained \ 一行，我理解的是它能够用你提供训练好的模型权重，在新数据集上进行微调，这样理解对吗？

CoderXuans avatar Jan 07 '25 08:01 CoderXuans

我找到train_pretrain.py文件，正在看它的逻辑

CoderXuans avatar Jan 07 '25 09:01 CoderXuans

@deepcs233 你好，运行cd vision_encoder bash scripts/train.sh我理解的是默认会使用预训练模型，会从lmdriver_config.py中读取内容，如果没有就会从头进行训练，而如果有的话就会使用预训练模型微调，我这样理解对吗？

CoderXuans avatar Jan 07 '25 09:01 CoderXuans