如何用自己的灰度图数据集进行训练？🥲

[NCUCyy]

我的数据集是（1，256，256）的灰度图，在训练DACLIP的时候会报如下的错误：🥲🥲🥲

我把如下路径中的conv1中的in_channel从3改为了1

结果会报错：
RuntimeError: Error(s) in loading state_dict for CLIP: size mismatch for visual.conv1.weight: copying a param with shape torch.Size([768, 3, 32, 32]) from checkpoint, the shape in current model is torch.Size([768, 1, 32, 32]).

请问如何解决呀？

[Algolzw]

你好我建议不要修改网络哈最好可以在加载图像时把gray转化为RGB（当然看起来还是灰度图），比如在这里加一个.convert('RGB')

[gejutaixiao22]

您好，我也在用daclip训练灰度图，能交流一下具体的细节么。非常感谢