2020-10-13 13:13:24 +08:00
|
|
|
export CUDA_VISIBLE_DEVICES='0,1,2,3,4,5,6,7'
|
|
|
|
|
export HOROVOD_GPU_ALLREDUCE=NCCL
|
|
|
|
|
export HOROVOD_GPU_ALLGATHER=NCCL
|
|
|
|
|
export HOROVOD_GPU_BROADCAST=NCLL
|
|
|
|
|
export MXNET_CPU_WORKER_NTHREADS=3
|
|
|
|
|
|
|
|
|
|
# use `which python` to get the absolute path of your python interpreter
|
|
|
|
|
#
|
2021-01-02 15:57:12 +08:00
|
|
|
PYTHON_EXEC=/usr/bin/python
|
2020-10-13 13:13:24 +08:00
|
|
|
${PYTHON_EXEC} train_memory.py \
|
|
|
|
|
--dataset glint360k_8GPU \
|
2020-10-26 14:15:16 +08:00
|
|
|
--loss cosface \
|
2020-10-13 13:13:24 +08:00
|
|
|
--network r100 \
|
2021-01-02 15:57:12 +08:00
|
|
|
--models-root /data/anxiang/opensource/glint360k_8GPU_r100FC_1.0_fp32_cosface
|
