New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Gumbel噪声的计算会出现-inf #26

Closed

wangxin211 opened this issue Jan 27, 2024 · 1 comment

wangxin211 commented Jan 27, 2024

在AutoCF代码中计算Gumble噪声时，会计算出-inf，进而导致infoloss的值变为inf

Collaborator

Re-bin commented Jan 27, 2024 •

edited

Loading

嗨，感谢关注👋

谢谢指正，可以尝试一下把42行替换成

noise = t.rand(scores.shape).cuda()
noise[noise == 0] = 1e-8

这样避免sample到0的情况，应该能解决问题 :)

Best regards,
Xubin

HKUDS closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment