PyTorch-Deep-Dream icon indicating copy to clipboard operation
PyTorch-Deep-Dream copied to clipboard

Published 20 hours ago • verified publisher icon eriklindernoren

Fix to Use Cuda Tensors

Open 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

ss32 avatar Sep 06 '19 01:09 ss32