作者:NewGuy | 来源:互联网 | 2023-09-05 14:12
spnasnethttps:github.comfireoilpytorch_template_codeblob3484140d0b6df8713a75e98152e831968
spnasnet
https://github.com/fireoil/pytorch_template_code/blob/3484140d0b6df8713a75e98152e831968d0311fc/model/spnasnet.py
https://github.com/iofthetiger/pkuad/blob/069a3cd746f955deaf1b3c0891d32eadac874bc2/pytorchcv/models/spnasnet.py
不带分类的backbo:
"""Single-Path NASNet for ImageNet-1K, implemented in PyTorch.Original paper: 'Single-Path NAS: Designing Hardware-Efficient ConvNets in less than 4 Hours,'https://arxiv.org/abs/1904.02877.
"""
import timeimport torch__all__ = ['SPNASNet', 'spnasnet']import os
import torch.nn as nn
import torch.nn.init as init
from net.common impor