Skip to content

Commit

Permalink
feat: run bash scripts for train/clean of segan+
Browse files Browse the repository at this point in the history
  • Loading branch information
santi-pdp committed Nov 13, 2018
1 parent 14281e6 commit f180519
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 0 deletions.
19 changes: 19 additions & 0 deletions run_segan+_clean.sh
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
7 changes: 7 additions & 0 deletions run_segan+_train.sh
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

0 comments on commit f180519

Please sign in to comment.