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_16gpu.sh

8 lines
401 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. mpirun --allow-run-as-root -np 16 -mca btl_tcp_if_include enp97s0f0 -x NCCL_SOCKET_IFNAME=enp97s0f0 -x PYTHONPATH=$depsdir/python -H daim117:8,daim118:8 python ${mainpy} --model $1 --dataset $2 --learning-rate 0.000625 --validate --timing --comm-mode AllReduce