Closed houqian2180320171 closed 4 years ago
hi,i can not find the code of spm in you model,can you show me where is it? thanks
You can find it in models/resnet.py
def init __ (self, nDenselayer, channels, growth, norm_layer = nn.BatchNorm2d, kwargs): super (RDB1, self) .** init __ () modules = [] channels_buffer = channels for i in range (nDenselayer): modules.append (make_dense1 (channels_buffer, growth, 3, norm_layer, self._up_kwargs))
you can ignore it.
def init (self, nDenselayer, channels, growth, norm_layer = nn.BatchNorm2d, ** kwargs): super (RDB1, self) . init () modules = [] channels_buffer = channels for i in range (nDenselayer): modules.append (make_dense1 (channels_buffer, growth, 3, norm_layer, self._up_kwargs))