PKU-DAIR
/
Hetu
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 6df869f592
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '6df869f592'
${ noResults }
Hetu/examples/cnn/scripts/hetu_8gpu.sh

9 lines
327 B
Raw Blame History 

  1. #!/bin/bash

  2. 

  3. workdir=$(cd $(dirname $0); pwd)

  4. mainpy=${workdir}/../main.py

  5. depsdir=${workdir}/../../..

  6. echo $depsdir

  7. ### validate and timing

  8. NCCL_DEBUG=INFO mpirun --allow-run-as-root -np 8 -x PYTHONPATH=$depsdir/python python ${mainpy} --model $1 --dataset $2 --learning-rate 0.00125 --validate --timing --comm-mode AllReduce