diff --git a/docs/zh_CN/user_guides/finetune.md b/docs/zh_CN/user_guides/finetune.md index acb28d00f..bcbba3da6 100644 --- a/docs/zh_CN/user_guides/finetune.md +++ b/docs/zh_CN/user_guides/finetune.md @@ -41,7 +41,7 @@ _base_ = [ ## 修改模型 -在进行模型微调是,我们通常希望在主干网络(backbone)加载预训练模型,再用我们的数据集训练一个新的分类头(head)。 +在进行模型微调时,我们通常希望在主干网络(backbone)加载预训练模型,再用我们的数据集训练一个新的分类头(head)。 为了在主干网络加载预训练模型,我们需要修改主干网络的初始化设置,使用 `Pretrained` 类型的初始化函数。另外,在初始化设置中,我们使用 `prefix='backbone'` diff --git a/mmcls/datasets/transforms/auto_augment.py b/mmcls/datasets/transforms/auto_augment.py index e289c216e..8e97cb245 100644 --- a/mmcls/datasets/transforms/auto_augment.py +++ b/mmcls/datasets/transforms/auto_augment.py @@ -7,10 +7,9 @@ from typing import List, Optional, Sequence, Tuple, Union import mmcv import numpy as np -from mmcv import BaseTransform, RandomChoice -from mmcv.transforms import Compose +from mmcv.transforms import BaseTransform, Compose, RandomChoice from mmcv.transforms.utils import cache_randomness -from mmengine import is_list_of, is_seq_of +from mmengine.utils import is_list_of, is_seq_of from mmcls.registry import TRANSFORMS