mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-24 06:29:09 +00:00
Remove duplication registry (#1078)
This commit is contained in:
parent
bcab249565
commit
d66ffb4df4
@ -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")
|
||||
|
Loading…
Reference in New Issue
Block a user