(self)
| 186 | ) |
| 187 | |
| 188 | def before_epoch(self): |
| 189 | logger.info("---> start train epoch{}".format(self.epoch + 1)) |
| 190 | |
| 191 | if self.epoch + 1 == self.max_epoch - self.exp.no_aug_epochs or self.no_aug: |
| 192 | |
| 193 | logger.info("--->No mosaic aug now!") |
| 194 | self.train_loader.close_mosaic() |
| 195 | logger.info("--->Add additional L1 loss now!") |
| 196 | if self.is_distributed: |
| 197 | self.model.module.head.use_l1 = True |
| 198 | else: |
| 199 | self.model.head.use_l1 = True |
| 200 | |
| 201 | self.exp.eval_interval = 1 |
| 202 | if not self.no_aug: |
| 203 | self.save_ckpt(ckpt_name="last_mosaic_epoch") |
| 204 | |
| 205 | def after_epoch(self): |
| 206 | if self.use_model_ema: |
no test coverage detected