AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
Jingyong Hou 2021-11-11 09:30:37 +08:00
parent 0942092426
commit 3326c6d37f
5 changed files with 4 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/hi_xiaowen/s0/conf/mdtc.yaml
View File

@ -1,15 +1,10 @@
dataset_conf: dataset_conf:
filter_conf: filter_conf:
max_length: 2048 max_length: 2048
min_length: 0 min_length: 0
resample_conf: resample_conf:
resample_rate: 16000 resample_rate: 16000
speed_perturb: false speed_perturb: false
feature_extraction_conf: feature_extraction_conf:
feature_type: 'mfcc' feature_type: 'mfcc'
num_ceps: 80 num_ceps: 80
@ -17,17 +12,13 @@ dataset_conf:
frame_shift: 10 frame_shift: 10
frame_length: 25 frame_length: 25
dither: 1.0 dither: 1.0
feature_dither: 0.0 feature_dither: 0.0
spec_aug: true spec_aug: true
spec_aug_conf: spec_aug_conf:
#warp_for_time: false
num_t_mask: 1 num_t_mask: 1
num_f_mask: 1 num_f_mask: 1
max_t: 20 max_t: 20
max_f: 40 max_f: 40
shuffle: true shuffle: true
shuffle_conf: shuffle_conf:
shuffle_size: 1500 shuffle_size: 1500
@ -35,8 +26,6 @@ dataset_conf:
batch_size: 100 batch_size: 100
model: model:
input_dim: 80
output_dim: 2
hidden_dim: 64 hidden_dim: 64
preprocessing: preprocessing:
type: none type: none
@ -50,7 +39,6 @@ model:
optim: adam optim: adam
optim_conf: optim_conf:
lr: 0.001 lr: 0.001
#warmup_step: 2000
training_config: training_config:
grad_clip: 5 grad_clip: 5

6
kws/bin/train.py
View File

@ -161,9 +161,9 @@ def main():
# !!!IMPORTANT!!! # !!!IMPORTANT!!!
# Try to export the model by script, if fails, we should refine # Try to export the model by script, if fails, we should refine
# the code to satisfy the script export requirements # the code to satisfy the script export requirements
#if args.rank == 0: # if args.rank == 0:
#script_model = torch.jit.script(model) # script_model = torch.jit.script(model)
#script_model.save(os.path.join(args.model_dir, 'init.zip')) # script_model.save(os.path.join(args.model_dir, 'init.zip'))
executor = Executor() executor = Executor()
# If specify checkpoint, load some info from checkpoint # If specify checkpoint, load some info from checkpoint
if args.checkpoint is not None: if args.checkpoint is not None:

1
kws/model/loss.py
View File

@ -13,7 +13,6 @@
# limitations under the License. # limitations under the License.
import torch import torch
import torch.nn.functional as F
from kws.utils.mask import padding_mask from kws.utils.mask import padding_mask

2
kws/model/mdtc.py
View File

@ -13,8 +13,6 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F

2
kws/model/tcn.py
View File

@ -13,7 +13,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from typing import List, Optional from typing import Optional
import torch import torch
import torch.nn as nn import torch.nn as nn