DenseNet 测试

最新推荐文章于 2022-02-24 12:30:42 发布

AI算法网奇

最新推荐文章于 2022-02-24 12:30:42 发布

阅读量2.5k

点赞数

分类专栏：深度学习宝典

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

版权

深度学习宝典专栏收录该内容

986 篇文章 713 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文测试了DenseNet在深度为40，分辨率32*32时，在CPU上的运行情况，平均需要46毫秒完成运算。

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

测试，网络深了也挺慢的。

深度40 分辨率32*32 cpu需要46ms


from __future__ import print_function, division, absolute_import

import time
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.utils.model_zoo as model_zoo
from torch.autograd import Variable
import numpy as np

import torch

import torch.nn as nn
import torch.optim as optim

import torch.nn.functional as F
from torch.autograd import Variable

import torchvision.datasets as dset
import torchvision.transforms as transforms
from torch.utils.data import DataLoader

import torchvision.models as models

import sys
import math

class Bottleneck(nn.Module):
    def __init__(self, nChannels, growthRate):
        super(Bottleneck, self).__init__()
        interC

了解本专栏