prevent tuner to choose BS=0 (#207)

MaximilienLC · Maximilien Le Clei · commit 65c9a039547c · 2024-01-24T22:03:36.000Z
diff --git a/cneuromax/fitting/deeplearning/utils/lightning.py b/cneuromax/fitting/deeplearning/utils/lightning.py
@@ -189,6 +189,8 @@ def find_good_per_device_batch_size(
         len(datamodule_copy.train_dataloader())
         // (100 * num_computing_devices),
     )
+    if per_device_batch_size == 0:
+        per_device_batch_size = 1
     logging.info(f"Best `batch_size` parameter: {per_device_batch_size}.")
     return per_device_batch_size
 

Original file line number	Diff line number	Diff line change
`@@ -189,6 +189,8 @@ def find_good_per_device_batch_size(`
`189`	`189`	`len(datamodule_copy.train_dataloader())`
`190`	`190`	`// (100 * num_computing_devices),`
`191`	`191`	`)`
	`192`	`+ if per_device_batch_size == 0:`
	`193`	`+ per_device_batch_size = 1`
`192`	`194`	logging.info(f"Best `batch_size` parameter: {per_device_batch_size}.")
`193`	`195`	`return per_device_batch_size`
`194`	`196`