dataloader中需要设置参数drop_last=True,因为你样本的总数不是批次的倍数,有余数存在,所以最后一个批次会出现错误,设置丢弃掉最后一个次数不足的批次就可以了,或者你reshape平均分布一下
报错问题应该是卷积的问题,卷积的两个输入尺寸不一致导致的,input是6416 ,weight是102464的,没法相乘报错。
import torch.nn.functional as F
由于问题中没有给出报错提示和具体代码,无法确定问题根源在哪里,也无法给出具体的解决方案。建议在提问时给出完整的信息,以便更好地得到帮助。