from local

This commit is contained in:
dohe0342 2023-02-25 14:55:04 +09:00
parent f13cf45283
commit a0d62ce58d
3 changed files with 2 additions and 0 deletions

View File

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