Method _make_conv

sat/sgm/modules/diffusionmodules/model.py:644–645

(self)

        return ResnetBlock

    def _make_conv(self) -> Callable:
        return torch.nn.Conv2d

    def get_last_layer(self, **kwargs):
        return self.conv_out.weight