Pytorch
4篇文章 · 0订阅
  • ResNet50的Pytorch实现
    ```python import torch import torch.nn as nn from torchvision.models import resnet50 from torchvisio
    PyTorch
  • Pytorch-book之数据处理(一)
    1. 数据处理模块 __getitem__：返回一条数据，或一个样本。obj[index]等价于obj.__getitem__(index) __len__：返回样本的数量。len(obj)等价于obj.__len__() 如果要对图片进行多个操作，可通过Compose函数将这…
    深度学习
  • Pytorch-book之前馈神经网络(二)
    1. 前馈神经网络 requires_grad：该变量有两个值，True 或者 False，如果为True，则加入到反向传播图中参与计算。 grad：该属性存储着相关的梯度值。当requires_grad为False时，该属性为None。即使 requires_grad为Tru…
    深度学习
  • Pytorch-book之Tensor基础(一)
    1. Tensor Tensor是PyTorch中重要的数据结构，可认为是一个高维数组。它可以是一个数（标量）、一维数组（向量）、二维数组（矩阵）以及更高维的数组。Tensor和Numpy的ndarrays类似，但Tensor可以使用GPU进行加速。 t.abs()/sqrt/…
    深度学习