uma-pi1 / kge

LibKGE - A knowledge graph embedding library for reproducible research
MIT License
765 stars 124 forks source link

Frequency based sampling broken #272

Open AdrianKs opened 1 year ago

AdrianKs commented 1 year ago

In frequency based sampling, we are using _multinomial_alias_setup and _multinomial_alias_draw. However, these methods were removed as of torch 1.8.

Therefore, frequency based sampling does not work anymore with any newer version of pytorch.

See https://github.com/pytorch/pytorch/issues/21257 and https://github.com/pytorch/pytorch/pull/50489