Method forward

models/unet_model_google.py:89–90

(self, x)

        self.conv = nn.Conv2d(dim, dim, 3, 2, 1)

    def forward(self, x):
        return self.conv(x)


# building block modules