cyto_dl.nn.discriminators.multi_scale_discriminator module#

class cyto_dl.nn.discriminators.multi_scale_discriminator.MultiScaleDiscriminator(n_scales: int = 2, dim: int = 3, **kwargs)[source]#

Bases: Module

Modified version of Pix2PixHD discriminator, which returns discriminator activations at multiple spatial scales.

Parameters:

  • n_scales (int=2) – Number of spatial scales to

  • **kwargs – Arguments to pass to NLayerDiscriminator

forward(x, real, pred)[source]#