GitHub
Colab Tutorials
Train and inference with shell commands
Train and inference with Python APIs
Docs
MMEngine
MMCV
MMPreTrain
MMDetection
MMSegmentation
MMDeploy
MMRotate
VBTI-Development
OneDL
VBTI
GitHub
OpenMMLab

Table of Contents

Get Started

Prerequisites
Installation

User Guides

Learn about Configs
Prepare Dataset
Inference with existing models
Train
Test
Downstream tasks

Advanced Guides

Adding New Dataset
Customize Data Pipeline
Customize Models
Customize Training Schedule
Customize Runtime Settings
Customize Evaluation Metrics
Convention in MMPretrain

Model Zoo

Visualization

Dataset Visualization
Hyper-parameter Scheduler Visualization
Class Activation Map (CAM) Visualization
t-Distributed Stochastic Neighbor Embedding (t-SNE) Visualization

Analysis Tools

How to Get the Complete Config
Verify Dataset
Log and Results Analysis
Model Complexity Analysis
Confusion Matrix
Shape Bias Tool Usage

Deployment

Torchserve Deployment

Migration

Migration
General change of config
Migration from MMClassification 0.x
Migration from MMSelfSup 0.x

API Reference

mmpretrain.apis
mmpretrain.engine
mmpretrain.datasets
Data Process
mmpretrain.models
mmpretrain.structures
mmpretrain.visualization
mmpretrain.evaluation
mmpretrain.utils

Notes

Contributing to MMPreTrain
Projects based on MMPretrain
Changelog (MMPreTrain)
Changelog (MMClassification)
Frequently Asked Questions
How to Pretrain with Custom Dataset
How to Fine-tune with Custom Dataset

Device Support

NPU (HUAWEI Ascend)

Docs >
mmpretrain.models >
mmpretrain.models.build_head

Shortcuts

mmpretrain.models.build_head¶

mmpretrain.models.build_head(cfg)[source]¶: Build head.

Next

© Copyright 2025, VBTI.

Built with Sphinx using a theme provided by Read the Docs.

mmpretrain.models.build_head
- build_head()

GitHub
Colab Tutorials

Train and inference with shell commands
Train and inference with Python APIs