Skip to content

GitLab

Unverified Commit d66ffb4d authored by junxu's avatar junxu Committed by GitHub
Browse files

Remove duplication registry (#1078)

parent bcab2495
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
colossalai/registry/__init__.py
View file @ d66ffb4d
...@@ -5,7 +5,6 @@ import torch.optim as optim ...@@ -5,7 +5,6 @@ import torch.optim as optim
from .registry import Registry from .registry import Registry
LAYERS = Registry("layers", third_party_library=[nn]) LAYERS = Registry("layers", third_party_library=[nn])
LOSSES = Registry("losses")
MODELS = Registry("models") MODELS = Registry("models")
OPTIMIZERS = Registry("optimizers", third_party_library=[optim, dist_optim]) OPTIMIZERS = Registry("optimizers", third_party_library=[optim, dist_optim])
DATASETS = Registry("datasets") DATASETS = Registry("datasets")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment