beichenzbc / Long-CLIP

[ECCV 2024] official code for "Long-CLIP: Unlocking the Long-Text Capability of CLIP"
Apache License 2.0
620 stars 30 forks source link

微调LongClip #34

Closed nimtecv closed 3 months ago

nimtecv commented 3 months ago

尊敬的研究人员: 我想用自己的数据集微调您的LongClip,但是发现有点无从下手,真诚的希望您能给与一点点帮助。 万分感谢

beichenzbc commented 3 months ago

您好,请问您能描述一下您碰到的问题吗

nimtecv commented 3 months ago

image image 我zai在train中将模型加载为预训练好的模型,用我的数据集训练,torch.save(self.model.state_dict(), 'E:/XM/Long-CLIP-main/checkpoints/' + name)通过这个进行保存后,我想将他用到别的地方,但是一直无法加载

beichenzbc commented 3 months ago

有什么报错呢

nimtecv commented 3 months ago

你好 问题解决啦 谢谢