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

Update executor.py

Browse Source
This commit is contained in:
xiaohou 2021-11-18 16:19:42 +08:00 committed by GitHub
parent a9a3fc3da9
commit 2700c9ef9b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kws/utils/executor.py
View File

@ -22,7 +22,7 @@ from kws.model.max_pooling_RHE import max_pooling_RHE_binary_CE
from kws.model.ce import cross_entropy from kws.model.ce import cross_entropy
criterion_dict = {'CE': cross_entropy, criterion_dict = {'CE': cross_entropy,
'max_pooling': max_pooling_loss, 'max_pooling': max_pooling_loss,
'RHE': max_pooling_RHE_binary_CE} 'RHE': max_pooling_RHE_binary_CE}