Python Registry.add_from_module Examples

Example #1

def _samplers_loader(r: Registry):
    from torch.utils.data import sampler as s
    factories = {
        k: v
        for k, v in s.__dict__.items() if "Sampler" in k and k != "Sampler"
    }
    r.add(**factories)
    from catalyst.data import sampler
    r.add_from_module(sampler)

Example #2

def _schedulers_loader(r: Registry):
    from catalyst.contrib import schedulers as m
    r.add_from_module(m)

Example #3

def _optimizers_loader(r: Registry):
    from catalyst.contrib import optimizers as m
    r.add_from_module(m)

Example #4

def _modules_loader(r: Registry):
    from catalyst.contrib import modules as m
    r.add_from_module(m)

Example #5

def _criterion_loader(r: Registry):
    from catalyst.contrib import criterion as m
    r.add_from_module(m)

Example #6

def _grad_clip_loader(r: Registry):
    from torch.nn.utils import clip_grad as m
    r.add_from_module(m)

Example #7

def _schedulers_loader(r: Registry):
    from torch.optim import lr_scheduler as m
    r.add_from_module(m)