Skip to content

Files

Latest commit

45f1185 · Nov 12, 2024

History

History
29 lines (22 loc) · 893 Bytes

TrainEval_pred.md

File metadata and controls

29 lines (22 loc) · 893 Bytes

Training & Evaluation for WOMD prediction

Baseline Model Selection

Choose a baseline model to train through config files in womd/tools/cfg/:

BeTopNet-full: cfg/BeTopNet_full_64.yaml
BeTopNet-e2e: cfg/BeTopNet_e2e_6.yaml
MTR++: cfg/MTR_PlusPlus.yaml
Wayformer: cfg/Wayformer.yaml

Training

To start the training pipeline, run the following command:

cd womd/tools
bash scripts/dist_train.sh N_GPUS --cfg_file cfg/YOUR_CHOSEN_BASELINE.yaml --epoch 30 --batch_size BATCH --extra_tag XXX

[TIPS] If you have not cached the data, your may set BATCH = 10*N_GPUS using A100(80G) gpus. Otherwise, you may leverage a larger batch size.

Evaluation

To evaluate the trained model, use the following command:

cd womd/tools
bash scripts/dist_test.sh N_GPUS --cfg_file cfg/YOUR_CHOSEN_BASELINE.yaml --batch_size BATCH --ckpt YOUR_MODEL_CKPT