mmagic.models.editors.ttsr.ttsr_disc

Module Contents

Classes

TTSRDiscriminator

A discriminator for TTSR.

class mmagic.models.editors.ttsr.ttsr_disc.TTSRDiscriminator(in_channels=3, in_size=160, init_cfg=None)

Bases: mmengine.model.BaseModule

A discriminator for TTSR.

Parameters

• in_channels (int) – Channel number of inputs. Default: 3.

• in_size (int) – Size of input image. Default: 160.

• init_cfg (dict, optional) – Initialization config dict.

forward(x)

Forward function.

Parameters

x (Tensor) – Input tensor with shape (n, c, h, w).

Returns

Forward results.

Return type

Tensor