| 61 | |
| 62 | |
| 63 | class MotionPixelNorm(nn.Module): |
| 64 | def __init__(self): |
| 65 | super().__init__() |
| 66 | |
| 67 | def forward(self, input): |
| 68 | return input * torch.rsqrt(torch.mean(input ** 2, dim=2, keepdim=True) + 1e-8) |
| 69 | |
| 70 | class Upsample(nn.Module): |
| 71 | def __init__(self, kernel, factor=2): |
nothing calls this directly
no outgoing calls
no test coverage detected