-
Notifications
You must be signed in to change notification settings - Fork 3
/
_create_json_vb.py
80 lines (73 loc) · 3.31 KB
/
_create_json_vb.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Copyright (c) 2022 by Inria
Authoried by Xiaoyu BIE ([email protected])
License agreement in LICENSE.txt
"""
import os
import json
import soundfile as sf
"""
write launch script for VoiceBank
"""
saved_model = 'ALL_2022-01-02-17h48_SRNN-ss_z_dim=16'
state_dict_file = 'SRNN_final_epoch225.pt'
cpu_type = 'cpu_med'
# cpu_type = 'cpu_prod'
lr = '1e-3'
iter_num = 100
speakers = ['232', '257']
for spk in speakers:
for i in range(40):
lines = ['#!/bin/bash\n',
'#SBATCH --job-name=p{}_{}\n'.format(spk, i),
'#SBATCH --output=/workdir/biex/saved_models_se/log_se_vb/se_vb_{}_{}.stdout\n'.format(spk, i),
'#SBATCH --error=/workdir/biex/saved_models_se/log_se_vb/se_vb_{}_{}.stderr\n'.format(spk, i),
'#SBATCH [email protected]\n',
'#SBATCH --mail-type=ALL\n',
'#SBATCH --nodes=1\n',
'#SBATCH --ntasks=1\n',
'#SBATCH --cpus-per-task=4\n',
'#SBATCH --partition={}\n'.format(cpu_type),
'#SBATCH --mem=16G\n',
'#SBATCH --time=4:00:00\n',
'\n\n',
'# load module\n',
'module purge\n',
'module load anaconda3/2021.05/gcc-9.2.0\n',
'source activate dvae\n',
'\n\n',
'# submit job, w/0 pre-train\n',
'python /gpfs/users/biex/GitPublish/dvae-se/test_voicebank.py \\\n',
' --exp_name p{}_{} \\\n'.format(spk, i),
' --saved_model /workdir/biex/saved_models_se/{} \\\n'.format(saved_model),
' --state_dict_file {} \\\n'.format(state_dict_file),
' --json_file /workdir/biex/data/VoiceBankDemand/json_acc/p{}_sub{}.json \\\n'.format(spk, i),
' --mix_dir /workdir/biex/data/VoiceBankDemand/noisy_testset_wav_16k \\\n',
' --clean_dir /workdir/biex/data/VoiceBankDemand/clean_testset_wav_16k \\\n',
' --ckpt_dir /workdir/biex/saved_models_se/{}/ckpt_vb_iter{}_lr{} \\\n'.format(saved_model, iter_num, lr),
' --enhance_dir /workdir/biex/saved_models_se/{}/ckpt_vb_iter{}_lr{}/enhance_wav \\\n'.format(saved_model, iter_num, lr),
' --log_type 1 \\\n',
' --niter {} \\\n'.format(iter_num),
' --nmf_rank 8 \\\n',
' --nepochs_E_step 1 \\\n',
' --nsamples_E_step 1 \\\n',
' --nsamples_WF 1 \\\n',
' --lr {} \\\n'.format(lr),
' --device cpu']
filename = './script_vb/ruche_test_vb_p{}_sub{}.sh'.format(spk, i)
with open(filename, 'w') as f:
f.writelines(lines)
os.chmod(filename, 0o777)
launch_list = []
for spk in speakers:
for i in range(40):
launch_list.append(f'sbatch ./script_vb/ruche_test_vb_p{spk}_sub{i}.sh\n')
launch_file = './ruche_test_vb.sh'
with open(launch_file, 'w') as f:
f.writelines(launch_list)
os.chmod(launch_file, 0o777)
enhance_dir = f'/workdir/biex/saved_models_se/{saved_model}/ckpt_vb_iter{iter_num}_lr{lr}/enhance_wav'
if not os.path.isdir(enhance_dir):
os.makedirs(enhance_dir)