Getting errors in running the code with python3

[stilllearningsoumya]

While running the code in python3 environment, I am getting this error -- RuntimeError: Mismatch in shape: grad_output[0] has a shape of torch.Size([1]) and output[0] has a shape of torch.Size([]).

Any help in this regard will be highly appreciated. Thanks in advance.

[dars16]

Thanks to jalola/improved-wgan-pytorch#11 (comment), the solution is to replace line 185 in klcpd.py with one = torch.tensor(1, dtype=torch.float)