![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 70
饿不坏的企鹅
我想当警察
展开
-
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错
pytorch使用tensorboardx可视化网络模型,graph位置报错,TypeError: graph() got an unexpected keyword argument 'use_strict_trace'原创 2023-07-26 11:38:37 · 1135 阅读 · 2 评论 -
pytorch中使用MultiStepLR动态调整学习率遇到的问题(求助帖)
首先放上原博文:pytorch中的学习率调整函数首先看一下原博文中通过MultiStepLR实现的部分model = AlexNet(num_classes=2)optimizer = optim.SGD(params = model.parameters(), lr=0.05)#在指定的epoch值,如[10,15,25,30]处对学习率进行衰减,lr = lr * gammascheduler = lr_scheduler.MultiStepLR(optimizer, milestones原创 2020-09-23 10:22:44 · 1723 阅读 · 3 评论 -
pytorch迁移学习详解(提取整个网络or提取网络中的指定层+Segnet)
pytorch迁移学习1.完整代码2.迁移学习**1.首先分析“vgg.features.children()”这句话的精髓,理解了这句代码,一切就好办了。****First**,我们理解feature从哪儿来的。**Second**,现在理解vgg.features.children()中的**children()**2. 最后,把目光转向 self.enc1 = nn.Sequential(*features[3:7])本例子以图像语义分割最终的Segnet网络为例进行说明,简单明了,希望对大家有所帮原创 2020-07-30 11:07:46 · 3591 阅读 · 3 评论 -
pytorch网络结构可视化(torchsummary+tensorboardX+网络多输通道入情况+多分支网络)
**torchsummary和tensorboardX的使用****1.单通道输入网络**结构1结构2实例012.多通道输入网络结构实例02(只用了卷积层进行演示)**3.参考链接:**1.单通道输入网络单通道输入的情况大致有以下两种结构:结构1只有一条路可以走结构2输入为一条路,输出为多条路以上两种的输入只有一个input,这种是经常遇到的情况。实例01import torchimport torch.nn as nnimport torch.nn.functional as F原创 2020-07-20 12:51:29 · 6265 阅读 · 1 评论