报错:
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [512, 512, 3, 3], but got 2-dimensional input of size [4, 512] instead
conv2d卷积层输入大小不匹配
要求输入 [512, 512, 3, 3]
实际输入 [4, 512]
临时解法:把2d tensor expand成4d
someTensor = someTensor.unsqueeze(2)