search

Thinklab-SJTU / pygmtools

A Python Graph Matching Toolkit.
https://pygmtools.readthedocs.io/
Other
275 stars 19 forks source link

about sinkhorn #64

Closed lansfair closed 11 months ago

lansfair commented 11 months ago

I'm trying to use pygm.sinkhorn in my model, but it causes the gradients to vanish. Why does this happen?

rogerwwww commented 11 months ago

Perhaps you should try printing the output of pygm.sinkhorn -- if it is very close to 0/1 values, it may cause the issue. And try increasing the value of tau in this case.

rogerwwww commented 11 months ago

I am closing this issue since it seems no longer active. Plz feel free to reopen it.