Skip to content

Commit b440236

Browse files
committed
Fix optimal mixing ratio
1 parent 696eb3c commit b440236

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

users/dorian_koch/datasets/MixingDataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def _run_seq_idx(self, seq_idx):
226226
# so just start loading them at the beginning again
227227
if all(self.datasets_exhausted):
228228
self.is_chooser_done = True
229-
print(f"MixingDataset: optimal mixing ratio = {child_indices[1] / max(1, child_indices[0]+child_indices[1])}", file=log.v4)
229+
print(f"MixingDataset: optimal mixing ratio = {self.datalens[1] / max(1, self.datalens[0]+self.datalens[1])}", file=log.v4)
230230
break
231231
# the modulo operator below will wrap around
232232

0 commit comments

Comments
 (0)