sccross.models.layers.Discriminator

class sccross.models.layers.Discriminator(in_features, out_features, n_batches=0, h_depth=2, h_dim=256, dropout=0.2)[source]

Bases: torch.nn.modules.container.Sequential, torch.nn.modules.module.Module

Domain discriminator

Parameters

  • in_features (int) – Input dimensionality

  • out_features (int) – Output dimensionality

  • h_depth (int) – Hidden layer depth

  • h_dim (Optional[int]) – Hidden layer dimensionality

  • dropout (float) – Dropout rate

Methods

forward

Define the computation performed at every call.

Attributes