sigma-vae-pytorch icon indicating copy to clipboard operation
sigma-vae-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon orybkin

Update model.py

Open jmyoon1 opened this issue 3 years ago • 2 comments

type change of self.log_sigma to enable requires_grad.

jmyoon1 avatar Nov 09 '21 09:11 jmyoon1

Hmm, is this required now? The code was running in a previous pytorch version.

orybkin avatar Nov 09 '21 18:11 orybkin

Yes. If we do not typecast, we return an error message that indicates that float or complex type is required to be require_grad.

jmyoon1 avatar Nov 09 '21 21:11 jmyoon1