Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
8 lines
268 B
Bash
Executable File
8 lines
268 B
Bash
Executable File
export CUDA_VISIBLE_DEVICES=0,1,2,3
|
|
export NGPUS=4
|
|
export OMP_NUM_THREADS=1 # you can change this value according to your number of cpu cores
|
|
|
|
|
|
python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py configs/culane.py
|
|
# python train.py configs/tusimple.py
|