spandrel.architectures.DRCT.DRCT

class documentation

class DRCT(nn.Module): (source)

Constructor: DRCT(img_size, patch_size, in_chans, embed_dim, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward Undocumented
Method forward_features Undocumented
Method no_weight_decay Undocumented
Method no_weight_decay_keywords Undocumented
Class Variable hyperparameters Undocumented
Instance Variable absolute_pos_embed Undocumented
Instance Variable ape Undocumented
Instance Variable conv_after_body Undocumented
Instance Variable conv_before_upsample Undocumented
Instance Variable conv_first Undocumented
Instance Variable conv_last Undocumented
Instance Variable embed_dim Undocumented
Instance Variable img_range Undocumented
Instance Variable layers Undocumented
Instance Variable mean Undocumented
Instance Variable mlp_ratio Undocumented
Instance Variable norm Undocumented
Instance Variable num_features Undocumented
Instance Variable num_layers Undocumented
Instance Variable patch_embed Undocumented
Instance Variable patch_norm Undocumented
Instance Variable patch_unembed Undocumented
Instance Variable patches_resolution Undocumented
Instance Variable pos_drop Undocumented
Instance Variable shift_size Undocumented
Instance Variable upsample Undocumented
Instance Variable upsampler Undocumented
Instance Variable upscale Undocumented
Instance Variable window_size Undocumented
Method _init_weights Undocumented
def __init__(self, *, img_size=64, patch_size=1, in_chans=3, embed_dim=180, depths=(6, 6, 6, 6, 6, 6), num_heads=(6, 6, 6, 6, 6, 6), window_size=16, mlp_ratio=2.0, qkv_bias=True, qk_scale=None, drop_rate=0.0, attn_drop_rate=0.0, drop_path_rate=0.1, norm_layer=nn.LayerNorm, ape=False, patch_norm=True, upscale=1, img_range=1.0, upsampler='', resi_connection='1conv', gc=32): (source)

Undocumented

def forward(self, x): (source)

Undocumented

def forward_features(self, x): (source)

Undocumented

@torch.jit.ignore
def no_weight_decay(self): (source)

Undocumented

@torch.jit.ignore
def no_weight_decay_keywords(self): (source)

Undocumented

hyperparameters: dict = (source)

Undocumented

absolute_pos_embed = (source)

Undocumented

ape = (source)

Undocumented

conv_after_body = (source)

Undocumented

conv_before_upsample = (source)

Undocumented

conv_first = (source)

Undocumented

conv_last = (source)

Undocumented

embed_dim = (source)

Undocumented

img_range = (source)

Undocumented

layers = (source)

Undocumented

mean = (source)

Undocumented

mlp_ratio = (source)

Undocumented

norm = (source)

Undocumented

num_features = (source)

Undocumented

num_layers = (source)

Undocumented

patch_embed = (source)

Undocumented

patch_norm = (source)

Undocumented

patch_unembed = (source)

Undocumented

patches_resolution = (source)

Undocumented

pos_drop = (source)

Undocumented

shift_size = (source)

Undocumented

upsample = (source)

Undocumented

upsampler = (source)

Undocumented

upscale = (source)

Undocumented

window_size = (source)

Undocumented

def _init_weights(self, m): (source)

Undocumented