fix: remove debugger

lamalab-org · Apr 11, 2024 · b5d5943 · b5d5943
1 parent 4a11e4a
commit b5d5943
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/experiments/train.py b/experiments/train.py
@@ -65,13 +65,15 @@ def train_molbind(config: DictConfig):
             ]
 
     combined_loader = load_combined_loader(
+        central_modality=config.data.central_modality,
         data_modalities=train_modality_data,
         batch_size=config.data.batch_size,
         shuffle=True,
         num_workers=1,
     )
 
     valid_dataloader = load_combined_loader(
+        central_modality=config.data.central_modality,
         data_modalities=valid_modality_data,
         batch_size=config.data.batch_size,
         shuffle=False,
@@ -87,9 +89,7 @@ def train_molbind(config: DictConfig):
 
 @hydra.main(config_path="../configs", config_name="train.yaml")
 def main(config: DictConfig):
-    # train_molbind(config)
-    import pdb; pdb.set_trace()
-    config = instantiate(config)
+    train_molbind(config)
 
 
 if __name__ == "__main__":