spandrel.architectures.RestoreFormer.RestoreFormer

class documentation

class RestoreFormer(nn.Module): (source)

Constructor: RestoreFormer(n_embed, embed_dim, ch, out_ch, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method decode Undocumented
Method encode Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable decoder Undocumented
Instance Variable encoder Undocumented
Instance Variable post_quant_conv Undocumented
Instance Variable quant_conv Undocumented
Instance Variable quantize Undocumented
def __init__(self, *, n_embed=1024, embed_dim=256, ch=64, out_ch=3, ch_mult=(1, 2, 2, 4, 4, 8), num_res_blocks=2, attn_resolutions=(16), dropout=0.0, in_channels=3, resolution=512, z_channels=256, double_z=False, enable_mid=True, fix_decoder=False, fix_codebook=True, fix_encoder=False, head_size=8): (source)

Undocumented

def decode(self, quant, hs): (source)

Undocumented

def encode(self, x): (source)

Undocumented

def forward(self, input, **kwargs): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

decoder = (source)

Undocumented

encoder = (source)

Undocumented

post_quant_conv = (source)

Undocumented

quant_conv = (source)

Undocumented

quantize = (source)

Undocumented