SYMBOL INDEX (461 symbols across 38 files)

FILE: src/data/__init__.py
  class MyConcatDataset (line 7) | class MyConcatDataset(ConcatDataset):
    method __init__ (line 8) | def __init__(self, datasets):
    method set_scale (line 12) | def set_scale(self, idx_scale):
  class Data (line 16) | class Data:
    method __init__ (line 17) | def __init__(self, args):

FILE: src/data/benchmark.py
  class Benchmark (line 13) | class Benchmark(srdata.SRData):
    method __init__ (line 14) | def __init__(self, args, name='', train=True, benchmark=True):
    method _scan (line 18) | def _scan(self):
    method _set_filesystem (line 39) | def _set_filesystem(self, dir_data):

FILE: src/data/common.py
  function get_patch (line 8) | def get_patch(*args, patch_size=96, scale=2, multi=False, input_large=Fa...
  function set_channel (line 34) | def set_channel(*args, n_channels=3):
  function np2Tensor (line 49) | def np2Tensor(*args, rgb_range=255):
  function augment (line 59) | def augment(*args, hflip=True, rot=True):

FILE: src/data/demo.py
  class Demo (line 11) | class Demo(data.Dataset):
    method __init__ (line 12) | def __init__(self, args, name='Demo', train=False, benchmark=False):
    method __getitem__ (line 26) | def __getitem__(self, idx):
    method __len__ (line 34) | def __len__(self):
    method set_scale (line 37) | def set_scale(self, idx_scale):

FILE: src/data/div2k.py
  class DIV2K (line 4) | class DIV2K(srdata.SRData):
    method __init__ (line 5) | def __init__(self, args, name='DIV2K', train=True, benchmark=False):
    method _scan (line 20) | def _scan(self):
    method _set_filesystem (line 27) | def _set_filesystem(self, dir_data):

FILE: src/data/div2kjpeg.py
  class DIV2KJPEG (line 5) | class DIV2KJPEG(div2k.DIV2K):
    method __init__ (line 6) | def __init__(self, args, name='', train=True, benchmark=False):
    method _set_filesystem (line 12) | def _set_filesystem(self, dir_data):

FILE: src/data/sr291.py
  class SR291 (line 3) | class SR291(srdata.SRData):
    method __init__ (line 4) | def __init__(self, args, name='SR291', train=True, benchmark=False):

FILE: src/data/srdata.py
  class SRData (line 15) | class SRData(data.Dataset):
    method __init__ (line 16) | def __init__(self, args, name='', train=True, benchmark=False):
    method _scan (line 71) | def _scan(self):
    method _set_filesystem (line 87) | def _set_filesystem(self, dir_data):
    method _check_and_load (line 94) | def _check_and_load(self, ext, img, f, verbose=True):
    method __getitem__ (line 101) | def __getitem__(self, idx):
    method __len__ (line 109) | def __len__(self):
    method _get_index (line 115) | def _get_index(self, idx):
    method _load_file (line 121) | def _load_file(self, idx):
    method get_patch (line 140) | def get_patch(self, lr, hr):
    method set_scale (line 158) | def set_scale(self, idx_scale):

FILE: src/data/video.py
  class Video (line 12) | class Video(data.Dataset):
    method __init__ (line 13) | def __init__(self, args, name='Video', train=False, benchmark=False):
    method __getitem__ (line 27) | def __getitem__(self, idx):
    method __len__ (line 39) | def __len__(self):
    method set_scale (line 42) | def set_scale(self, idx_scale):

FILE: src/dataloader.py
  function _ms_loop (line 22) | def _ms_loop(dataset, index_queue, data_queue, done_event, collate_fn, s...
  class _MSDataLoaderIter (line 68) | class _MSDataLoaderIter(_DataLoaderIter):
    method __init__ (line 70) | def __init__(self, loader):
  class MSDataLoader (line 148) | class MSDataLoader(DataLoader):
    method __init__ (line 150) | def __init__(self, cfg, *args, **kwargs):
    method __iter__ (line 156) | def __iter__(self):

FILE: src/loss/__init__.py
  class Loss (line 14) | class Loss(nn.modules.loss._Loss):
    method __init__ (line 15) | def __init__(self, args, ckp):
    method forward (line 69) | def forward(self, sr, hr):
    method step (line 86) | def step(self):
    method start_log (line 91) | def start_log(self):
    method end_log (line 94) | def end_log(self, n_batches):
    method display_loss (line 97) | def display_loss(self, batch):
    method plot_loss (line 105) | def plot_loss(self, apath, epoch):
    method get_loss_module (line 119) | def get_loss_module(self):
    method save (line 125) | def save(self, apath):
    method load (line 129) | def load(self, apath, cpu=False):

FILE: src/loss/adversarial.py
  class Adversarial (line 12) | class Adversarial(nn.Module):
    method __init__ (line 13) | def __init__(self, args, gan_type):
    method forward (line 35) | def forward(self, fake, real):
    method state_dict (line 95) | def state_dict(self, *args, **kwargs):
    method bce (line 101) | def bce(self, real, fake):

FILE: src/loss/discriminator.py
  class Discriminator (line 5) | class Discriminator(nn.Module):
    method __init__ (line 9) | def __init__(self, args):
    method forward (line 50) | def forward(self, x):

FILE: src/loss/vgg.py
  class VGG (line 8) | class VGG(nn.Module):
    method __init__ (line 9) | def __init__(self, conv_index, rgb_range=1):
    method forward (line 24) | def forward(self, sr, hr):

FILE: src/main.py
  function main (line 13) | def main():

FILE: src/model/__init__.py
  class Model (line 10) | class Model(nn.Module):
    method __init__ (line 11) | def __init__(self, args, ckp):
    method forward (line 39) | def forward(self, x, idx_scale):
    method save (line 60) | def save(self, apath, epoch, is_best=False):
    method load (line 73) | def load(self, apath, pre_train='', resume=-1, cpu=False):
    method forward_chop (line 106) | def forward_chop(self, x, shave=10, min_size=160000):
    method forward_x8 (line 147) | def forward_x8(self, *args, forward_function=None):

FILE: src/model/common.py
  function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True):
  class MeanShift (line 12) | class MeanShift(nn.Conv2d):
    method __init__ (line 13) | def __init__(
  class BasicBlock (line 24) | class BasicBlock(nn.Sequential):
    method __init__ (line 25) | def __init__(
  class ResBlock (line 37) | class ResBlock(nn.Module):
    method __init__ (line 38) | def __init__(
    method forward (line 54) | def forward(self, x):
  class Upsampler (line 60) | class Upsampler(nn.Sequential):
    method __init__ (line 61) | def __init__(self, conv, scale, n_feats, bn=False, act=False, bias=True):

FILE: src/model/dcn/deform_conv.py
  class DeformConvFunction (line 15) | class DeformConvFunction(Function):
    method forward (line 17) | def forward(ctx, input, offset, weight, stride=1, padding=0, dilation=...
    method backward (line 51) | def backward(ctx, grad_output):
    method _output_size (line 82) | def _output_size(input, weight, padding, dilation, stride):
  class ModulatedDeformConvFunction (line 97) | class ModulatedDeformConvFunction(Function):
    method forward (line 99) | def forward(ctx, input, offset, mask, weight, bias=None, stride=1, pad...
    method backward (line 124) | def backward(ctx, grad_output):
    method _infer_shape (line 145) | def _infer_shape(ctx, input, weight):
  class DeformConv (line 161) | class DeformConv(nn.Module):
    method __init__ (line 162) | def __init__(self, in_channels, out_channels, kernel_size, stride=1, p...
    method reset_parameters (line 188) | def reset_parameters(self):
    method forward (line 195) | def forward(self, x, offset):
  class DeformConvPack (line 200) | class DeformConvPack(DeformConv):
    method __init__ (line 201) | def __init__(self, *args, **kwargs):
    method init_offset (line 211) | def init_offset(self):
    method forward (line 215) | def forward(self, x):
  class ModulatedDeformConv (line 221) | class ModulatedDeformConv(nn.Module):
    method __init__ (line 222) | def __init__(self, in_channels, out_channels, kernel_size, stride=1, p...
    method reset_parameters (line 243) | def reset_parameters(self):
    method forward (line 252) | def forward(self, x, offset, mask):
  class ModulatedDeformConvPack (line 258) | class ModulatedDeformConvPack(ModulatedDeformConv):
    method __init__ (line 259) | def __init__(self, *args, extra_offset_mask=False, **kwargs):
    method init_offset (line 270) | def init_offset(self):
    method forward (line 274) | def forward(self, x):

FILE: src/model/dcn/setup.py
  function make_cuda_ext (line 5) | def make_cuda_ext(name, sources):

FILE: src/model/dcn/src/deform_conv_cuda.cpp
  function shape_check (line 61) | void shape_check(at::Tensor input, at::Tensor offset, at::Tensor *gradOu...
  function deform_conv_forward_cuda (line 151) | int deform_conv_forward_cuda(at::Tensor input, at::Tensor weight,
  function deform_conv_backward_input_cuda (line 260) | int deform_conv_backward_input_cuda(at::Tensor input, at::Tensor offset,
  function deform_conv_backward_parameters_cuda (line 373) | int deform_conv_backward_parameters_cuda(
  function modulated_deform_conv_cuda_forward (line 486) | void modulated_deform_conv_cuda_forward(
  function modulated_deform_conv_cuda_backward (line 566) | void modulated_deform_conv_cuda_backward(
  function PYBIND11_MODULE (line 681) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: src/model/ddbpn.py
  function make_model (line 10) | def make_model(args, parent=False):
  function projection_conv (line 13) | def projection_conv(in_channels, out_channels, scale, up=True):
  class DenseProjection (line 29) | class DenseProjection(nn.Module):
    method __init__ (line 30) | def __init__(self, in_channels, nr, scale, up=True, bottleneck=True):
    method forward (line 55) | def forward(self, x):
  class DDBPN (line 68) | class DDBPN(nn.Module):
    method __init__ (line 69) | def __init__(self, args):
    method forward (line 112) | def forward(self, x):

FILE: src/model/edsr.py
  function make_model (line 14) | def make_model(args, parent=False):
  class EDSR (line 17) | class EDSR(nn.Module):
    method __init__ (line 18) | def __init__(self, args, conv=common.default_conv):
    method forward (line 55) | def forward(self, x):
    method load_state_dict (line 67) | def load_state_dict(self, state_dict, strict=True):

FILE: src/model/han.py
  function make_model (line 6) | def make_model(args, parent=False):
  class CALayer (line 10) | class CALayer(nn.Module):
    method __init__ (line 11) | def __init__(self, channel, reduction=16):
    method forward (line 23) | def forward(self, x):
  class LAM_Module (line 28) | class LAM_Module(nn.Module):
    method __init__ (line 30) | def __init__(self, in_dim):
    method forward (line 37) | def forward(self,x):
  class CSAM_Module (line 60) | class CSAM_Module(nn.Module):
    method __init__ (line 62) | def __init__(self, in_dim):
    method forward (line 71) | def forward(self,x):
  class RCAB (line 99) | class RCAB(nn.Module):
    method __init__ (line 100) | def __init__(
    method forward (line 114) | def forward(self, x):
  class ResidualGroup (line 121) | class ResidualGroup(nn.Module):
    method __init__ (line 122) | def __init__(self, conv, n_feat, kernel_size, reduction, act, res_scal...
    method forward (line 132) | def forward(self, x):
  class HAN (line 138) | class HAN(nn.Module):
    method __init__ (line 139) | def __init__(self, args, conv=common.default_conv):
    method forward (line 181) | def forward(self, x):
    method load_state_dict (line 212) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/matrixmodel.py
  function initialize_weights (line 30) | def initialize_weights(net_l, scale=1):
  class ResBlock (line 49) | class ResBlock(nn.Module):
    method __init__ (line 50) | def __init__(
    method forward (line 65) | def forward(self, x):
  class BFN (line 71) | class BFN(nn.Module):
    method __init__ (line 72) | def __init__(self, num_channels, kernel_size, reduction, n_blocks, blo...
    method _make_blocks (line 90) | def _make_blocks(self, in_channels, num_channels, kernel_size, reducti...
    method forward (line 98) | def forward(self, x):
  class BFN1 (line 117) | class BFN1(nn.Module):
    method __init__ (line 118) | def __init__(self, num_channels, kernel_size, reduction, n_blocks, blo...
    method _make_blocks (line 128) | def _make_blocks(self, in_channels, num_channels, kernel_size, reducti...
    method forward (line 136) | def forward(self, x):
  class BFN2 (line 144) | class BFN2(nn.Module):
    method __init__ (line 145) | def __init__(self, num_channels, kernel_size, reduction, n_blocks, blo...
    method _make_blocks (line 159) | def _make_blocks(self, in_channels, num_channels, kernel_size, reducti...
    method forward (line 167) | def forward(self, x):
  class EoctResBlock (line 181) | class EoctResBlock(nn.Module):
    method __init__ (line 184) | def __init__(self, in_channels, num_channels, stride=1, downsample=Non...
    method forward (line 193) | def forward(self, x):
  class EoctBottleneck (line 213) | class EoctBottleneck(nn.Module):
    method __init__ (line 214) | def __init__(self, in_channels, num_channels, stride=1, downsample=Non...
    method forward (line 227) | def forward(self, x):
  class CALayer (line 250) | class CALayer(nn.Module):
    method __init__ (line 251) | def __init__(self, in_channels, num_channels, reduction=16):
    method forward (line 259) | def forward(self, x):
  class CAEoctResBlock (line 270) | class CAEoctResBlock(nn.Module):
    method __init__ (line 271) | def __init__(self, in_channels, num_channels, reduction, bias=True, re...
    method forward (line 279) | def forward(self, x):
  function make_model (line 302) | def make_model(args, parent=False):
  class MatrixModel (line 305) | class MatrixModel(nn.Module):
    method __init__ (line 306) | def __init__(self, args):
    method _make_blocks (line 344) | def _make_blocks(self, in_channels, num_channels, kernel_size, reducti...
    method forward (line 352) | def forward(self, x):
    method load_state_dict (line 369) | def load_state_dict(self, state_dict, strict=False):
  class RERB (line 395) | class RERB(nn.Module):
    method __init__ (line 396) | def __init__(self, in_channels, num_channels, kernel_size, reduction, ...
    method _make_blocks (line 404) | def _make_blocks(self, in_channels, num_channels, kernel_size, reducti...
    method forward (line 412) | def forward(self, x):
  class MatrixModelB (line 421) | class MatrixModelB(nn.Module):
    method __init__ (line 422) | def __init__(self, args):
    method forward (line 496) | def forward(self, x):
    method load_state_dict (line 548) | def load_state_dict(self, state_dict, strict=False):
  class PDF (line 574) | class PDF(nn.Module):
    method __init__ (line 580) | def __init__(self, nf=64, groups=8):
    method forward (line 607) | def forward(self, nbr_fea_l):
  class PD (line 643) | class PD(nn.Module):
    method __init__ (line 649) | def __init__(self, nf=64, groups=8):
    method forward (line 681) | def forward(self, nbr_fea_l):
  class MatrixModelC (line 724) | class MatrixModelC(nn.Module):
    method __init__ (line 725) | def __init__(self, args):
    method forward (line 774) | def forward(self, x):
    method load_state_dict (line 813) | def load_state_dict(self, state_dict, strict=False):
  class MatrixModelD (line 840) | class MatrixModelD(nn.Module):
    method __init__ (line 841) | def __init__(self, args):
    method forward (line 908) | def forward(self, x):
    method load_state_dict (line 949) | def load_state_dict(self, state_dict, strict=False):
  class MatrixModelE (line 976) | class MatrixModelE(nn.Module):
    method __init__ (line 977) | def __init__(self, args):
    method forward (line 1047) | def forward(self, x):
    method load_state_dict (line 1085) | def load_state_dict(self, state_dict, strict=False):
  class MatrixModelF (line 1112) | class MatrixModelF(nn.Module):
    method __init__ (line 1113) | def __init__(self, args):
    method forward (line 1186) | def forward(self, x):
    method load_state_dict (line 1224) | def load_state_dict(self, state_dict, strict=False):
  class MatrixModelG (line 1251) | class MatrixModelG(nn.Module):
    method __init__ (line 1252) | def __init__(self, args):
    method forward (line 1325) | def forward(self, x):
    method load_state_dict (line 1369) | def load_state_dict(self, state_dict, strict=False):
  class MatrixModelG2 (line 1395) | class MatrixModelG2(nn.Module):
    method __init__ (line 1396) | def __init__(self, args):
    method forward (line 1474) | def forward(self, x):
    method load_state_dict (line 1520) | def load_state_dict(self, state_dict, strict=False):
  class MatrixModelF2 (line 1546) | class MatrixModelF2(nn.Module):
    method __init__ (line 1547) | def __init__(self, args):
    method forward (line 1640) | def forward(self, x):
    method load_state_dict (line 1694) | def load_state_dict(self, state_dict, strict=False):
  class PAM_Module (line 1720) | class PAM_Module(nn.Module):
    method __init__ (line 1723) | def __init__(self, in_dim):
    method forward (line 1733) | def forward(self, x):
  class CAM_Module (line 1755) | class CAM_Module(nn.Module):
    method __init__ (line 1757) | def __init__(self, in_dim):
    method forward (line 1764) | def forward(self,x):
  class GAM_Module (line 1786) | class GAM_Module(nn.Module):
    method __init__ (line 1788) | def __init__(self, in_dim):
    method forward (line 1795) | def forward(self,x):
  class DAM_Module (line 1818) | class DAM_Module(nn.Module):
    method __init__ (line 1820) | def __init__(self, in_dim):
    method forward (line 1827) | def forward(self,x):
  class MatrixModelH (line 1850) | class MatrixModelH(nn.Module):
    method __init__ (line 1851) | def __init__(self, args):
    method forward (line 1928) | def forward(self, x):
    method load_state_dict (line 1972) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/mdsr.py
  function make_model (line 10) | def make_model(args, parent=False):
  class MDSR (line 13) | class MDSR(nn.Module):
    method __init__ (line 14) | def __init__(self, args, conv=common.default_conv):
    method forward (line 51) | def forward(self, x):
    method set_scale (line 65) | def set_scale(self, scale_idx):

FILE: src/model/ops.py
  class EoctConv (line 10) | class EoctConv(nn.Module):
    method __init__ (line 11) | def __init__(self, in_channels, num_channels, kernel_size=3, stride=1,...
    method forward (line 55) | def forward(self, data):
  function relu (line 126) | def relu(data):
  function sigmoid (line 146) | def sigmoid(data):
  function bn (line 165) | def bn(data, num_channels):
  function max_pool2d (line 182) | def max_pool2d(data, l=(2,2)):
  function avg_pool2d (line 201) | def avg_pool2d(data):
  function dropout (line 221) | def dropout(data, l):
  function dataSum (line 241) | def dataSum(a, b):
  function tupleSum (line 252) | def tupleSum(a,b):
  class MeanShift (line 256) | class MeanShift(nn.Conv2d):
    method __init__ (line 257) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1):
  class _UpsampleBlock (line 266) | class _UpsampleBlock(nn.Module):
    method __init__ (line 267) | def __init__(self,
    method forward (line 288) | def forward(self, x):
  function tupleMultiply (line 302) | def tupleMultiply(a, b):

FILE: src/model/rcan.py
  function make_model (line 7) | def make_model(args, parent=False):
  class CALayer (line 11) | class CALayer(nn.Module):
    method __init__ (line 12) | def __init__(self, channel, reduction=16):
    method forward (line 24) | def forward(self, x):
  class Ada_conv (line 31) | class Ada_conv(nn.Module):
    method __init__ (line 32) | def __init__(self, in_channels, out_channels, kernel_size, bias=True, ...
    method forward (line 45) | def forward(self, x):
  class ResAda_conv (line 56) | class ResAda_conv(nn.Module):
    method __init__ (line 57) | def __init__(self, in_channels, out_channels, kernel_size, bias=True, ...
    method forward (line 70) | def forward(self, x):
  class RCAB (line 81) | class RCAB(nn.Module):
    method __init__ (line 82) | def __init__(
    method forward (line 97) | def forward(self, x):
  class ResidualGroup (line 104) | class ResidualGroup(nn.Module):
    method __init__ (line 105) | def __init__(self, conv, n_feat, kernel_size, reduction, act, res_scal...
    method forward (line 116) | def forward(self, x):
  class RCAN (line 122) | class RCAN(nn.Module):
    method __init__ (line 123) | def __init__(self, args, conv=common.default_conv):
    method forward (line 161) | def forward(self, x):
    method load_state_dict (line 173) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/rcan1.py
  function make_model (line 8) | def make_model(args, parent=False):
  class CALayer (line 12) | class CALayer(nn.Module):
    method __init__ (line 13) | def __init__(self, channel, reduction=16):
    method forward (line 25) | def forward(self, x):
  class Dis (line 30) | class Dis(nn.Module):
    method __init__ (line 31) | def __init__(self, loss_type='L1', batchsize=16):
    method forward (line 39) | def forward(self, x1, x2):
    method L1Loss (line 51) | def L1Loss(self, x1, x2):
    method L2Loss (line 56) | def L2Loss(self, x1, x2):
    method bit_product_sum (line 61) | def bit_product_sum(self, x, y):
    method cosine_similarity (line 65) | def cosine_similarity(self, x, y, norm=True):
  class FullConvRes (line 87) | class FullConvRes(nn.Module):
    method __init__ (line 89) | def __init__(self, out_channels=64, in_channels=64, K=9):
    method forward (line 108) | def forward(self,x):
  class FullConvRes1 (line 189) | class FullConvRes1(nn.Module):
    method __init__ (line 191) | def __init__(self, out_channels=64, in_channels=64, kernel_size=3):
    method forward (line 210) | def forward(self,x):
  class FullConv (line 273) | class FullConv(nn.Module):
    method __init__ (line 275) | def __init__(self, out_channels=64, in_channels=64, kernel_size=3):
    method forward (line 288) | def forward(self,x):
  class RCAB (line 328) | class RCAB(nn.Module):
    method __init__ (line 329) | def __init__(
    method forward (line 343) | def forward(self, x):
  class ResidualGroup (line 350) | class ResidualGroup(nn.Module):
    method __init__ (line 351) | def __init__(self, conv, n_feat, kernel_size, reduction, act, res_scal...
    method forward (line 362) | def forward(self, x):
  class RCAN (line 368) | class RCAN(nn.Module):
    method __init__ (line 369) | def __init__(self, args, conv=common.default_conv):
    method forward (line 409) | def forward(self, x):
    method load_state_dict (line 421) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/rcan3.py
  function make_model (line 9) | def make_model(args, parent=False):
  class CALayer (line 13) | class CALayer(nn.Module):
    method __init__ (line 14) | def __init__(self, channel, reduction=16):
    method forward (line 26) | def forward(self, x):
  class MSCALayer (line 31) | class MSCALayer(nn.Module):
    method __init__ (line 32) | def __init__(self):
  class Dis (line 35) | class Dis(nn.Module):
    method __init__ (line 36) | def __init__(self, loss_type='L1', B=4):
    method forward (line 44) | def forward(self, x1, x2):
    method L1Loss (line 56) | def L1Loss(self, x1, x2):
    method L2Loss (line 61) | def L2Loss(self, x1, x2):
    method bit_product_sum (line 66) | def bit_product_sum(self, x, y):
    method cosine_similarity (line 70) | def cosine_similarity(self, x, y, norm=True):
  class DAM_Module (line 91) | class DAM_Module(nn.Module):
    method __init__ (line 93) | def __init__(self, in_dim):
    method forward (line 100) | def forward(self,x):
  class SEDAM_Module (line 123) | class SEDAM_Module(nn.Module):
    method __init__ (line 125) | def __init__(self, in_dim):
    method forward (line 136) | def forward(self,x):
  class MSAM_Module (line 172) | class MSAM_Module(nn.Module):
    method __init__ (line 174) | def __init__(self, in_dim):
    method forward (line 195) | def forward(self,x):
    method attention (line 223) | def attention(self, x):
    method one_scale (line 228) | def one_scale(self, x, scale=2):
    method multi_scale (line 240) | def multi_scale(self, x):
  class SAM_Module (line 252) | class SAM_Module(nn.Module):
    method __init__ (line 254) | def __init__(self, in_dim):
    method forward (line 271) | def forward(self,x):
    method depixel_shuffle (line 313) | def depixel_shuffle(self, x, upscale_factor=2):
    method squaremax (line 334) | def squaremax(self, x, dim=-1):
    method logmax (line 340) | def logmax(self,x):
    method absmax (line 346) | def absmax(self,x):
  class SECAM_Module (line 352) | class SECAM_Module(nn.Module):
    method __init__ (line 353) | def __init__(self, in_dim):
    method forward (line 365) | def forward(self,x):
  class LAM_Module (line 402) | class LAM_Module(nn.Module):
    method __init__ (line 404) | def __init__(self, in_dim):
    method forward (line 413) | def forward(self,x):
  class GAM_Module (line 458) | class GAM_Module(nn.Module):
    method __init__ (line 461) | def __init__(self, in_dim):
    method forward (line 470) | def forward(self,x):
  class RCAB (line 498) | class RCAB(nn.Module):
    method __init__ (line 499) | def __init__(
    method forward (line 513) | def forward(self, x):
  class ResidualGroup (line 520) | class ResidualGroup(nn.Module):
    method __init__ (line 521) | def __init__(self, conv, n_feat, kernel_size, reduction, act, res_scal...
    method forward (line 532) | def forward(self, x):
  class RCAN (line 538) | class RCAN(nn.Module):
    method __init__ (line 539) | def __init__(self, args, conv=common.default_conv):
    method forward (line 582) | def forward(self, x):
    method load_state_dict (line 614) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/rcan4.py
  function make_model (line 6) | def make_model(args, parent=False):
  class CALayer (line 10) | class CALayer(nn.Module):
    method __init__ (line 11) | def __init__(self, channel, reduction=16):
    method forward (line 23) | def forward(self, x):
  class DAM_Module (line 28) | class DAM_Module(nn.Module):
    method __init__ (line 30) | def __init__(self, in_dim):
    method forward (line 37) | def forward(self,x):
  class GAM_Module (line 60) | class GAM_Module(nn.Module):
    method __init__ (line 63) | def __init__(self, in_dim):
    method forward (line 72) | def forward(self,x):
  class RCAB (line 100) | class RCAB(nn.Module):
    method __init__ (line 101) | def __init__(
    method forward (line 115) | def forward(self, x):
  class ResidualGroup (line 122) | class ResidualGroup(nn.Module):
    method __init__ (line 123) | def __init__(self, conv, n_feat, kernel_size, reduction, act, res_scal...
    method forward (line 133) | def forward(self, x):
  class RCAN (line 139) | class RCAN(nn.Module):
    method __init__ (line 140) | def __init__(self, args, conv=common.default_conv):
    method forward (line 183) | def forward(self, x):
    method load_state_dict (line 220) | def load_state_dict(self, state_dict, strict=False):

FILE: src/model/rdn.py
  function make_model (line 10) | def make_model(args, parent=False):
  class RDB_Conv (line 13) | class RDB_Conv(nn.Module):
    method __init__ (line 14) | def __init__(self, inChannels, growRate, kSize=3):
    method forward (line 23) | def forward(self, x):
  class RDB (line 27) | class RDB(nn.Module):
    method __init__ (line 28) | def __init__(self, growRate0, growRate, nConvLayers, kSize=3):
    method forward (line 42) | def forward(self, x):
  class RDN (line 45) | class RDN(nn.Module):
    method __init__ (line 46) | def __init__(self, args):
    method forward (line 93) | def forward(self, x):

FILE: src/model/rdn1.py
  function make_model (line 10) | def make_model(args, parent=False):
  class RDB_Conv (line 13) | class RDB_Conv(nn.Module):
    method __init__ (line 14) | def __init__(self, inChannels, growRate, kSize=(3,3,3)):
    method forward (line 24) | def forward(self, x):
  class DAM_Module (line 31) | class DAM_Module(nn.Module):
    method __init__ (line 33) | def __init__(self):
    method forward (line 39) | def forward(self,x):
  class RDB (line 62) | class RDB(nn.Module):
    method __init__ (line 63) | def __init__(self, growRate0, growRate, nConvLayers, kSize=3):
    method forward (line 79) | def forward(self, x):
  class RDN (line 84) | class RDN(nn.Module):
    method __init__ (line 85) | def __init__(self, args):
    method forward (line 133) | def forward(self, x):

FILE: src/model/rdn2.py
  function make_model (line 10) | def make_model(args, parent=False):
  class RDB_Conv (line 13) | class RDB_Conv(nn.Module):
    method __init__ (line 14) | def __init__(self, inChannels, growRate, kSize=3):
    method forward (line 24) | def forward(self, x):
  class DAM_Module (line 33) | class DAM_Module(nn.Module):
    method __init__ (line 35) | def __init__(self):
    method forward (line 41) | def forward(self,x):
  class RDB (line 64) | class RDB(nn.Module):
    method __init__ (line 65) | def __init__(self, growRate0, growRate, nConvLayers, kSize=3):
    method forward (line 81) | def forward(self, x):
  class RDN (line 88) | class RDN(nn.Module):
    method __init__ (line 89) | def __init__(self, args):
    method forward (line 137) | def forward(self, x):

FILE: src/model/vdsr.py
  function make_model (line 10) | def make_model(args, parent=False):
  class VDSR (line 13) | class VDSR(nn.Module):
    method __init__ (line 14) | def __init__(self, args, conv=common.default_conv):
    method forward (line 39) | def forward(self, x):

FILE: src/template.py
  function set_template (line 1) | def set_template(args):

FILE: src/trainer.py
  class Trainer (line 12) | class Trainer():
    method __init__ (line 13) | def __init__(self, args, loader, my_model, my_loss, ckp):
    method train (line 30) | def train(self):
    method test (line 78) | def test(self):
    method prepare (line 136) | def prepare(self, *args):
    method terminate (line 144) | def terminate(self):

FILE: src/utility.py
  class timer (line 19) | class timer():
    method __init__ (line 20) | def __init__(self):
    method tic (line 24) | def tic(self):
    method toc (line 27) | def toc(self, restart=False):
    method hold (line 32) | def hold(self):
    method release (line 35) | def release(self):
    method reset (line 41) | def reset(self):
  class checkpoint (line 44) | class checkpoint():
    method __init__ (line 45) | def __init__(self, args):
    method get_path (line 82) | def get_path(self, *subdir):
    method save (line 85) | def save(self, trainer, epoch, is_best=False):
    method add_log (line 94) | def add_log(self, log):
    method write_log (line 97) | def write_log(self, log, refresh=False):
    method done (line 104) | def done(self):
    method plot_psnr (line 107) | def plot_psnr(self, epoch):
    method begin_background (line 126) | def begin_background(self):
    method end_background (line 143) | def end_background(self):
    method save_results (line 148) | def save_results(self, dataset, filename, save_list, scale):
  function quantize (line 161) | def quantize(img, rgb_range):
  function calc_psnr (line 165) | def calc_psnr(sr, hr, scale, rgb_range, dataset=None):
  function make_optimizer (line 183) | def make_optimizer(args, target):

FILE: src/videotester.py
  class VideoTester (line 12) | class VideoTester():
    method __init__ (line 13) | def __init__(self, args, my_model, ckp):
    method test (line 22) | def test(self):
    method prepare (line 65) | def prepare(self, *args):