Skip to content
Snippets Groups Projects
Commit 0c6085d9 authored by eca1g19's avatar eca1g19
Browse files

fixed string typo

parent 24a0d00f
Branches
Tags
No related merge requests found
...@@ -65,7 +65,7 @@ class BertLightning(LightningModule): ...@@ -65,7 +65,7 @@ class BertLightning(LightningModule):
decoded_outputs = self.tokenizer.batch_decode(outputs, skip_special_tokens=True) decoded_outputs = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)
scores = rouge_score(decoded_outputs, decoded_lables) scores = rouge_score(decoded_outputs, decoded_lables)
for key, value in scores.items(): for key, value in scores.items():
self.log(f'test_greedy_{key}', value) self.log(f'val_greedy_{key}', value)
if self.beam_search_decode: if self.beam_search_decode:
beams, beam_scores, beam_lengths = beam_search_bert(outputs, length_penalty_enabled=False, beam_size=10) beams, beam_scores, beam_lengths = beam_search_bert(outputs, length_penalty_enabled=False, beam_size=10)
best_beam_indices = torch.argmax(beam_scores, dim=-1) best_beam_indices = torch.argmax(beam_scores, dim=-1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment