【pytorch报错】RuntimeError: mat1 dim 1 must match mat2 dim 0

错误描述: 网络输入参数不匹配,即前一层网络的输出channel与后一层网络的输入channel不一致。
解决方法: 将fc1的(2048,1024)改为(512,256)即可。
【pytorch报错】RuntimeError: mat1 dim 1 must match mat2 dim 0_第1张图片
debug方法:
根据打印输出确定channel数目
【pytorch报错】RuntimeError: mat1 dim 1 must match mat2 dim 0_第2张图片

你可能感兴趣的:(pytorch报错,python)