- s = "".join([self.id2char[k.item()] for k in result[n]])
- log_string(f"test_after {s}")
- s = "".join([self.id2char[k.item()] for k in correct[n]])
- log_string(f"correct {s}")
+ comment="GOOD" if (result[n]-input[n]).abs().max()==0 else ""
+ log_string(f"test_after {self.seq2str(result[n])} {comment}")
+ log_string(f"correct {self.seq2str(correct[n])}")