(self)
| 61 | self.update_block = BasicUpdateBlock(self.args, hidden_dim=hdim) |
| 62 | |
| 63 | def freeze_bn(self): |
| 64 | for m in self.modules(): |
| 65 | if isinstance(m, nn.BatchNorm2d): |
| 66 | m.eval() |
| 67 | |
| 68 | def initialize_flow(self, img): |
| 69 | """ Flow is represented as difference between two coordinate grids flow = coords1 - coords0""" |
nothing calls this directly
no outgoing calls
no test coverage detected