Output 0 of PermuteBackward0

[xchuwenbo]

File "/data00/home/huwenbo.ml/flowplusplus/train.py", line 84, in main train(epoch, net, trainloader, device, optimizer, scheduler, File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context return func(*args, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/train.py", line 99, in train z, sldj = net(x, reverse=False) File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/flowplusplus.py", line 63, in forward x, sldj = self.dequantize(x, sldj) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/flowplusplus.py", line 71, in dequantize x, sldj = self.dequant_flows(x, sldj) File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/flowplusplus.py", line 247, in forward u, sldj = flow(u, sldj, aux=aux) if i % 4 == 2 else flow(u, sldj) File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/coupling.py", line 26, in forward a, b, pi, mu, s = self.nn(x_id, aux) File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/nn.py", line 46, in forward x = conv(x, aux) File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/nn.py", line 76, in forward x = self.attn(x) + x File "/data00/home/huwenbo.ml/lib/conda/envs/f++/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/data00/home/huwenbo.ml/flowplusplus/models/flowplusplus/nn.py", line 121, in forward q *= key_depth_per_head ** -0.5 RuntimeError: Output 0 of PermuteBackward0 is a view and is being modified inplace. This view is the output of a function that returns multiple views. Such functions do not allow the output views to be modified inplace. You should replace the inplace operation by an out-of-place one.