(self, net, inp, corr, flow, upsample=True)
| 125 | nn.Conv2d(256, 64*9, 1, padding=0)) |
| 126 | |
| 127 | def forward(self, net, inp, corr, flow, upsample=True): |
| 128 | motion_features = self.encoder(flow, corr) |
| 129 | inp = torch.cat([inp, motion_features], dim=1) |
| 130 | |
| 131 | net = self.gru(net, inp) |
| 132 | delta_flow = self.flow_head(net) |
| 133 | |
| 134 | # scale mask to balence gradients |
| 135 | mask = .25 * self.mask(net) |
| 136 | return net, mask, delta_flow |
| 137 | |
| 138 | |
| 139 |
nothing calls this directly
no outgoing calls
no test coverage detected