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):
)
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
# store intermediate correlation calculation resulting from the comparison
self.update_corr(rst[0][1]['inner_states'], rst[1][1]['inner_states'])
......
Supports Markdown
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