hpcaitech / mmdetection-examples

Train mmdetection models with ColossalAI.

mmdetection3d with ColossalAI examples

Requirements

Please install mmdetection3d following mmdetection getting started first.

Then

pip install -r requirements.txt

Run

We provide a shell script to launch training on single-node 4-gpu.

./train_3dssd_zero2_4gpu.sh

This script trains 3dssd model with random generated data using Zero2 implemented by ColossalAI.