Hi,
I am facing a strange issue. I am trying to get WGAN-GP model running in  pytorch. It seems like while training for first few iteration everything seems to run fine, but after that the model for generator throws the error ‘Variable’ object is not callable which is strange because for the previous iteration everything was fine. For example:

image.png797×574 71.2 KB


image.png794×367 39.3 KB


image.png891×337 32.1 KB

My code can be found here: https://github.com/udion/gmmGAN/blob/master/testbed.ipynb 2
The reference code I used in my notebook is here: https://github.com/caogang/wgan-gp/blob/master/gan_toy.py 1
please help.