wTsAI1b

Replacing .sum() with .mean() in dola.py?

2

Hi Team, great work on the project! I noticed something in `dola.py` line 217 that might need attention. The code is: ```python log_probs = diff_logits[range(diff_logits.shape[0]), continue_ids].sum().item() ``` I am wondering...