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