Update file trainer.py

trainer.py

@@ -12,6 +12,9 @@ class Trainer:
     def log(self, loss: float):
         print(f"Epoch: {self.epoch} / {self.num_epochs}\t\tBatch: {self.batch} / {self.num_batches}\t\tLoss: {round(loss, 4)}")
         
+        args = {'epoch': self.epoch, 'batch': self.batch, 'loss': self.loss}
+        wandb(args)
+
         if self.batch % 20 == 0:
             print(f'{self.model.generate_text(self.model.tokenizer, self.inference.seed_text, self.inference.n_predict)}')