超级会员免费看
在1060显卡上，尝试运行Bagnet模型进行图像识别任务，发现在处理640*640分辨率的输入时出现性能瓶颈，模型的权重大小约为60MB。
1060上，1个640*640都跑不了
权重60多m
import time
import torch.nn as nn
import math
import torch
from collections import OrderedDict
from torch.utils import model_zoo
import os
dir_path = os.path.dirname(os.path.realpath(__file__))
__all__ = ['bagnet9', 'bagnet17', 'bagnet33']
model_urls = {
'bagnet9': 'https://bitbucket.org/wielandbrendel/bag-of-feature-pretrained-models/raw/249e8fa82c0913623a807d9d35eeab9da7dcc2a8/bagnet8-34f4ccd2.pth.tar',
'bagnet17': 'https://bitbucket.org/wielandbrendel/bag-of-feature-pretrained-models/raw/249e8fa82c0913623a807d9d35eeab9da7dcc2a8/bagnet16-105524de.pth.tar',
'bagnet33': 'https://bitbucket.org/wielandbrendel/ba
订阅专栏 解锁全文
扫一扫
3184
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
