run_irra.sh 207 B

123456789101112
  1. #!/bin/bash
  2. DATASET_NAME="CUHK-PEDES"
  3. CUDA_VISIBLE_DEVICES=0 \
  4. python train.py \
  5. --name iira \
  6. --img_aug \
  7. --batch_size 64 \
  8. --MLM \
  9. --dataset_name $DATASET_NAME \
  10. --loss_names 'sdm+mlm+id' \
  11. --num_epoch 60