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

Merge branch 'master' of https://github.com/blessyyyu/wenet-kws

Browse Source
This commit is contained in:
blessyyyu 2022-03-23 14:32:04 +08:00
commit c2572d9abf
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kws/model/loss.py
View File

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
import torch import torch
import torch.nn as nn import torch.nn.functional as F
from kws.utils.mask import padding_mask from kws.utils.mask import padding_mask
@ -105,8 +105,7 @@ def cross_entropy(logits: torch.Tensor, target: torch.Tensor):
(float): loss of current batch (float): loss of current batch
(float): accuracy of current batch (float): accuracy of current batch
""" """
cross_entropy = nn.CrossEntropyLoss() loss = F.cross_entropy(logits, target)
loss = cross_entropy(logits, target)
acc = acc_frame(logits, target) acc = acc_frame(logits, target)
return loss, acc return loss, acc