from local

This commit is contained in:
dohe0342 2023-02-25 15:15:42 +09:00
parent b565274274
commit ed44167dab
2 changed files with 0 additions and 5 deletions

View File

@ -319,11 +319,6 @@ 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)
if type(nnet_output) == tuple:
nnet_output = nnet_output[0]
# nnet_output is (N, T, C)
print(nnet_output.size())
exit()
supervision_segments = torch.stack( supervision_segments = torch.stack(
( (