spandrel.architectures.IPT.IPT

class documentation

class IPT(nn.Module): (source)

Constructor: IPT(patch_size, patch_dim, n_feats, rgb_range, ...)

Undocumented

Method	`__init__`	Undocumented
Method	`forward`	Undocumented
Method	`set_scale`	Undocumented
Class Variable	`hyperparameters`	Undocumented
Instance Variable	`add_mean`	Undocumented
Instance Variable	`body`	Undocumented
Instance Variable	`head`	Undocumented
Instance Variable	`patch_size`	Undocumented
Instance Variable	`rgb_range`	Undocumented
Instance Variable	`scale`	Undocumented
Instance Variable	`scale_idx`	Undocumented
Instance Variable	`sub_mean`	Undocumented
Instance Variable	`tail`	Undocumented

def __init__(self, *, patch_size: int = 48, patch_dim: int = 3, n_feats: int = 64, rgb_range: float = 255, n_colors: int = 3, scale: Sequence[int] = [1], num_heads: int = 12, num_layers: int = 12, num_queries: int = 1, dropout_rate: float = 0, mlp=True, pos_every=False, no_pos=False, no_norm=False, conv=default_conv): (source) ¶

Undocumented

def forward(self, x): (source) ¶

Undocumented

def set_scale(self, scale_idx): (source) ¶

Undocumented

hyperparameters: dict = (source) ¶

Undocumented

add_mean = (source) ¶

Undocumented

body = (source) ¶

Undocumented

head = (source) ¶