Pytorch常用包

  • torch:张量的有关运算。如创建、索引、连接、转置、加减乘除、切片等
  • torch.nn: 包含搭建神经网络层的模块(Modules)和一系列loss函数。如全连接、卷积、BN批处理、dropout、CrossEntryLoss、MSELoss等
  • torch.nn.functional:常用的激活函数relu、leaky_relu、sigmoid等
  • torch.autograd:提供Tensor所有操作的自动求导方法
  • torch.optim:各种参数优化方法,例如SGD、AdaGrad、Adam、RMSProp等
  • torch.utils.data:用于加载数据
  • torch.nn.init:可以用它更改nn.Module的默认参数初始化方式

  • torchvision.datasets:常用数据集。MNIST、COCO、CIFAR10、Imagenet等
  • torchvision.modules:常用模型。AlexNet、VGG、ResNet、DenseNet等
  • torchvision.transforms:图片相关处理。裁剪、尺寸缩放、归一化等
    -torchvision.utils:将给定的Tensor保存成image文件
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
PyTorch中,有一些常用的代码片段可以帮助我们进行深度学习任务。以下是一些常见的代码片段: 1. 定义网络模型: ```python import torch import torch.nn as nn class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.fc = nn.Linear(10, 5) def forward(self, x): x = self.fc(x) return x net = Net() ``` 在这个示例中,我们定义了一个简单的神经网络模型Net,它含一个全连接层nn.Linear。我们可以使用这个模型来进行前向传播。 2. 数据加载: ```python import torch from torch.utils.data import DataLoader, Dataset class CustomDataset(Dataset): def __init__(self, data): self.data = data def __getitem__(self, index): x, y = self.data[index] return torch.tensor(x), torch.tensor(y) def __len__(self): return len(self.data) dataset = CustomDataset(data) dataloader = DataLoader(dataset, batch_size=32, shuffle=True) ``` 在这个示例中,我们定义了一个自定义的数据集CustomDataset,并使用DataLoader将数据集加载为批量大小为32的数据加载器。 3. 模型训练: ```python import torch import torch.optim as optim import torch.nn as nn criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9) for epoch in range(num_epochs): running_loss = 0.0 for inputs, labels in dataloader: optimizer.zero_grad() outputs = net(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print(f"Epoch {epoch+1}: Loss = {running_loss/len(dataloader)}") ``` 在这个示例中,我们使用交叉熵损失函数nn.CrossEntropyLoss和随机梯度下降优化器optim.SGD来训练我们的模型。我们使用for循环迭代数据加载器中的批次,并在每个批次上进行前向传播、计算损失、反向传播和优化模型参数。 4. 模型保存和加载: ```python torch.save(net.state_dict(), 'model.pth') # 加载模型 model = Net() model.load_state_dict(torch.load('model.pth')) ``` 在这个示例中,我们使用torch.save将模型的状态字典保存到文件'model.pth'中,并使用model.load_state_dict加载模型的参数。 这些是PyTorch中一些常见的代码片段,可以帮助我们进行深度学习任务。需要根据具体任务的需求进行适当的修改和调整。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值