(self, source, targets)
| 45 | print(f"Resuming training from snapshot at Epoch {self.epochs_run}") |
| 46 | |
| 47 | def _run_batch(self, source, targets): |
| 48 | self.optimizer.zero_grad() |
| 49 | output = self.model(source) |
| 50 | loss = F.cross_entropy(output, targets) |
| 51 | loss.backward() |
| 52 | self.optimizer.step() |
| 53 | |
| 54 | def _run_epoch(self, epoch): |
| 55 | b_sz = len(next(iter(self.train_data))[0]) |