Drop historical sample counts

daviswer · web-flow · commit b9fa538fd24a · 2025-05-09T14:10:05.000-04:00
diff --git a/fms_fsdp/utils/dataset_utils.py b/fms_fsdp/utils/dataset_utils.py
@@ -1366,7 +1366,7 @@ def __init__(
         self.tokens_seen = [0] * len(self.datasets)
 
         self.current_iterator = -1
-        self.state_params = ["tokens_seen", "current_iterator"]
+        self.state_params = ["current_iterator"]
 
     def setup(self):
         if not self.is_setup:
