import torchfrom collections import OrderedDictimport torchimport torchnn as nndef make_layersblock no_relu_layers layers = for layer_name v in blockitems if pool in layer_name
这段代码中的卷积方式为普通的卷积(nn.Conv2d)。
原文地址: https://www.cveoy.top/t/topic/hdae 著作权归作者所有。请勿转载和采集!