Skip to content

Latest commit

 

History

History
8 lines (8 loc) · 238 Bytes

README.md

File metadata and controls

8 lines (8 loc) · 238 Bytes

Prepare dataset & Configuration

  • Modify the config.json under ./static

Training Regressor

Multi-GPU training

export CUDA_VISIBLE_DEVICES=0,1,2...
accelerate launch main.py --gpu_ids="0,1,2..." --multi-gpu