![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 53
粟悟饭&龟波功
这个作者很懒,什么都没留下…
展开
-
Pytorch中DistributedDataParallel基本使用
使用DistributedDataParallel进行并行训练原创 2023-01-03 16:17:38 · 1339 阅读 · 1 评论 -
timm库使用
一、timm库简介PyTorch Image Models,简称timm,是一个巨大的PyTorch代码集合,整合了常用的models、layers、utilities、optimizers、schedulers、data-loaders/augmentations和reference training/validation scripts。二、安装pip install timm三、使用...原创 2022-04-08 19:50:59 · 13399 阅读 · 4 评论 -
pytorch特征可视化——特征图
原创 2022-03-25 15:15:46 · 4753 阅读 · 7 评论 -
深度可分离卷积vs标准卷积
本文只比较深度可分离卷积与标准卷积的参数量、运算量、时间,模型性能请看Mobilenet系列论文。1. Model深度可分离卷积import torchimport torch.nn as nnimport torch.nn.functional as Fimport timefrom torchstat import statclass Depthpoint_conv(nn.Module): def __init__(self, C_in, C_out, kernel_si原创 2021-12-20 10:27:29 · 1331 阅读 · 0 评论 -
Pytorch查看模型的参数量和计算量
1. stat缺点:仅支持单输入模型Exampleimport torch.nn as nnimport torch.nn.functional as Ffrom torchstat import statclass Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(1, 10, kernel_size=5)原创 2021-12-19 22:06:29 · 2171 阅读 · 2 评论