Closed barghit closed 9 months ago
@barghit can you give more info on how you are defining the model with code snippet or log to understand your issue?
Hi! I attached the exact code snippet and the error I get. I don't get the same error with the "Reparameterization" option, it works perfectly. (I also thought it was a problem of the summary method but it does the same when i try to run the training).
fixed with commit 1180b87dc7c0039426d0e41be8c9f52a34db9fc7
The problem seems to be in line 544 of conv_flipout.py, the line is self.kl = kl inside the forward() method, i get the error "local variable 'kl' referenced before assignment" as an UnboundLocalError.
Thank you in advance.