tgn icon indicating copy to clipboard operation
tgn copied to clipboard

Published 20 hours ago • verified publisher icon twitter-research

Run script failed due to assertion

Open zcxsythenew opened this issue 1 year ago • 0 comments

Hi team, when I tried to run script using Reddit dataset, it failed after 6 training epoch.

Screenshot_2023-12-21-21-43-10-33_b925948ebc6526d76511b6ea3af63aeb.jpg

I run the script on NVIDIA RTX 3080ti, which does not support CUDA 10.1. Therefore, I upgraded torch version to 2.1 with CUDA 12.1. Other requirements (pandas, scikit-learn, python3.7) stated in README.md are unchanged.

How could this happen?

If you need any more details about my environment, please comment and I will provide them.

zcxsythenew avatar Dec 21 '23 13:12 zcxsythenew