add comments

Author: ramanishsingh

test/stateful_dataloader/test_state_dict.py

@@ -1591,7 +1591,7 @@ def get_map_dl(self, data_size, num_workers, batch_size, sampler_cls):
         )
 
     def _run(self, data_size, num_workers, batch_size, interrupt, sampler_cls):
-        torch.manual_seed(0)
+        torch.manual_seed(0)  # Fixing seed for deterministic results
         dataloader1 = self.get_map_dl(
             data_size=data_size,
             num_workers=num_workers,
@@ -1606,7 +1606,9 @@ def _run(self, data_size, num_workers, batch_size, interrupt, sampler_cls):
                 break
         state_dict = dataloader1.state_dict()
 
-        torch.manual_seed(0)
+        torch.manual_seed(
+            0
+        )  # We need to fix seed again so that before fast forwarding we are at the same state of gen as before
         resumed_dataloader1 = self.get_map_dl(
             data_size=data_size,
             num_workers=num_workers,