for s, prob_s in zip(subjects, prob_subjects):
if s in s_done:
continue
else:
s_done.add(s)
and this:
for o, prob_o in zip(objects, prob_objects):
if o in o_done:
continue
else:
o_done.add(o)
does not return a true for the if-statement. because e.g., tensor(53) is not equal to tensor(53) - these are different hashs.
Instead, you'd have to modify it to
for s, prob_s in zip(subjects, prob_subjects):
if s.item() in s_done:
continue
else:
s_done.add(s.item())
This (l.232 ff):
and this:
does not return a true for the if-statement. because e.g., tensor(53) is not equal to tensor(53) - these are different hashs. Instead, you'd have to modify it to
(and same for objects)