Type cast bath size to int

Slightly increase mouse sensitivity
OHDSI · Oct 17, 2023 · 2d0eadd · 2d0eadd
1 parent 357b14e
commit 2d0eadd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inst/python/LrFinder.py b/inst/python/LrFinder.py
@@ -56,7 +56,7 @@ def __init__(self,
         self.scheduler = ExponentialSchedulerPerBatch(self.optimizer, self.max_lr, self.num_lr)
 
         self.criterion = estimator_settings["criterion"]()
-        self.batch_size = estimator_settings['batch_size']
+        self.batch_size = int(estimator_settings['batch_size'])
         self.losses = None
         self.loss_index = None