res2net 测试

最新推荐文章于 2024-04-14 20:50:36 发布

AI算法网奇

最新推荐文章于 2024-04-14 20:50:36 发布

阅读量3.1k

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/91815976

版权

pytorch知识宝典专栏收录该内容

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

在尝试运行 res2net 模型时遇到内存溢出的问题，详细记录了在 GitHub 上找到的 pytorch-res2net 实现中遇到的困难。

摘要由CSDN通过智能技术生成

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

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 convoluti

了解本专栏