sub_opence.lsf 1011 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#!/bin/bash
# Begin LSF directives
#BSUB -P stf011
#BSUB -J sc21
#BSUB -o logs/sc21.o%J
#BSUB -W 0:30
#BSUB -nnodes 1
#BSUB -alloc_flags "nvme smt4"
####BSUB -N
# End LSF directives and begin shell commands

nnodes=$(cat ${LSB_DJOB_HOSTFILE} | sort | uniq | grep -v login | grep -v batch | wc -l)
DATA_DIR=/gpfs/wolf/gen166/proj-shared/atsaris/imagenet/data/
LOG_DIR=logs/

source /gpfs/wolf/stf011/proj-shared/28t/env/open-ce/activate.sh
#source /gpfs/wolf/stf011/proj-shared/28t/env/ibm-wml/activate.sh

jsrun --smpiargs="-disable_gpu_hooks" -n${nnodes} -a1 -c42 -g1 -r1 \
    --bind=proportional-packed:7 --launch_distribution=packed \
    bash -c "\
    source export_DDP_envvars.sh && \
    python -u ../imagenet/main.py \
    --arch resnet50 \
    -j 28 \
    -p 10 \
    -b 128\
    --training-only \
    --raport-file ${LOG_DIR}/pytorch.opence.1GPU.json \
    --epochs 1 \
    --prof 100 \
    --no-checkpoints \
    --data-backend pytorch \
    --amp \
    --memory-format nhwc \
    ${DATA_DIR}
    "