(self, planes, blocks, stride=1)
| 127 | self.attnpool = AttentionPool2d(input_resolution // 32, embed_dim, heads, output_dim) |
| 128 | |
| 129 | def _make_layer(self, planes, blocks, stride=1): |
| 130 | layers = [Bottleneck(self._inplanes, planes, stride)] |
| 131 | |
| 132 | self._inplanes = planes * Bottleneck.expansion |
| 133 | for _ in range(1, blocks): |
| 134 | layers.append(Bottleneck(self._inplanes, planes)) |
| 135 | |
| 136 | return nn.Sequential(*layers) |
| 137 | |
| 138 | def forward(self, x): |
| 139 | def stem(x): |