一位浪漫的极客

权重初始化： emb那种可以直接from_pretrained 固定权重 利用requires_grad=False，这样只是不更新，但是还是会计算并占用显存 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d...

why pytorch？ 计算图分为 动态图(pytorch)&静态图(tensorflow)，可用利用python建图，而不需用如 tf专门的建图语言；动态图修改更灵活，一般而言也省内存 配置 查看 GPU是否可用：torch.cuda.is_available()；数量： torch.cuda.device_count() 基本操作 数据类型的定义与运算（注意 max/sum、i...