pytorch
wtj318_
做个有理想,有追求的好青年
展开
-
五 . Pytorch数据加载 : Dataset +Dataloader
1. Dataset 提供一种方式去获取数据及其label. 如何获取每一个数据及其label 告诉我们总共有多少数据,才能知道迭代多少次 from torch.utils.data import Dataset help(Dataset) Dataset?? def __getitem__(self, index) -> T_co: raise NotImplementedError def __add__(self, other: 'Dataset[T..原创 2022-03-10 14:23:13 · 1758 阅读 · 0 评论 -
四. torchvision 数据集
1. 基本功能函数使用讲解 import torchvision # root 数据集位置,train 训练集或测试集,download 是否下载 train_set = torchvision.datasets.CIFAR10( root ="./dataset",train=True,download=True) test_set = torchvision.datasets.CIFAR10(root="./dataset",train=False,download=True) print(t.原创 2022-03-10 11:16:36 · 1471 阅读 · 0 评论 -
二. Pytorch Tensorboard 使用
1. writer.add_scalar() from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter("logs") # 两个方法 # writer.add_image() for i in range(100): writer.add_scalar("y=x", i ,2*i) #(title,value,step) writer.close() 命令行输入: tensorboard原创 2022-03-08 17:48:30 · 163 阅读 · 0 评论 -
一. Pytorch 工具箱
import torch dir(torch) dir(torch.cuda) help(torch.cuda.is_available) # print(torch.cuda.is_available) 1. dir(torch) ['Any', 'BFloat16Storage', 'BFloat16Tensor', 'BoolStorage', 'BoolTensor', 'ByteStorage', 'ByteTensor', 'CharStorage', 'CharTens.原创 2022-03-08 10:10:10 · 1626 阅读 · 0 评论 -
一、张量-pytorch
import numpy as np import torch # data->tensor data = [[1, 2], [3, 4]] x_data = torch.tensor(data) print(x_data) # np->tensor np_array = np.array(data) x_np = torch.from_numpy(np_array) print(x_np) # tensor->tensor x_ones = torch.ones_like(x_.原创 2022-01-20 00:40:59 · 1074 阅读 · 0 评论