-
Notifications
You must be signed in to change notification settings - Fork 109
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: run bash scripts for train/clean of segan+
- Loading branch information
Showing
2 changed files
with
26 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
#!/bin/bash | ||
|
||
CKPT_PATH="ckpt_segan+" | ||
|
||
# please specify the path to your G model checkpoint | ||
# as in weights_G-EOE_<iter>.ckpt | ||
G_PRETRAINED_CKPT="" | ||
|
||
# please specify the path to your folder containing | ||
# noisy test files, each wav in there will be processed | ||
TEST_FILES_PATH="" | ||
|
||
# please specify the output folder where cleaned files | ||
# will be saved | ||
SAVE_PATH="" | ||
|
||
python -u clean.py --g_pretrained_ckpt $CKPT_PATH/$G_PRETRAINED_CKPT \ | ||
--test_files $TEST_FILES_PATH --cfg_file $CKPT_PATH/train.opts \ | ||
--synthesis_path $SAVE_PATH --soundfile |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,7 @@ | ||
#!/bin/bash | ||
|
||
|
||
python -u train.py --save_path ckpt_segan+ \ | ||
--clean_trainset data_veu4/expanded_segan1_additive/clean_trainset \ | ||
--noisy_trainset data_veu4/expanded_segan1_additive/noisy_trainset \ | ||
--cache_dir data_tmp --no_train_gen --batch_size 300 |