flpelerin commited on
Commit
f2ce311
·
1 Parent(s): 4b2f426

Update file trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +2 -2
trainer.py CHANGED
@@ -15,8 +15,8 @@ class Trainer:
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
 
22
  def train(self, batches):
 
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
 
22
  def train(self, batches):