Update file trainer.py
Browse files- trainer.py +1 -1
trainer.py
CHANGED
|
@@ -15,7 +15,7 @@ class Trainer:
|
|
| 15 |
args = {'epoch': self.epoch, 'batch': self.batch, 'loss': loss}
|
| 16 |
self.wandb(args)
|
| 17 |
|
| 18 |
-
if self.batch %
|
| 19 |
print(f'{self.model.generate_text(self.model.tokenizer, self.inference.seed_text, self.inference.n_predict)}')
|
| 20 |
|
| 21 |
|
|
|
|
| 15 |
args = {'epoch': self.epoch, 'batch': self.batch, 'loss': loss}
|
| 16 |
self.wandb(args)
|
| 17 |
|
| 18 |
+
if self.batch % 200 == 0:
|
| 19 |
print(f'{self.model.generate_text(self.model.tokenizer, self.inference.seed_text, self.inference.n_predict)}')
|
| 20 |
|
| 21 |
|