box-embeddings
box-embeddings copied to clipboard
make loss.backward() work
I was trying to use gumbel_intersection in my training code. However, when calling the loss.backward(), there would be an error message. Imgur which shows that there should not be an in-place operation with tunable parameter. After I made some changes in the intersection function, the code worked out.
Thanks for creating the PR. I am not entirely sure if this issue is coming up due to the box-embeddings library—we have been able to backprop successfully while using gumbel intersection. Would you be able to create a small test case that isolates this issue this issue?