(self, x)
| 187 | self.encode = nn.Sequential(*layers) |
| 188 | |
| 189 | def forward(self, x): |
| 190 | if self.downsample: |
| 191 | x = self.maxpool(x) |
| 192 | x = self.encode(x) |
| 193 | return x |
| 194 | |
| 195 | class Transformer(nn.Module): |
| 196 | def __init__(self, in_channels1, in_channels2, feat_size1, feat_size2, hidden_size=args['D']): |
nothing calls this directly
no outgoing calls
no test coverage detected