用自己数据训练时遇到问题 #3
Description
您好！根据您给的提示，我运用自己的数据微调预训练zero123模型时出现如下错误“RuntimeError: Given groups=1, weight of size [320, 8, 3, 3], expected input[8, 20, 32, 32] to have 8 channels, but got 20 channels instead”。我将配置文件yaml中的in_channels参数改为8时可以正常训练，但是运用训练后ckpt文件进行推理时就会出现如下错误“RuntimeError: Error(s) in loading state_dict for TeethDreamer:
size mismatch for model.diffusion_model.input_blocks.0.0.weight: copying a param with shape torch.Size([320, 8, 3, 3]) from checkpoint, the shape in current model is torch.Size([320, 20, 3, 3]).”，请问需要如何解决呢？期待您的回复，谢谢！