(self, seed)
| 178 | tmp.input_var[k] = torch.autograd.Variable(v.cuda()) |
| 179 | |
| 180 | def _set_randomseed(self, seed): |
| 181 | random.seed(seed) |
| 182 | np.random.seed(seed) |
| 183 | torch.manual_seed(seed) |
| 184 | torch.cuda.manual_seed(seed) |
| 185 | torch.cuda.manual_seed_all(seed) |