diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-20 18:09:06 +0100 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-20 18:09:06 +0100 |
commit | 7e8e54e84c63171e748bbf09516fd517e6821ace (patch) | |
tree | 996093f75a5d488dddf7ea1f159ed343a561ef89 /src/tasks/train.sh | |
parent | b0719d84138b6bbe5f04a4982dfca673aea1a368 (diff) |
Inital commit for refactoring to lightning
Diffstat (limited to 'src/tasks/train.sh')
-rwxr-xr-x | src/tasks/train.sh | 68 |
1 files changed, 0 insertions, 68 deletions
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 |