From c45a9b8ea4b116c382f48fe094fd2718f03fa83b Mon Sep 17 00:00:00 2001
From: lxueaa <kungfu.big.head@hotmail.com>
Date: Tue, 24 Aug 2021 16:11:10 +0800
Subject: [PATCH] identify pretrain model path

---
 deeprapper/generate.sh | 2 +-
 deeprapper/train.sh    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/deeprapper/generate.sh b/deeprapper/generate.sh
index 310303c0..a80d4aec 100644
--- a/deeprapper/generate.sh
+++ b/deeprapper/generate.sh
@@ -10,7 +10,7 @@ python generate.py \
                       --save_samples \
                       --save_samples_dir 'samples_save_dir' \
                       --samples_sign 'sample_name' \
-                      --model_dir 'model/lyrics/lyrics_samples_reverse/samples' \
+                      --model_dir 'model/deeprapper-model' \
                       --model_config 'config.json' \
                       --enable_final \
                       --enable_sentence \
diff --git a/deeprapper/train.sh b/deeprapper/train.sh
index cdef87ea..efc229d0 100644
--- a/deeprapper/train.sh
+++ b/deeprapper/train.sh
@@ -1,5 +1,5 @@
 python train.py \
-  --device 'cpu' \
+  --device '0,1' \
   --stride 1024 \
   --model_config 'config/model_config_small.json' \
   --model_dir 'model' \