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.
17 lines
316 B
17 lines
316 B
#!/bin/bash
|
|
|
|
source path.sh
|
|
# only demos
|
|
|
|
# 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 ckpt/checkpoints/step-3284
|
|
|
|
# infer model
|
|
CUDA_VISIBLE_DEVICES=0 bash ./local/infer.sh ckpt/checkpoints/step-3284
|