Deploying to gh-pages from main @ 49b0bc0 🚀

AllenCellModeling · Oct 1, 2024 · 4b01738 · 4b01738
1 parent 7d964c5
commit 4b01738
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/_modules/cyto_dl/utils/checkpoint.html b/_modules/cyto_dl/utils/checkpoint.html
@@ -449,7 +449,7 @@ <h1>Source code for cyto_dl.utils.checkpoint</h1><div class="highlight"><pre>
             <span class="s2">&quot;ckpt_path&quot;</span>
         <span class="p">),</span> <span class="s2">&quot;ckpt_path must be provided to with argument weights_only=True&quot;</span>
         <span class="c1"># load model from state dict to get around trainer.max_epochs limit, useful for resuming model training from existing weights</span>
-        <span class="n">state_dict</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">load_params</span><span class="p">[</span><span class="s2">&quot;ckpt_path&quot;</span><span class="p">])[</span><span class="s2">&quot;state_dict&quot;</span><span class="p">]</span>
+        <span class="n">state_dict</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">load_params</span><span class="p">[</span><span class="s2">&quot;ckpt_path&quot;</span><span class="p">],</span> <span class="n">map_location</span><span class="o">=</span><span class="s2">&quot;cpu&quot;</span><span class="p">)[</span><span class="s2">&quot;state_dict&quot;</span><span class="p">]</span>
         <span class="n">model</span><span class="o">.</span><span class="n">load_state_dict</span><span class="p">(</span><span class="n">state_dict</span><span class="p">,</span> <span class="n">strict</span><span class="o">=</span><span class="n">load_params</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;strict&quot;</span><span class="p">,</span> <span class="kc">True</span><span class="p">))</span>
         <span class="c1"># set ckpt_path to None to avoid loading checkpoint again with model.fit/model.test</span>
         <span class="n">load_params</span><span class="p">[</span><span class="s2">&quot;ckpt_path&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>