Update file trainer.py

trainer.py

@@ -13,7 +13,7 @@ class Trainer:
         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': loss}
-        wandb(args)
+        self.wandb(args)
 
         if self.batch % 20 == 0:
             print(f'{self.model.generate_text(self.model.tokenizer, self.inference.seed_text, self.inference.n_predict)}')