- 博客(7)
- 收藏
- 关注
原创 lambda表达式,embed = lambda x, eo=embedder_obj : eo.embed(x)解读
Nerf-pytorch项目中的lambda函数表达式的解析。
2022-07-18 09:23:03
269
原创 pytorch中torch.stack()的功能
首先看一下官方文档对torch.stack()函数功能的描述: 将一系列tensor沿着dim维度方向串联起来,所有的tensor必须具备相同的大小,其中维度的大小必须是在0到需要被串联的tensor的维度大小之间,如被串联的tensor是一个二维平面,则dim不能大于2。简单来说:就是把多个2维张量串联成3维张量,把多个3维张量串联为4维张量,以此类推就是在增加新的维度进行堆叠。举个例子:将3个2维张量进行torch.stack()操作。首先准备3个2维的3行4列数据如下图: 数据形状如下
2022-07-14 08:43:13
1006
原创 一张图看懂Resnet50与Resnet101算法
直接上流程图,算法很清晰。仅包括卷积层和全连接层,不包括池化层,正好50层。相比于ResNet_50,ResNet_101就是在上图第3个大虚线框多了17个bottleneck,17*3+50=101,说白了就是将下图复制17个加入上图的第3个大虚线框内:画图不易,转载请注明出处! ...
2022-05-31 09:23:01
19966
5
原创 Ubuntu20.04安装NVIDIA驱动+cuda+cudnn+anaconda+pytorch+pycharm经验
配置环境是深度学习的第一道门槛,也是必须迈过的一道门槛,之前一直在windows平台上跑pytorch,但很多包无法安装,必须要使用Linux。在经历了很多坑,无数次重装系统,崩溃,坚持后终于在Ubuntu20.04上成功安装pytorch,现将经验分享,希望能对初学者有一定参考。一、安装Ubantu20.04系统 在清华源上下载系统安装包,找一个空U盘,用第三方软件制作系统启动盘。注意一定是空U盘,制作启动盘时会将U盘格式化。Index of /ubuntu-releases/2...
2022-04-18 12:01:00
4831
5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人