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.
10 lines
187 B
10 lines
187 B
3 years ago
|
#!/bin/bash
|
||
|
|
|
||
|
|
preprocess_path=$1
|
||
|
|
train_output_path=$2
|
||
|
|
|
||
|
|
python3 ${BIN_DIR}/train.py \
|
||
|
|
--data=${preprocess_path} \
|
||
|
|
--output=${train_output_path} \
|
||
|
|
--device="gpu" \
|
||
|
|
--nprocs=1
|