Selected training excerpt · lines 473–473 · source SHA-256 5718458b537d
473 masked, in_loss = token_flags_r6(grid, masks) # (B, total); in_loss == masked