Inplace operation error occurs with batch size 1 about glow-pytorch HOT 5 CLOSED

I was getting a similar error when using some of the code - looks like it was coming from the in-place ops in the coupling layers (e.g. z2 += z1). Just sent a PR to fix it.

from glow-pytorch.

@kolchinski Thanks a lot.
@wmjung It seems the bugs are fixed.

from glow-pytorch.

Hi Wonmo,
Have you ever tried to change nn.ReLU(inplace=True) into nn.ReLU(inplace=False)?
In thops.py, squeeze_() and scatter_() could be problem too.
It seems that there are no other inplace operations in my code.

I cannot run codes these days. I will fix this later. You can try to modify the above three mentioned places first.
Thanks.

from glow-pytorch.

Thank you for your reply.
I'll try with those modifications :)

from glow-pytorch.

I tried to modify some places and run on my cpu, but failed to figure out the problem.
I have no idea what happened. It could be a small bug.
If you found something, please tell me.
Thank you.

from glow-pytorch.