for n in range(result.size(0)):
s = "".join([self.id2char[k.item()] for k in result[n]])
log_string(f"test_before {s}")
masked_inplace_autoregression(
model, self.batch_size, result, ar_mask, device=self.device
)
for n in range(result.size(0)):
s = "".join([self.id2char[k.item()] for k in result[n]])
log_string(f"test_before {s}")
masked_inplace_autoregression(
model, self.batch_size, result, ar_mask, device=self.device
)