Full training logic
(self)
| 156 | return model |
| 157 | |
| 158 | def train(self): |
| 159 | """ |
| 160 | Full training logic |
| 161 | """ |
| 162 | for epoch in range(self.start_epoch + 1, self.epochs + 1): |
| 163 | self.epoch_result = self._train_epoch(epoch) |
| 164 | self._on_epoch_finish() |
| 165 | if paddle.distributed.get_rank() == 0 and self.visualdl_enable: |
| 166 | self.writer.close() |
| 167 | self._on_train_finish() |
| 168 | |
| 169 | def _train_epoch(self, epoch): |
| 170 | """ |
no test coverage detected