- #!/bin/bash
- DATASET_NAME="CUHK-PEDES"
- CUDA_VISIBLE_DEVICES=0 \
- python train.py \
- --name iira \
- --img_aug \
- --batch_size 64 \
- --MLM \
- --dataset_name $DATASET_NAME \
- --loss_names 'sdm+mlm+id' \
- --root_dir '/mnt/vos-s9gjtkm2/reid/dataset/cross_reid' \
- --num_epoch 60
|