Amar Saini

Results 2 issues of Amar Saini

This is a great implementation! One small note: On line 151: ``` grid_i = torch.arange(start=0.0, end=N, device=self.device, requires_grad=True,).view(1, -1) ``` I believe you should set the `arange` to be from...

This is a great implementation! One small note: On line 151: ``` grid_i = torch.arange(start=0.0, end=N, device=self.device, requires_grad=True,).view(1, -1) ``` I believe you should set the `arange` to be from...