deploy: 07511b9

courtois-neuromod · Dec 11, 2024 · 0c8257a · 0c8257a
1 parent 8fda97b
commit 0c8257a
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
diff --git a/_modules/cneuromax/fitting/deeplearning/utils/lightning.html b/_modules/cneuromax/fitting/deeplearning/utils/lightning.html
@@ -487,7 +487,7 @@ <h1>Source code for cneuromax.fitting.deeplearning.utils.lightning</h1><div clas
     <span class="k">else</span><span class="p">:</span>
         <span class="n">per_device_batch_size</span> <span class="o">=</span> <span class="n">datamodule</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">fixed_per_device_batch_size</span>
 
-    <span class="k">if</span> <span class="ow">not</span> <span class="n">datamodule</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">fixed_per_device_num_workers</span><span class="p">:</span>
+    <span class="k">if</span> <span class="n">datamodule</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">fixed_per_device_num_workers</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
         <span class="n">proposed_per_device_num_workers</span> <span class="o">=</span> <span class="n">find_good_per_device_num_workers</span><span class="p">(</span>
             <span class="n">datamodule</span><span class="o">=</span><span class="n">datamodule</span><span class="p">,</span>
             <span class="n">per_device_batch_size</span><span class="o">=</span><span class="n">per_device_batch_size</span><span class="p">,</span>