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

[fix] fix training and export error (#81)

Browse Source
This commit is contained in:
Binbin Zhang 2022-08-28 16:49:24 +08:00 committed by GitHub
parent 50354a38e0
commit 490a474d4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
wekws/bin/export_onnx.py
View File

@ -17,6 +17,7 @@ import argparse
import torch import torch
import yaml import yaml
import onnx
import onnxruntime as ort import onnxruntime as ort
from wekws.model.kws_model import init_model from wekws.model.kws_model import init_model

2
wekws/model/kws_model.py
View File

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
import sys import sys
from typing import Optional from typing import Optional, Tuple
import torch import torch
import torch.nn as nn import torch.nn as nn