from local

This commit is contained in:
dohe0342 2023-02-25 14:51:13 +09:00
parent a7b72c592a
commit cca9987f85
3 changed files with 2 additions and 0 deletions

View File

@ -319,6 +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))
exit()
# nnet_output is (N, T, C) # nnet_output is (N, T, C)
supervision_segments = torch.stack( supervision_segments = torch.stack(