File size: 430 Bytes
08ec965 | 1 2 3 4 5 6 7 8 9 | # Copyright (c) Meta Platforms, Inc. and affiliates.
#!/bin/bash
export DETECTRON2_DATASETS=/path/to/DETECTRON2_DATASETS/
MASTER_NODE=$(scontrol show hostname "$SLURM_NODELIST" | head -n1)
DIST_URL="tcp://$MASTER_NODE:12399"
SOCKET_NAME=$(ip r | grep default | awk '{print $5}')
export GLOO_SOCKET_IFNAME=$SOCKET_NAME
python -u train_net.py --num-gpus 8 --num-machines 1 --machine-rank "$SLURM_NODEID" --dist-url "$DIST_URL" "$@" |