(self)
| 59 | self.model.eval() |
| 60 | |
| 61 | def predict(self): |
| 62 | |
| 63 | test_iter = data_loader.DataLoader(self.args, data_loader.load_dataset(self.args, 'test', shuffle=False), |
| 64 | self.args.batch_size, self.device, shuffle=False, is_test=True) |
| 65 | trainer = build_trainer(self.args, self.device_id, self.model, None) |
| 66 | trainer.predict(test_iter, self.step) |
| 67 | |
| 68 | |
| 69 | if __name__ == '__main__': |
no test coverage detected