From d66ffb4df40fe18bebc7284bda1311677f284d1d Mon Sep 17 00:00:00 2001 From: junxu Date: Wed, 8 Jun 2022 07:47:24 +0800 Subject: [PATCH] Remove duplication registry (#1078) --- colossalai/registry/__init__.py | 1 - 1 file changed, 1 deletion(-) 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")