You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
360 B
19 lines
360 B
#!/bin/bash
|
|
set -e
|
|
|
|
source path.sh
|
|
|
|
# prepare data
|
|
bash ./local/data.sh
|
|
|
|
# train model
|
|
CUDA_VISIBLE_DEVICES=0,1,2,3 bash ./local/train.sh
|
|
|
|
# test model
|
|
CUDA_VISIBLE_DEVICES=0 bash ./local/test.sh
|
|
|
|
# infer model
|
|
CUDA_VISIBLE_DEVICES=0 bash ./local/infer.sh ckpt/checkpoints/step-3284
|
|
|
|
# export model
|
|
bash ./local/export.sh ckpt/checkpoints/step-3284 jit.model |