def get_segmentation_net(self): encoder = self.encoder segmentation_decoder = self.segmentation_decoder encoder.return_gradient_reversal = False segmentation_decoder.return_gradient_reversal = False return unet_from_encoder_decoder(encoder, segmentation_decoder)
def get_segmentation_net(self): return unet_from_encoder_decoder(self.encoder, self.segmentation_decoder)