(self, data_loader, model, optimizer, stage, epoch)
| 102 | ModeType.TRAIN) |
| 103 | |
| 104 | def eval(self, data_loader, model, optimizer, stage, epoch): |
| 105 | model.eval() |
| 106 | return self.run(data_loader, model, optimizer, stage, epoch) |
| 107 | |
| 108 | def run(self, data_loader, model, optimizer, stage, |
| 109 | epoch, mode=ModeType.EVAL): |