def _samplers_loader(r: hydra_slayer.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)
def _engines_loader(r: hydra_slayer.Registry):
from catalyst.core.engine import IEngine
r.add(IEngine)
from catalyst import engines as m
r.add_from_module(m)
def _callbacks_loader(r: hydra_slayer.Registry):
from catalyst.core.callback import Callback, CallbackWrapper
r.add(Callback)
r.add(CallbackWrapper)
from catalyst import callbacks as m
r.add_from_module(m)
def _runners_loader(r: hydra_slayer.Registry):
from catalyst.core.runner import IRunner
r.add(IRunner)
r.add(IRunner)
from catalyst import runners as m
r.add_from_module(m)
def _dataloaders_loader(r: hydra_slayer.Registry):
from torch.utils.data import DataLoader
r.add(DataLoader)