Pytorch深度学习思考【精】
文章平均质量分 86
Hurri_cane
自动驾驶算法工程师
展开
-
Pytorch深度学习中分段函数计算加速Tips
Pytorch深度学习中计算加速Tips1.配置环境2.问题描述3.问题解决4.结束语1.配置环境使用环境:python3.7平台:Windows10工具:pytorchIDE:PyCharm2.问题描述博主在搭建网络过程中,最后一步设计loss时由于要进行分情况讨论,所以采用了for循环加if判断来实现不同情况loss的计算。这样导致模型训练速度大打折扣,为了解决这个问题,思考了一天终于找到解决方法,如果你也遇到同样的问题或许这篇文章可以对你有所启发将问题简化我们有一组数据data,根据原创 2021-11-17 15:47:45 · 1255 阅读 · 0 评论 -
基于卷积神经网络的手写数字识别(附数据集+完整代码+操作说明)
基于卷积神经网络的手写数字识别(附数据集+代码)配置环境1.前言2.问题描述3.解决方案4.实现步骤4.1数据集选择4.2构建网络4.3训练网络4.4测试网络4.5图像预处理4.6传入网络进行计算5.代码实现5.1文件说明5.2使用方法5.3 训练模型5.4配置环境使用环境:python3.8平台:Windows10IDE:PyCharm1.前言手写数字识别,作为机器视觉入门项目,无论是基于传统的OpenCV方法还是基于目前火热的深度学习、神经网络的方法都有这不错的训练效果。当然,这个项目也常常原创 2020-12-06 17:11:52 · 77274 阅读 · 166 评论 -
《动手学深度学习》(PyTorch版)代码注释 - 32 【RNN_with_zero】
目录说明配置环境此节说明代码修改之后的代码说明本博客代码来自开源项目:《动手学深度学习》(PyTorch版)并且在博主学习的理解上对代码进行了大量注释,方便理解各个函数的原理和用途配置环境使用环境:python3.8平台:Windows10IDE:PyCharm此节说明此节对应书本上5.7节此节功能为:使用重复元素的网络(VGG)由于次节相对容易理解,代码注释量较少并且,值得注意的是,次节中博主对其中数据操作有些困惑,自己重新实现了一版,在文末附上代码# 本书链接https://原创 2020-08-28 19:59:16 · 615 阅读 · 0 评论 -
深度学习中训练迭代次数理解【源码阅读技巧分享】【深度学习循环迭代理解】【for X, y in train_iter:】
这里写目录标题起因问题探索第一步:由近及远三级目录本文源码起因近日,博主在学习《动手学深度学习》(PyTorch版)时,用fashion_mnist复现LeNet时想知道这个for循环运行了多少次:代码如下:(在文末会给出整个代码) for X, y in train_iter: X = X.to(device) y = y.to(device) y_hat = net(X) l = los原创 2020-08-23 13:04:23 · 8884 阅读 · 5 评论