From 7e8e54e84c63171e748bbf09516fd517e6821ace Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 20 Mar 2021 18:09:06 +0100 Subject: Inital commit for refactoring to lightning --- src/tasks/train.sh | 68 ------------------------------------------------------ 1 file changed, 68 deletions(-) delete mode 100755 src/tasks/train.sh (limited to 'src/tasks/train.sh') diff --git a/src/tasks/train.sh b/src/tasks/train.sh deleted file mode 100755 index 60cbd23..0000000 --- a/src/tasks/train.sh +++ /dev/null @@ -1,68 +0,0 @@ -#!/bin/bash - - -# Add checkpoint and resume experiment -usage() { - cat << EOF - usage: ./tasks/train_crnn_line_ctc_model.sh - -f | --experiment_config Name of the experiment config. - -c | --checkpoint (Optional) The experiment name to continue from. - -p | --pretrained_weights (Optional) Path to pretrained weights. - -n | --notrain (Optional) Evaluates a trained model. - -t | --test (Optional) If set, evaluates the model on test set. - -v | --verbose (Optional) Sets the verbosity. - -h | --help Shows this message. -EOF -exit 1 -} - -experiment_config="" -checkpoint="" -pretrained_weights="" -notrain="" -test="" -verbose="" -train_command="" - -while getopts 'f:c:p:nthv' flag; do - case "${flag}" in - f) experiment_config="${OPTARG}" ;; - c) checkpoint="${OPTARG}" ;; - p) pretrained_weights="${OPTARG}" ;; - n) notrain="--notrain" ;; - t) test="--test" ;; - v) verbose="${verbose}v" ;; - h) usage ;; - *) error "Unexpected option ${flag}" ;; - esac -done - - -if [ -z ${experiment_config} ]; -then - echo "experiment_config not specified!" - usage - exit 1 -fi - -experiments_filename="training/experiments/${experiment_config}" -train_command=$(bash tasks/prepare_experiments.sh $experiments_filename) - -if [ ${checkpoint} ]; -then - train_command="${train_command} --checkpoint $checkpoint" -fi - -if [ ${pretrained_weights} ]; -then - train_command="${train_command} --pretrained_weights $pretrained_weights" -fi - -if [ ${verbose} ]; -then - train_command="${train_command} -$verbose" -fi - -train_command="${train_command} $test $notrain" -echo $train_command -eval $train_command -- cgit v1.2.3-70-g09d2