klcpd_code
klcpd_code copied to clipboard
Getting errors in running the code with python3
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.