Update trainer.py

gagan3012 · Jul 2, 2021 · dc6242f · dc6242f
1 parent 58dd23b
commit dc6242f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keytotext/trainer.py b/keytotext/trainer.py
@@ -214,6 +214,7 @@ def validation_step(self, batch, batch_size):
         )
         val_acc = self.val_acc(outputs.logits.argmax(1), labels)
         self.log("val_loss", loss, prog_bar=True, logger=True)
+        self.log(f"val_acc", val_acc, prog_bar=True,logger=True)
         return loss
 
     def test_step(self, batch, batch_size):