dyad: fixes bug in DYAD data loader

This commit fixes a bug in the DYAD PyTorch data loader that causes 'brokers_per_node' to not be set before reference.
flux-framework · Aug 21, 2024 · e0d6589 · e0d6589
1 parent 7af6d1c
commit e0d6589
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...AWS/JupyterNotebook/tutorial/supplementary/dyad/dlio_extensions/dyad_torch_data_loader.py b/...AWS/JupyterNotebook/tutorial/supplementary/dyad/dlio_extensions/dyad_torch_data_loader.py
@@ -48,11 +48,11 @@ def __init__(self, format_type, dataset_type, epoch, num_samples, num_workers, b
         self.reader = None
         self.num_images_read = 0
         self.batch_size = batch_size
+        self.broker_per_node = 1
         args = ConfigArguments.get_instance()
         self.serial_args = pickle.dumps(args)
         if num_workers == 0:
             self.worker_init(-1)
-        self.broker_per_node = 1
 
     def worker_init(self, worker_id):
         # Configure PyTorch components