r.add_from_module(p, prefix=["A.", "albu.", "albumentations."])
from catalyst.contrib.data.cv import transforms as t
r.add_from_module(t, prefix=["catalyst.", "C."])
except ImportError as ex:
if settings.albumentations_required:
logger.warning(
"albumentations not available, to install albumentations, "
"run `pip install albumentations`.")
raise ex
TRANSFORMS = Registry("transform")
TRANSFORMS.late_add(_transforms_loader)
Transform = TRANSFORMS.add
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)
Scheduler,
SCHEDULERS,
Transform,
TRANSFORMS,
)
from catalyst.tools.registry import Registry
def _callbacks_loader(r: Registry):
from catalyst.core import callbacks as m
r.add_from_module(m)
CALLBACKS = Registry("callback")
CALLBACKS.late_add(_callbacks_loader)
Callback = CALLBACKS.add
__all__ = [
"Callback",
"Criterion",
"Optimizer",
"Scheduler",
"Module",
"Model",
"Sampler",
"Transform",
"CALLBACKS",
"CRITERIONS",
"GRAD_CLIPPERS",
"MODELS",
