ResidualDiscriminator

class torchelie.models.ResidualDiscriminator(arch: List[Union[str, int]])

add_minibatch_stddev() → torchelie.models.snres_discr.ResidualDiscriminator

forward(x: torch.Tensor, y: Optional[torch.Tensor] = None) → torch.Tensor

set_input_specs(in_channels: int) → torchelie.models.snres_discr.ResidualDiscriminator

to_equal_lr(leak=0.2) → torchelie.models.snres_discr.ResidualDiscriminator

Experimental

Warning

ResidualDiscriminator.to_equal_lr() is experimental, and may change or be deleted soon if not already broken

.

to_projection_discr(num_classes: int) → torchelie.models.snres_discr.ResidualDiscriminator

to_spectral_norm() → torchelie.models.snres_discr.ResidualDiscriminator

Experimental

Warning

ResidualDiscriminator.to_spectral_norm() is experimental, and may change or be deleted soon if not already broken

.

classifier: Union[torchelie.models.classifier.ClassificationHead, torchelie.models.classifier.ProjectionDiscr]