spandrel_extra_arches.architectures.FeMaSR.FeMaSR

class documentation

class FeMaSR(nn.Module): (source)

Constructor: FeMaSR(in_channel, codebook_params, gt_resolution, LQ_stage, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method encode_and_decode Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable after_quant_group Undocumented
Instance Variable before_quant_group Undocumented
Instance Variable codebook_scale Undocumented
Instance Variable decoder_group Undocumented
Instance Variable gt_res Undocumented
Instance Variable in_channel Undocumented
Instance Variable LQ_stage Undocumented
Instance Variable max_depth Undocumented
Instance Variable multiscale_encoder Undocumented
Instance Variable out_conv Undocumented
Instance Variable quantize_group Undocumented
Instance Variable scale_factor Undocumented
Instance Variable use_quantize Undocumented
Instance Variable use_residual Undocumented
def __init__(self, *, in_channel=3, codebook_params=[[32, 1024, 512]], gt_resolution=256, LQ_stage=False, norm_type='gn', act_type='silu', use_quantize=True, scale_factor=1, use_residual=True): (source)

Undocumented

def encode_and_decode(self, input): (source)

Undocumented

def forward(self, input): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

after_quant_group = (source)

Undocumented

before_quant_group = (source)

Undocumented

codebook_scale = (source)

Undocumented

decoder_group = (source)

Undocumented

gt_res = (source)

Undocumented

in_channel = (source)

Undocumented

LQ_stage = (source)

Undocumented

max_depth = (source)

Undocumented

multiscale_encoder = (source)

Undocumented

out_conv = (source)

Undocumented

quantize_group = (source)

Undocumented

scale_factor = (source)

Undocumented

use_quantize = (source)

Undocumented

use_residual = (source)

Undocumented