Commit 8a8d5453 authored by Torge Berckmann's avatar Torge Berckmann
Browse files

Show where to save sample representations

parent db1c3eeb
...@@ -560,6 +560,9 @@ class TransformerMultiInputDecoder(FairseqDecoder): ...@@ -560,6 +560,9 @@ class TransformerMultiInputDecoder(FairseqDecoder):
) )
if self.run_corr_arr: if self.run_corr_arr:
#torch.save(rst[0][1]['inner_states'], "rst01inner.pt")
#torch.save(rst[1][1]['inner_states'], "rst11inner.pt")
#exit(1)
# Compare internal decoder states resulting from text and speech inputs # Compare internal decoder states resulting from text and speech inputs
# store intermediate correlation calculation resulting from the comparison # store intermediate correlation calculation resulting from the comparison
self.update_corr(rst[0][1]['inner_states'], rst[1][1]['inner_states']) self.update_corr(rst[0][1]['inner_states'], rst[1][1]['inner_states'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment