01 2021 档案
摘要:DataLoader?直接切片更香!(pytorth GPU加速探讨) 写文章时作者刚看了两天pytorch,对其底层原理不甚了解。 这不是一篇技术文章,请谨慎参考 仅从实际效果来讲一下遇到的情况 使用pytorch的小伙伴一定不会陌生这样的代码: train_loader = DataLoader
阅读全文
摘要:关于CIFAR-10数据集向量到三维图片中的转置操作 在CIFAR-10数据集中,图片的储存方式为一维向量,如下图所示 错误方法: a = x_data[3,:] # np.ndarray shape:(3072,) a = a.reshape(32,32,3) plt.imshow(a) 正确方法
阅读全文
摘要:#%% import torch as th import numpy as np import matplotlib.pyplot as plt from torch.utils.data import Dataset from torch.utils.data import DataLoader
阅读全文

浙公网安备 33010602011771号