Skip to content
Snippets Groups Projects
Select Git revision
  • master default
  • nikhil_rayaprolu-branch
  • submission-v1.2
  • submission-v1.1
  • submission-vnikhil_rayaprolu2
  • submission-vnikhil_rayaprolu1
  • submission-v0.1
  • submission-scaling
  • submission-scaling-v2
9 results
Blame Permalink
Forked from nikhil_rayaprolu / food-round2
2 commits ahead of the upstream repository.
dist_test.sh 257 B
1
2
3
4
5
6
7
8
9
10
11
12 






#!/usr/bin/env bash

PYTHON=${PYTHON:-"python"}

CONFIG=$1
CHECKPOINT=$2
GPUS=$3
PORT=${PORT:-29500}

$PYTHON -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
    $(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}