(self, x)
| 105 | |
| 106 | |
| 107 | def forward(self, x): |
| 108 | y = x |
| 109 | y = self.relu(self.norm1(self.conv1(y))) |
| 110 | y = self.relu(self.norm2(self.conv2(y))) |
| 111 | y = self.relu(self.norm3(self.conv3(y))) |
| 112 | |
| 113 | if self.downsample is not None: |
| 114 | x = self.downsample(x) |
| 115 | |
| 116 | return self.relu(x+y) |
| 117 | |
| 118 | class BasicEncoder(nn.Module): |
| 119 | def __init__(self, output_dim=128, norm_fn='batch', dropout=0.0): |
nothing calls this directly
no outgoing calls
no test coverage detected