Hello, how can I solve the error?

jxbbb / TOD3Cap

[ECCV 2024] TOD3Cap: Towards 3D Dense Captioning in Outdoor Scenes

101 stars 5 forks source link

Hello, how can I solve the error? #7

Open fubai-666 opened 2 months ago

fubai-666 commented 2 months ago

File "/workspace/nkh1szh/TOD3Cap/llama/utils.py", line 16, in sample_top_p next_token = torch.multinomial(probs_sort, num_samples=1) RuntimeError: probability tensor contains either inf, nan or element < 0