spandrel.architectures.GRL

package documentation
(source)

Undocumented

From __init__.py:

Class GRL Image restoration transformer with global, non-local, and local connections
Class GRLArch Undocumented
Function _clean_up_checkpoint Undocumented
Function _get_anchor_params Undocumented
Function _get_output_params Undocumented
Function _inv_div_add Assuming that a = b + b // d, this will return b.
Constant _NON_PERSISTENT_BUFFERS Undocumented
_NON_PERSISTENT_BUFFERS: list[str] = (source)

Undocumented

Value
['table_w',
 'table_sh',
 'table_sv',
 'index_w',
 'index_sh_a2w',
 'index_sh_w2a',
 'index_sv_a2w',
...
def _clean_up_checkpoint(state_dict: StateDict) -> StateDict: (source)

Undocumented

def _get_output_params(state_dict: StateDict, in_channels: int): (source)

Undocumented

def _get_anchor_params(state_dict: StateDict, default_down_factor: int) -> tuple[bool, str, int]: (source)

Undocumented

def _inv_div_add(a: int, d: int) -> int: (source)

Assuming that a = b + b // d, this will return b.