diff --git a/docs/en/tutorials/customize_models.md b/docs/en/tutorials/customize_models.md index f637fd6f0..93ce8f04e 100644 --- a/docs/en/tutorials/customize_models.md +++ b/docs/en/tutorials/customize_models.md @@ -85,7 +85,7 @@ Here we show how to develop new components with an example of MobileNet. ```python import torch.nn as nn -from ..registry import BACKBONES +from ..builder import BACKBONES @BACKBONES.register_module diff --git a/docs/zh_cn/tutorials/customize_models.md b/docs/zh_cn/tutorials/customize_models.md index c92d7db23..69fb09b4b 100644 --- a/docs/zh_cn/tutorials/customize_models.md +++ b/docs/zh_cn/tutorials/customize_models.md @@ -83,7 +83,7 @@ MMSegmentation 里主要有2种组件: ```python import torch.nn as nn -from ..registry import BACKBONES +from ..builder import BACKBONES @BACKBONES.register_module