mirror of
https://github.com/zhigang1992/PointRCNN.git
synced 2026-01-12 22:49:40 +08:00
12 lines
189 B
Bash
12 lines
189 B
Bash
#!/bin/bash
|
|
partition=$1
|
|
node=$2
|
|
|
|
gpu_num=1
|
|
|
|
srun --mpi=pmi2 --gres=gpu:${gpu_num} \
|
|
-p $partition -n1 \
|
|
--ntasks-per-node=1 \
|
|
-K -w ${node}\
|
|
python -u train_rcnn.py ${@:3}
|