(self, data_loader, model, optimizer, stage, epoch)
| 96 | self.conf.task_info.hierar_taxonomy, label_map) |
| 97 | |
| 98 | def train(self, data_loader, model, optimizer, stage, epoch): |
| 99 | model.update_lr(optimizer, epoch) |
| 100 | model.train() |
| 101 | return self.run(data_loader, model, optimizer, stage, epoch, |
| 102 | ModeType.TRAIN) |
| 103 | |
| 104 | def eval(self, data_loader, model, optimizer, stage, epoch): |
| 105 | model.eval() |