#!/usr/bin/env bash

SRC=$1
DEST=$2
DATADIR=$3
INIT=$4
EPOCHS=$5
BS=128

for i in `ls -d $SRC*/`
do
    echo "retrain model in ${i}"
    name=$(basename $i)
    python3 main.py --mode retrain --model_src ${i} --model_dest ${DEST}/${name} --init_epoch $INIT --epochs $EPOCHS --batch $BS --train ${DATADIR}

done