1#!/bin/bash 2 3 4case $# in 5 3) FOLDER=$1; MODEL=$2; PYTHON=$3;; 6 *) echo "update_output_folder.sh folder model python"; exit;; 7esac 8 9 10SCRIPTFOLDER=$(dirname "$0") 11 12 13# update setup 14echo "updating $FOLDER/setup.py..." 15$PYTHON $SCRIPTFOLDER/update_setups.py $FOLDER/setup.yml --model $MODEL 16 17# update checkpoints 18for fn in $(find $FOLDER -type f -name "checkpoint*.pth") 19do 20 echo "updating $fn..." 21 $PYTHON $SCRIPTFOLDER/update_checkpoints.py $fn --model $MODEL 22done