Слияние кода завершено, страница обновится автоматически
CONFIG=$1
PARTITION=$2
NUM_GPU=$3
SRUN="srun -p $PARTITION -n1 --mpi=pmi2 --gres=gpu:$NUM_GPU --ntasks-per-node=1 --cpus-per-task=8 --kill-on-bad-exit=0"
PYTHON="python -u "
echo "[INFO] Partition: $PARTITION, Used GPU Num: $NUM_GPU. "
echo "[INFO] SRUN: $SRUN"
echo "[INFO] PYTHON: $PYTHON"
SCRIPT1="main.py"
PYTHON_SCRIPT1="$PYTHON $SCRIPT1 --config $CONFIG "
echo "$SRUN $PYTHON_SCRIPT1"
$SRUN $PYTHON_SCRIPT1
Вы можете оставить комментарий после Вход в систему
Неприемлемый контент может быть отображен здесь и не будет показан на странице. Вы можете проверить и изменить его с помощью соответствующей функции редактирования.
Если вы подтверждаете, что содержание не содержит непристойной лексики/перенаправления на рекламу/насилия/вульгарной порнографии/нарушений/пиратства/ложного/незначительного или незаконного контента, связанного с национальными законами и предписаниями, вы можете нажать «Отправить» для подачи апелляции, и мы обработаем ее как можно скорее.
Опубликовать ( 0 )