
#!/bin/bash

####################################################
# Stage I: train with identity loss
####################################################


epochs=21
lr=1e-2
save_path="path to save models"
model="LightCNN-9" # or "LightCNN-29"
img_root_R='path to vis dataset'
train_list_R='path to vis data list'
img_root_F='path to generated nirvis dataset'
train_list_F='path to nirvis data list'


prefix=''
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python train_112_cosface_imglist.py \
      --img_root_R=$img_root_R --train_list_R=$train_list_R \
      --img_root_F=$img_root_F --train_list_F=$train_list_F \
      --save_path=$save_path \
      --model=$model\
      --lr=$lr\
      -b=512 \
      --wd=1e-4 \
      --input_type='grey'\
      --prefix=$prefix\
      --epochs=$epochs | tee $save_path'train_'$prefix'.log'