![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
计算机视觉
记录学习中的问题
Huranqingqing
这个作者很懒,什么都没留下…
展开
-
torch和tensorflow各版本下载地址,tensorflow与cuda版本对应关系
下载地址:https://download.pytorch.org/whl/torch_stable.html原创 2022-03-03 09:41:07 · 1000 阅读 · 0 评论 -
AttributeError: ‘str‘ object has no attribute ‘copy
在加载模型时出现 AttributeError: 'str' object has no attribute 'copy' 错误解决方案:将model.load_state_dict(state_dict) 修改为:model.load_state_dict(torch.load(PATH))原创 2022-03-01 09:57:18 · 3120 阅读 · 0 评论 -
vgg网络结构
最近在基于vgg的网络模型上进行改进,记录下几种常见的vgg模型结构:vgg11,vgg13,vgg16,vgg19vgg11VGG-11的网络结构: VGG( (features): Sequential( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (1): ReLU(inplace=True) (2): MaxPool2d(kernel_size=2, stride=2,原创 2021-07-17 13:40:35 · 609 阅读 · 0 评论