(self)
| 170 | self.model.eval() |
| 171 | |
| 172 | def prepare_data(self): |
| 173 | tmp = self.tmp |
| 174 | tmp.input_var = dict() |
| 175 | |
| 176 | for k,v in tmp.input.items(): |
| 177 | if not isinstance(v, list): |
| 178 | tmp.input_var[k] = torch.autograd.Variable(v.cuda()) |
| 179 | |
| 180 | def _set_randomseed(self, seed): |
| 181 | random.seed(seed) |