(self, size, dim=512, dim_motion=20)
| 238 | |
| 239 | class Encoder(nn.Module): |
| 240 | def __init__(self, size, dim=512, dim_motion=20): |
| 241 | super(Encoder, self).__init__() |
| 242 | |
| 243 | # appearance netmork |
| 244 | self.net_app = EncoderApp(size, dim) |
| 245 | |
| 246 | # motion network |
| 247 | fc = [EqualLinear(dim, dim)] |
| 248 | for i in range(3): |
| 249 | fc.append(EqualLinear(dim, dim)) |
| 250 | |
| 251 | fc.append(EqualLinear(dim, dim_motion)) |
| 252 | self.fc = nn.Sequential(*fc) |
| 253 | |
| 254 | def enc_app(self, x): |
| 255 |
nothing calls this directly
no test coverage detected