Method __init__

models/unet_model_google.py:73–77

(self, dim, save_fm=False)

class Upsample(nn.Module):
    def __init__(self, dim, save_fm=False):
        super().__init__()
        self.up = nn.Upsample(scale_factor=2, mode="nearest")
        self.conv = nn.Conv2d(dim, dim, 3, padding=1)
        self.save_fm = save_fm

    def forward(self, x):
        x = self.conv(self.up(x))