(self, net, inp, corr, flow)
| 114 | self.flow_head = FlowHead(hidden_dim, hidden_dim=128) |
| 115 | |
| 116 | def forward(self, net, inp, corr, flow): |
| 117 | motion_features = self.encoder(flow, corr) |
| 118 | inp = torch.cat([inp, motion_features], dim=1) |
| 119 | net = self.gru(net, inp) |
| 120 | delta_flow = self.flow_head(net) |
| 121 | |
| 122 | return net, None, delta_flow |
| 123 | |
| 124 | class BasicUpdateBlock(nn.Module): |
| 125 | def __init__(self, args, hidden_dim=128, input_dim=128): |
nothing calls this directly
no outgoing calls
no test coverage detected