run_irra.sh 247 B

12345678910111213
  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 16 \
  8. --MLM \
  9. --dataset_name $DATASET_NAME \
  10. --loss_names 'sdm+mlm+id' \
  11. --root_dir '/home/linkslinks/dataset' \
  12. --num_epoch 60