Shortcuts

Source code for mmpretrain.utils.misc

# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn
from mmengine.model import is_model_wrapper


[docs]def get_ori_model(model: nn.Module) -> nn.Module: """Get original model if the input model is a model wrapper. Args: model (nn.Module): A model may be a model wrapper. Returns: nn.Module: The model without model wrapper. """ if is_model_wrapper(model): return model.module else: return model
Read the Docs v: dev
Versions
latest
stable
mmcls-1.x
mmcls-0.x
dev
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.