search

rosinality / swapping-autoencoder-pytorch

Unofficial implementation of Swapping Autoencoder for Deep Image Manipulation (https://arxiv.org/abs/2007.00653) in PyTorch
Other
256 stars 47 forks source link

Update train.py #20

Closed UdonDa closed 3 years ago

UdonDa commented 3 years ago

To avoid the error, RuntimeError: No grad accumulator for a saved leaf!, I guess that the grad flag of real_img must be changed to False.

Actually, I encounter the above error.

rosinality commented 3 years ago

Thank you, I will merge this.