GET_STARTED.md

Getting Started

Train with a single GPU

• The default command is as follows:

  python train.py ${CONFIG_FILE}

Train with multiple GPUs

• The default command is as follows:

  bash dist_train.sh ${CONFIG_FILE} ${GPU_NUM}

Test with a single GPU

• The default command is as follows:

  python test.py ${CONFIG_FILE} ${CHECKPOINT_FILE}

Test with multiple GPUs

• The default command is as follows:

  bash dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM}

Note: If you want to test the model with TTA (test time augmentation), you need to add --tta argument, such as:

bash dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} --tta