search

rixwew / pytorch-fm

Factorization Machine models in PyTorch
MIT License
1.04k stars 225 forks source link

RuntimeError: CUDA error: device-side assert triggered #10

Closed p-null closed 5 years ago

p-null commented 5 years ago

Hi, It threw out this error RuntimeError: CUDA error: device-side assert triggered when running ffm model, can you please take a look at it and fix it? Thank you!

rixwew commented 5 years ago

I could not reproduce RuntimeError: CUDA error: device-side assert triggered @tingkai-zhang Could you tell me your pytorch version and dataset name?

p-null commented 5 years ago

I could not reproduce RuntimeError: CUDA error: device-side assert triggered @tingkai-zhang Could you tell me your pytorch version and dataset name?

Thanks for replying. Been busy these days.

Pytorch Version：1.2 Dataset: MovieLens1M

Let me know if there is anything I could provide.

rixwew commented 5 years ago

Thank you for your time! MovieLens1M error was reported in #6, and fixed with https://github.com/rixwew/pytorch-fm/commit/782d9598a25c37b77bbeb39ad28406087f2e5e01 If you are using pypi torchfm, please update the package.

pip install torchfm --upgrade
p-null commented 5 years ago

Sweet. Thank you