#!/usr/bin/env bash

export FLAGS_cudnn_exhaustive_search=true 
export FLAGS_fraction_of_gpu_memory_to_use=0.96
export FLAGS_eager_delete_tensor_gb=0.0
selected_gpus="0,1,2,3,4,5,6,7"
#selected_gpus="4,5,6"

python -m paddle.distributed.launch \
  --selected_gpus $selected_gpus \
  --log_dir mylog \
  do_train.py \
  --model=ResNet_ARCFACE50 \
  --loss_type=dist_softmax \
  --model_save_dir=output \
  --margin=0.5 \
  --train_batch_size 32 \
  --class_dim 85742 \
  --with_test=True