Skip to content

Commit

Permalink
remove debug
Browse files Browse the repository at this point in the history
  • Loading branch information
Ivan-Zhou committed Feb 13, 2024
1 parent 29538d6 commit 8d82a9c
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/levanter/data/mixture.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,6 @@ def __init__(
if stop_strategy not in [StopStrategy.FIRST_STOP_STRATEGY, StopStrategy.ALL_STOP_STRATEGY]:
raise ValueError(f"Stop strategy {stop_strategy} is not supported.")

print(f"=== class MixtureDataset: self.datasets.keys() = {self.datasets.keys()}, self.weights = {self.weights}, stop_strategy = {stop_strategy} ===")
self.stop_strategy = stop_strategy

if not isinstance(key, int):
Expand Down

0 comments on commit 8d82a9c

Please sign in to comment.