(self)
| 204 | |
| 205 | @th.no_grad() |
| 206 | def test(self): |
| 207 | self.test_lst = th.tensor(self.test_lst).long().to(self.device) |
| 208 | test_desc = self.val(self.test_lst, prefix="test") |
| 209 | test_desc["train_time"] = self.train_time |
| 210 | test_desc["model_param"] = self.model_param |
| 211 | return test_desc |
| 212 | |
| 213 | |
| 214 | def main(dataset, times): |