问题如下:RuntimeError: size mismatch, m1: [4 x 512], m2: [64 x 128]……
RuntimeError: size mismatch, m1: [4 x 512], m2: [64 x 128] at /opt/conda/conda-bld/pytorch_1573049304260/work/aten/src/THC/generic/THCTensorMathBlas.cu:290
RuntimeError: size mismatch, m1: [4 x 512], m2: [256 x 128] at /opt/conda/conda-bld/pytorch_1573049304260/work/aten/src/THC/generic/THCTensorMathBlas.cu:290
1. 查看网络参数:利用Pytorch搭建模型的参数,直接看程序
import torch
# 引入torch.nn并指定别名
import torch.nn as nn
import torch.nn.functional as F
class Net(nn.Module):
def __init__(self):
# nn.Module子类的函数必须在构造函数中执行父类的构造函数
super(Net, self).__init__()