from local
This commit is contained in:
parent
a0d62ce58d
commit
e006c770a3
Binary file not shown.
@ -321,8 +321,6 @@ def decode_one_batch(
|
|||||||
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:
|
if type(nnet_output) == tuple:
|
||||||
nnet_output = nnet_output[0]
|
nnet_output = nnet_output[0]
|
||||||
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(
|
||||||
|
|||||||
Reference in New Issue
Block a user