![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
我~
!
展开
-
pytorch 模型可视化
一. visualize.pyfrom graphviz import Digraphimport torchfrom torch.autograd import Variabledef make_dot(var, params=None): """ Produces Graphviz representation of PyTorch autograd graph ...原创 2018-07-26 15:46:46 · 5950 阅读 · 4 评论 -
深度学习——卷积神经网络 的经典网络(LeNet-5、AlexNet、ZFNet、VGG-16、GoogLeNet、ResNet)(转)
深度学习——卷积神经网络 的经典网络(LeNet-5、AlexNet、ZFNet、VGG-16、GoogLeNet、ResNet) 一、CNN卷积神经网络的经典网络综述下面图片参照博客:http://blog.csdn.net/cyh_24/article/details/...转载 2018-07-31 21:39:03 · 2072 阅读 · 0 评论 -
pytorch gpu设置
1. 显示当前GPU使用情况nvidia-smi2. 指定gpu使用2.1 终端指定CUDA_VISIBLE_DEVICES=2 python xxx.py2.2 代码中指定import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"原创 2018-07-28 09:42:51 · 15092 阅读 · 0 评论 -
pytorch 准备自己数据库
构建图片数据库一般有两种方法:1、所有图片放在一个文件夹内,另外有一个txt文件显示标签。2、不同类别的图片放在不同的文件夹内,文件夹就是图片的类别。方法2会用到 ImageFolderImageFolder 是一个通用的数据加载器,数据集中的数据以以下方式组织root/dog/xxx.pngroot/dog/xxy.pngroot/dog/xxz.pngroo...原创 2018-08-03 11:03:35 · 1181 阅读 · 0 评论