AVCE_FER

AVCE_FER copied to clipboard

Error : y.register_hook in cvx_utils.py file

1

Original line was : y.register_hook(lambda grad,b=batch : torch.solve(grad[:,None], J[b].transpose(0,1))[0][:,0]) I changed to torch.linalg.solve, but still got error as RuntimeError: linalg.solve: A must be batches of square matrices, but they are...

ravinalawade13