|
|
@ -20,8 +20,7 @@ if [ ${seed} != 0 ]; then
|
|
|
|
export FLAGS_cudnn_deterministic=True
|
|
|
|
export FLAGS_cudnn_deterministic=True
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
python3 -u ${BIN_DIR}/train.py \
|
|
|
|
python3 -m paddle.distributed.launch --gpus=${CUDA_VISIBLE_DEVICES} ${BIN_DIR}/train.py \
|
|
|
|
--ngpu ${ngpu} \
|
|
|
|
|
|
|
|
--config ${config_path} \
|
|
|
|
--config ${config_path} \
|
|
|
|
--output exp/${ckpt_name} \
|
|
|
|
--output exp/${ckpt_name} \
|
|
|
|
--model_type ${model_type} \
|
|
|
|
--model_type ${model_type} \
|
|
|
|