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...