(self, args, hidden_dim=96)
| 98 | |
| 99 | class SmallUpdateBlock(nn.Module): |
| 100 | def __init__(self, args, hidden_dim=96): |
| 101 | super(SmallUpdateBlock, self).__init__() |
| 102 | self.encoder = SmallMotionEncoder(args) |
| 103 | self.gru = ConvGRU(hidden_dim=hidden_dim, input_dim=82+64) |
| 104 | self.flow_head = FlowHead(hidden_dim, hidden_dim=128) |
| 105 | |
| 106 | def forward(self, net, inp, corr, flow): |
| 107 | motion_features = self.encoder(flow, corr) |
nothing calls this directly
no test coverage detected