from local

This commit is contained in:
dohe0342 2023-02-25 14:52:08 +09:00
parent cca9987f85
commit f10c8c3433
2 changed files with 2 additions and 2 deletions

View File

@ -319,8 +319,8 @@ def decode_one_batch(
supervisions = batch["supervisions"] supervisions = batch["supervisions"]
nnet_output, memory, memory_key_padding_mask = model(feature, supervisions) nnet_output, memory, memory_key_padding_mask = model(feature, supervisions)
print(type(nnet_output)) if type(nnet_output) == tuple:
exit() nnet_output = nnet_output[0]
# nnet_output is (N, T, C) # nnet_output is (N, T, C)
supervision_segments = torch.stack( supervision_segments = torch.stack(