16 lines
310 B
Bash
16 lines
310 B
Bash
|
#!/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
|