got an erro in backwrad process when i run the model with the cora_graphs_bipartite.pt data

[justbin-coder]

Hi @bwilder0,

First of all thanks for such a great work!

i got an error when i run the model with the cora_graphs_bipartite.pt matching data, the error occurs in the backwards process when solving KKT equations for the affine step like the following pic

i have no idea about this error, hope for your repose!

Thanks!