Pytorch
小白的深度学习之路。。。
boss-dog
当才华撑不起野心的时候,唯有安静读书......
展开
-
用Pytorch搭建LeNet-5模型且基于CIFAR-10数据集训练
博客中的实例是Pytorch官方的一个demo,这里我结合最近正在看的书《Python深度学习 基于Pytorch》P125 Pytorch实现CIFAR-10多分类这节,进行了梳理与个人的总结,以加深自己的理解。目的:①了解LeNet-5模型②加深Pytorch的学习③从头至尾介绍搭建网络并进行训练预测的全过程LeNet-5模型1.模型架构LeNet-5模型结构为输入层–卷积层–池化层–卷积层–池化层–全连接层–全连接层–输出,为串联模式,如上图所示2.模型特点a.每个卷积层包含3个部原创 2020-06-18 22:37:38 · 3345 阅读 · 3 评论 -
《Python深度学习 基于PyTorch》(吴茂贵)P73源码分析记录
一.TensorboardX介绍书上这一节主要是介绍Pytorch可视化工具——tensorboardX。首先简单介绍下该工具,tensorboardX可以记录训练数据、评估数据、网络结构、图像等。模块安装先安装tensorflow,然后再安装tensorboardX,目前tensorboardX版本仅支持到了2.0这是我模块版本,可以参考,运行没问题使用方法from tensorboardX import SummaryWriter#实例化SummaryWriter,并指明日志存放路径。原创 2020-06-08 16:17:27 · 1857 阅读 · 3 评论