diff --git a/colossalai/registry/__init__.py b/colossalai/registry/__init__.py index b4e48badf..b48d9e7d2 100644 --- a/colossalai/registry/__init__.py +++ b/colossalai/registry/__init__.py @@ -5,7 +5,6 @@ import torch.optim as optim from .registry import Registry LAYERS = Registry("layers", third_party_library=[nn]) -LOSSES = Registry("losses") MODELS = Registry("models") OPTIMIZERS = Registry("optimizers", third_party_library=[optim, dist_optim]) DATASETS = Registry("datasets")