Update 2 files

- /model.py
- /trainer.py

model.py

@@ -41,7 +41,6 @@ class Model:
         return lm_loss
 
 
-
     def generate_text(self, seed_text, num_predict):
         max_len = num_predict + len(seed_text)
 

trainer.py

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