SYMBOL INDEX (108 symbols across 7 files)

FILE: demosaicnet/dataset.py
  class Dataset (line 38) | class Dataset(TorchDataset):
    method __init__ (line 47) | def __init__(self, root, download=False,
    method __len__ (line 81) | def __len__(self):
    method __getitem__ (line 84) | def __getitem__(self, idx):
  function _download (line 441) | def _download(dst):
  function md5sum (line 517) | def md5sum(filename, blocksize=65536):

FILE: demosaicnet/modules.py
  class BayerDemosaick (line 18) | class BayerDemosaick(nn.Module):
    method __init__ (line 26) | def __init__(self, depth=15, width=64, pretrained=True, pad=False):
    method forward (line 67) | def forward(self, mosaic):
  class XTransDemosaick (line 92) | class XTransDemosaick(nn.Module):
    method __init__ (line 98) | def __init__(self, depth=11, width=64, pretrained=True, pad=False):
    method forward (line 134) | def forward(self, mosaic):
  function _crop_like (line 151) | def _crop_like(src, tgt):

FILE: demosaicnet/mosaic.py
  function bayer (line 9) | def bayer(im, return_mask=False):
  function xtrans_cell (line 58) | def xtrans_cell(torch=False):
  function xtrans (line 89) | def xtrans(im, return_mask=False):

FILE: demosaicnet/utils.py
  function crop_like (line 20) | def crop_like(src, tgt):
  class ExponentialMovingAverage (line 48) | class ExponentialMovingAverage(object):
    method __init__ (line 56) | def __init__(self, keys, alpha=0.999):
    method __getitem__ (line 61) | def __getitem__(self, key):
    method update (line 64) | def update(self, key, value):
  class BasicArgumentParser (line 75) | class BasicArgumentParser(argparse.ArgumentParser):
    method __init__ (line 78) | def __init__(self, *args, **kwargs):
  class ModelInterface (line 115) | class ModelInterface(metaclass=ABCMeta):
    method __init__ (line 118) | def __init__(self):
    method training_step (line 122) | def training_step(self, batch):
    method init_validation (line 137) | def init_validation(self):
    method validation_step (line 149) | def validation_step(self, batch, running_val_data):
    method __repr__ (line 164) | def __repr__(self):
  class Checkpointer (line 168) | class Checkpointer(object):
    method __init__ (line 188) | def __init__(self, root, model=None, meta=None, optimizers=None,
    method __repr__ (line 215) | def __repr__(self):
    method __path (line 218) | def __path(self, path, prefix=None):
    method save (line 223) | def save(self, path, extras=None):
    method try_and_init_from (line 257) | def try_and_init_from(self, path):
    method load (line 286) | def load(self, path):
    method load_latest (line 331) | def load_latest(self):
    method sorted_checkpoints (line 353) | def sorted_checkpoints(self):
    method delete (line 374) | def delete(self, path):
    method load_meta (line 386) | def load_meta(root, prefix=None):
  class Trainer (line 404) | class Trainer(object):
    method __init__ (line 416) | def __init__(self, interface):
    method interrupt_handler (line 426) | def interrupt_handler(self, signo, frame):
    method _stop (line 431) | def _stop(self):
    method add_callback (line 436) | def add_callback(self, callback):
    method train (line 447) | def train(self, dataloader, starting_epoch=None, num_epochs=None,
    method __repr__ (line 500) | def __repr__(self):
    method __training_start (line 504) | def __training_start(self, dataloader):
    method __training_end (line 508) | def __training_end(self):
    method __epoch_start (line 512) | def __epoch_start(self, epoch_idx):
    method __epoch_end (line 516) | def __epoch_end(self):
    method __batch_start (line 520) | def __batch_start(self, batch_idx, batch):
    method __batch_end (line 524) | def __batch_end(self, batch, train_step_data):
    method __val_batch_start (line 528) | def __val_batch_start(self, batch_idx, batch):
    method __val_batch_end (line 532) | def __val_batch_end(self, batch, running_val_data):
    method __validation_start (line 536) | def __validation_start(self, dataloader):
    method __validation_end (line 541) | def __validation_end(self, running_val_data):
    method __training_step (line 545) | def __training_step(self, batch):
    method __validation_step (line 548) | def __validation_step(self, batch, running_val_data):
  class Callback (line 552) | class Callback(object):
    method __repr__ (line 563) | def __repr__(self):
    method __init__ (line 566) | def __init__(self):
    method training_start (line 575) | def training_start(self, dataloader):
    method training_end (line 584) | def training_end(self):
    method epoch_start (line 588) | def epoch_start(self, epoch):
    method epoch_end (line 599) | def epoch_end(self):
    method validation_start (line 607) | def validation_start(self, dataloader):
    method validation_end (line 616) | def validation_end(self, val_data):
    method batch_start (line 620) | def batch_start(self, batch_idx, batch_data):
    method batch_end (line 629) | def batch_end(self, batch_data, train_step_data):
    method val_batch_start (line 639) | def val_batch_start(self, batch_idx, batch_data):
    method val_batch_end (line 648) | def val_batch_end(self, batch_data, running_val_data):
  class CheckpointingCallback (line 658) | class CheckpointingCallback(Callback):
    method __init__ (line 675) | def __init__(self, checkpointer, interval=600,
    method epoch_end (line 685) | def epoch_end(self):
    method training_end (line 692) | def training_end(self):
    method batch_end (line 696) | def batch_end(self, batch_data, train_step_data):
    method __purge_old_files (line 720) | def __purge_old_files(self):
  class KeyedCallback (line 746) | class KeyedCallback(Callback):
    method __init__ (line 758) | def __init__(self, keys=None, val_keys=None, smoothing=0.999):
    method batch_end (line 776) | def batch_end(self, batch_data, train_step_data):
  class ProgressBarCallback (line 780) | class ProgressBarCallback(KeyedCallback):
    method __init__ (line 787) | def __init__(self, keys=None, val_keys=None, smoothing=0.99, label=None):
    method training_start (line 796) | def training_start(self, dataloader):
    method training_end (line 800) | def training_end(self):
    method epoch_start (line 804) | def epoch_start(self, epoch):
    method epoch_end (line 812) | def epoch_end(self):
    method validation_start (line 817) | def validation_start(self, dataloader):
    method val_batch_end (line 823) | def val_batch_end(self, batch, running_val_data):
    method validation_end (line 826) | def validation_end(self, val_data):
    method batch_end (line 836) | def batch_end(self, batch_data, train_step_data):

FILE: scripts/demosaicnet_demo.py
  function main (line 16) | def main(args):

FILE: scripts/eval.py
  class PSNR (line 14) | class PSNR(th.nn.Module):
    method __init__ (line 15) | def __init__(self):
    method forward (line 18) | def forward(self, out, ref):
  function main (line 22) | def main(args):

FILE: scripts/train.py
  class PSNR (line 14) | class PSNR(th.nn.Module):
    method __init__ (line 15) | def __init__(self):
    method forward (line 18) | def forward(self, out, ref):
  class DemosaicnetInterface (line 23) | class DemosaicnetInterface(demosaicnet.utils.ModelInterface):
    method __init__ (line 31) | def __init__(self, model, lr=1e-4, cuda=th.cuda.is_available()):
    method training_step (line 41) | def training_step(self, batch):
    method forward (line 46) | def forward(self, batch):
    method backward (line 52) | def backward(self, batch, fwd_output):
    method init_validation (line 69) | def init_validation(self):
    method update_validation (line 72) | def update_validation(self, batch, fwd_output, running_data):
    method finalize_validation (line 87) | def finalize_validation(self, running_data):
  function main (line 93) | def main(args):