mmengine/examples/text_classification

Train a Text Classification Model

Install Dependencies

• Build MMEngine from Source

  git clone https://github.com/open-mmlab/mmengine.git
  cd mmengine
  pip install -e . -v
  

• Install thirty-party libraries

  pip install datasets transformers
  

Run the Example

• Single device training

  python examples/text_classification/train.py
  

• Distributed data parallel training

  tochrun -nnodes 1 -nproc_per_node 8 examples/text_classification/train.py --launcher pytorch