Daniil Krasilnikov

icon indicating an email [email protected]

icon company Yandex Research icon location Moscow, Russia icon location ML Researcher @yandex-research Student @yandexdataschool

Results 1 issues of Daniil Krasilnikov

compute_bins falls with CUDA out of memory for big datasets

Due to the [memory inefficient implementation of torch.quantile](https://github.com/pytorch/pytorch/issues/64947), I get `CUDA out of memory` error when trying to compute bins [here](https://github.com/yandex-research/rtdl-num-embeddings/blob/24173f7023088d48a1a81765029aedd632316d56/package/rtdl_num_embeddings.py#L338) Temporary workaround: copy your tensor on cpu before passing...