typo

mwalmsley · Nov 7, 2023 · da24384 · da24384
1 parent c83714c
commit da24384
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/zoobot/pytorch/training/train_with_pytorch_lightning.py b/zoobot/pytorch/training/train_with_pytorch_lightning.py
@@ -223,7 +223,7 @@ def train_default_zoobot_from_scratch(
             num_workers=num_workers,
             prefetch_factor=prefetch_factor
         )
-         use_distributed_sampler=True
+        use_distributed_sampler=True
     else:
         # this branch will use WebDataModule to load premade webdatasets
         datamodule = webdatamodule.WebDataModule(
@@ -235,7 +235,7 @@ def train_default_zoobot_from_scratch(
             cache_dir=cache_dir
             # TODO pass through the rest
         )
-         use_distributed_sampler=False
+        use_distributed_sampler=False
 
     datamodule.setup(stage='fit')
 
@@ -287,7 +287,7 @@ def train_default_zoobot_from_scratch(
         max_epochs=epochs,
         default_root_dir=save_dir,
         plugins=plugins,
-         use_distributed_sampler= use_distributed_sampler
+        use_distributed_sampler=use_distributed_sampler
     )
 
     logging.info((trainer.strategy, trainer.world_size,