MCPcopy Create free account
hub / github.com/thygate/stable-diffusion-webui-depthmap-script / _make_resnet_backbone

Function _make_resnet_backbone

dmidas/blocks.py:192–202  ·  view source on GitHub ↗
(resnet)

Source from the content-addressed store, hash-verified

190
191
192def _make_resnet_backbone(resnet):
193 pretrained = nn.Module()
194 pretrained.layer1 = nn.Sequential(
195 resnet.conv1, resnet.bn1, resnet.relu, resnet.maxpool, resnet.layer1
196 )
197
198 pretrained.layer2 = resnet.layer2
199 pretrained.layer3 = resnet.layer3
200 pretrained.layer4 = resnet.layer4
201
202 return pretrained
203
204
205def _make_pretrained_resnext101_wsl(use_pretrained):

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected