prepare-masks.sh

TASK=$1
MASK_LIST=(
  10
  10
  10
)

SINGLE_DIR=exp/$TASK/single
MTL_DIR=exp/$TASK/mtl
len=${#MASK_LIST[@]}

mkdir -p $MTL_DIR/mask
for((i=0; i<len; i++)) do
    mask_id=${MASK_LIST[$i]}
    cp -v $SINGLE_DIR/cp/$i/${mask_id}_* $MTL_DIR/mask/${i}_$mask_id
done
cp -v $SINGLE_DIR/cp/0/init_weights.th $MTL_DIR/mask/init_weights