weixin_44776845的博客

在训练期间，运行如下代码import torchimport torch.nn as nnfrom lenet import LeNet, MyNetfrom torchsummary import summarynet = LeNet(classes=2)net.initialize_weights()summary(net, input_size=(3, 32, 32))报...

猫狗大战的时候处理猫狗数据时出点小问题，在此记录先贴出数据分离的部分代码 dataset_dir = os.path.join("..", "data", "Dog_Cat_data") split_dir = os.path.join("..", "data", "cat_dog_split") train_dir = os.path.join(split_dir, "...

RMB分类模型实现数据集不一定必须是RMB数据集，其他数据集同理，注意读取路径的设置数据读取数据收集 ------ Img， label数据划分 ------ train, valid, test数据读取 ------ DataLoader ------Sampler 生成索引，即样本的序号 DataSet 根据索引读取图片和标签数据预处理 ----- transformsD...

逻辑回归模型的实现import torchimport torch.nn as nnimport matplotlib.pyplot as pltimport numpy as nptorch.manual_seed(10)# ============================= step 1/5 生成数据 ========================sample_nu...

torch.autograd.backward ------ 自动求取梯度torch.autograd.backward(tensors, grad_tensors=None, retain_graph=None, create_graph=False)tensors: 用于求导梯度, 如loss；retain_graph: 保存计算图；create_graph: 创建导数计算图， 用于高...

计算图：用来描述运算的有向无环图有两个主要元素，结点note 边edge结点表示数据，如向量，矩阵，张量边表示运算，如加减乘除使用计算图主要是为了求导方便， 只需要沿着计算图的方向找到需要求导对象的所有路径， 对路径上的所有导数求和即该求导对象的梯度。叶子结点：用户创建的结点，区分叶子结点的目的是节省内存，代码运行之后会释放点非叶子结点的数据；非叶子结点梯度为none是因为在反向传播之后非...

在运行之前需要matplotlib包，快速方便的方法可以看这个pycharm导入matplotlib可以尝试设置不同的学习率，创建不同大小的训练数据集，以及不同的斜率和截距import torchimport matplotlib.pyplot as plttorch.manual_seed(10)# 设定学习率为0.01lr = 0.01best_loss = float("in...

笔记为以下函数的实例**torch.manual_seed **-------生成随机数种子，并且使得每次得到的随机数是固定的张量的拼接，切分，索引，变换，压缩：torch.cattorch.stacktorch.chunkenumeratetorch.splittorch.index_selecttorch.masked_selecttorch.reshapetorch.t...

开始学习pytorch, 记录学习历程本博客为关于以下函数的实例np.onestorch.tensortorch.from_numpytorch.zerostorch.zeros_liketorch.fulltorch.arangetorch.linspacetorch.logspacetorch.eyetorch.randpermtorch.randinttorch....