随笔分类 - pytorch
数据加载报错
摘要:Pytorch(一) 一、Pytorch介绍 Pytorch 是Torch在Python上的衍生物 和Tensorflow相比: Pytorch建立的神经网络是动态的,而Tensorflow建立的神经网络是静态的 Tensorflow的高度工业化,它的底层代码很难看懂 官网:http://pytor
阅读全文
摘要:破天荒的谎言、谈敷衍 尽人事以听天命! Anaconda3详细安装使用教程及问题总结 Anaconda3详细安装使用教程及问题总结 注:本面文章来源于网络,转载请注明来自https://www.jianshu.com/p/026a2c43b081。 1.Anaconda是什么? 简单来说,Anaco
阅读全文
摘要:import torchprint(torch.__version__)1.8.1+cpu
阅读全文
摘要:pytorch中模型的保存pth,pkl,pt格式 pytorch保存模型非常简单,主要有两种方法: 只保存参数;(官方推荐) 保存整个模型 (结构+参数)。 由于保存整个模型将耗费大量的存储,故官方推荐只保存参数,然后在建好模型的基础上加载。本文介绍两种方法,但只就第一种方法进行举例详解。 模型保
阅读全文
摘要:只要你创建了一个模块(一个.py文件),这个模块就有一个内置属性name生成,该模块的 name 的值取决于如何应用这个模块。说人话就是,如果你直接运行该模块,那么__name__ == "__main__";如果你 import 一个模块,那么模块name 的值通常为模块文件名。 如,创建一个te
阅读全文
摘要:import torchimport torch.nn as nnimport torch.nn.functional as Fclass VGGbase(nn.Module): # 定义网络模型 def __int__(self): super(VGGbase, self).__init__()
阅读全文
摘要:#自定义加载cifar10数据集from torchvision import transformsfrom torch.utils.data import DataLoader, Datasetimport osfrom PIL import Imageimport numpy as npimpo
阅读全文
摘要:UserWarning: The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors. This means you can write to the underlying (s
阅读全文