MCPcopy Create free account
hub / github.com/CHENGY12/PLOT / stem

Method stem

plot-pp/clip/model.py:139–143  ·  view source on GitHub ↗
(x)

Source from the content-addressed store, hash-verified

137
138 def forward(self, x):
139 def stem(x):
140 for conv, bn in [(self.conv1, self.bn1), (self.conv2, self.bn2), (self.conv3, self.bn3)]:
141 x = self.relu(bn(conv(x)))
142 x = self.avgpool(x)
143 return x
144
145 x = x.type(self.conv1.weight.dtype)
146 x = stem(x)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected