随笔分类 -  Pytorch 使用知识点

摘要：首先、在pytorch中本身带有参数初始化。使用初始化主要是因为在模型构建中的需求 1.使用apply() from torch import nn def weights_init(mod): """对于卷积层和batchnorm进行初始化""" classname=mod.__class__._ 阅读全文

摘要：import torch import torch.nn as nn import torch.nn.functional as F Pytorch基础构建模型方法 方法一：较为通用的方法 class Net(nn.Module): def __init__(self): self.conv = n 阅读全文