spandrel_extra_arches.architectures.AdaCode.AdaCode

class documentation

class AdaCode(nn.Module): (source)

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

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
Instance Variable	`weight_predictor`	Undocumented
Instance Variable	`weight_softmax`	Undocumented

def __init__(self, *, in_channel=3, codebook_params: list[list[int]] = [[32, 256, 256]], gt_resolution=256, LQ_stage=False, norm_type='gn', act_type='silu', use_quantize=True, scale_factor=1, use_residual=True, weight_softmax=False): (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) ¶