Update file trainer.py

trainer.py

@@ -15,8 +15,9 @@ class Trainer:
         args = {'epoch': self.epoch, 'batch': self.batch, 'loss': loss}
         self.wandb(args)
 
-        if self.batch % self.inference.frequency == 0:
-            print(f'{self.model.generate_text(self.inference.seed_text, self.inference.n_predict)}')
+        if self.inference.frequency != 0:
+            if self.batch % self.inference.frequency == 0:
+                print(f'{self.model.generate_text(self.inference.seed_text, self.inference.n_predict)}')
 
 
     def train(self, batches):