ss32 opened this issue 5 years ago · 0 comments
Currently the code fails if using Cuda. Line 24 should read
- image.data += norm_lr * image.grad.data + image.data += Tensor([norm_lr]) * image.grad.data