Skip to content
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

ADM infer代码的一些疑惑 #15

Open
nzpeng opened this issue Apr 24, 2024 · 1 comment
Open

ADM infer代码的一些疑惑 #15

nzpeng opened this issue Apr 24, 2024 · 1 comment

Comments

@nzpeng
Copy link

nzpeng commented Apr 24, 2024

大佬的代码简洁至极！发现一些问题：ADM infer部分，当前预测的dt_predict 为小数，而非整数，直接拼在p_code上，作为下一步的输入。这种方式貌似和训练时输入都是整数不相符，有可能出现意想不到的结果。建议改为:
dt_predict = torch.round(dt_predict).clamp(1, self.max_duration_token)
@LSimon95
Copy link
Owner

👌有时间测试一下

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@nzpeng @LSimon95