res2net 测试

举报
风吹稻花香 发表于 2021/06/04 22:46:44 2021/06/04
【摘要】 这个都跑不起来。内存溢出。 https://github.com/4uiiurz1/pytorch-res2net   import time import torchimport torch.nn as nn __all__ = ['ImageNetRes2Net', 'res2net50', 'res2net101', 'res2net152', 're...

这个都跑不起来。内存溢出。

https://github.com/4uiiurz1/pytorch-res2net

 


      import time
      import torch
      import torch.nn as nn
      __all__ = ['ImageNetRes2Net', 'res2net50', 'res2net101',
      'res2net152', 'res2next50_32x4d', 'se_res2net50',
      'CifarRes2Net', 'res2next29_6cx24wx4scale',
      'res2next29_8cx25wx4scale', 'res2next29_6cx24wx6scale',
      'res2next29_6cx24wx4scale_se', 'res2next29_8cx25wx4scale_se',
      'res2next29_6cx24wx6scale_se']
      def conv3x3(in_planes, out_planes, stride=1, groups=1):
      """3x3 convolution with padding"""
      return nn.Conv2d(in_planes, out_planes, kernel_size=3, stride=stride,
       padding=1, groups=groups, bias=False)
      def conv1x1(in
  
 

文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/jacke121/article/details/91815976

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。