identify pretrain model path

microsoft · Aug 24, 2021 · c45a9b8 · c45a9b8
1 parent 716b5b6
commit c45a9b8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -1,5 +1,5 @@
 python train.py \
-  --device 'cpu' \
+  --device '0,1' \
   --stride 1024 \
   --model_config 'config/model_config_small.json' \
   --model_dir 'model' \