style-transfer-quality icon indicating copy to clipboard operation
style-transfer-quality copied to clipboard
verified publisher icon aigagror

pytorch style init needs transfer of tensor to cpu to convert to numpy

Open rvencu opened this issue 4 years ago • 0 comments

I succeeded to run it after changing line 63 to this

gen_hist.append(F.to_pil_image(gen_img.cpu().squeeze(0)))

rvencu avatar May 25 '21 10:05 rvencu