A bug in example generator
" self.size = size\n",
" self.t_min, self.t_max = t_min, t_max\n",
" if method=='uniform':\n",
- " self.get_examples = lambda: torch.rand(self.size, requires_grad=True)*(self.t_max-self.t_min) + self.t_min\n",
+ " self.examples = torch.zeros(self.size, requires_grad=True)\n",
+ " self.get_examples = lambda: self.examples + torch.rand(self.size)*(self.t_max-self.t_min) + self.t_min\n",
" elif method=='equally-spaced':\n",
" self.examples = torch.linspace(self.t_min, self.t_max, self.size, requires_grad=True)\n",
" self.get_examples = lambda: self.examples\n",
