Final Train Runs

2021-03-18 07:45:06 +01:00
parent f89f0f8528
commit fc4617c9d8
4 changed files with 49 additions and 18 deletions
--- a/modules/util.py
+++ b/modules/util.py
@ -207,15 +207,11 @@ class ShapeMixin:
            return shape


-class F_x(ShapeMixin, nn.Module):
+class F_x(ShapeMixin, nn.Identity):
    def __init__(self, in_shape):
        super(F_x, self).__init__()
        self.in_shape = in_shape

-    @staticmethod
-    def forward(x):
-        return x
-

 class SlidingWindow(ShapeMixin, nn.Module):
    def __init__(self, in_shape, kernel, stride=1, padding=0, keepdim=False):