deploy: e8873ae

courtois-neuromod · Dec 12, 2024 · be9efd3 · be9efd3
1 parent 0c8257a
commit be9efd3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
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/litmodule/store.html b/_modules/cneuromax/fitting/deeplearning/litmodule/store.html
@@ -353,6 +353,7 @@ <h1>Source code for cneuromax.fitting.deeplearning.litmodule.store</h1><div clas
 <span></span><span class="sd">&quot;&quot;&quot;:class:`.BaseLitModule` `Hydra &lt;https://hydra.cc&gt;`_ config store.&quot;&quot;&quot;</span>
 
 <span class="kn">from</span> <span class="nn">hydra_zen</span> <span class="kn">import</span> <span class="n">ZenStore</span>
+<span class="kn">from</span> <span class="nn">schedulefree</span> <span class="kn">import</span> <span class="n">AdamWScheduleFree</span>
 <span class="kn">from</span> <span class="nn">torch.optim</span> <span class="kn">import</span> <span class="n">SGD</span><span class="p">,</span> <span class="n">Adam</span><span class="p">,</span> <span class="n">AdamW</span>
 <span class="kn">from</span> <span class="nn">transformers</span> <span class="kn">import</span> <span class="p">(</span>
     <span class="n">get_constant_schedule</span><span class="p">,</span>
@@ -403,6 +404,11 @@ <h1>Source code for cneuromax.fitting.deeplearning.litmodule.store</h1><div clas
 <span class="sd">    &quot;&quot;&quot;</span>
     <span class="n">store</span><span class="p">(</span><span class="n">pfs_builds</span><span class="p">(</span><span class="n">Adam</span><span class="p">),</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;adam&quot;</span><span class="p">,</span> <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">)</span>
     <span class="n">store</span><span class="p">(</span><span class="n">pfs_builds</span><span class="p">(</span><span class="n">AdamW</span><span class="p">),</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;adamw&quot;</span><span class="p">,</span> <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">)</span>
+    <span class="n">store</span><span class="p">(</span>
+        <span class="n">pfs_builds</span><span class="p">(</span><span class="n">AdamWScheduleFree</span><span class="p">),</span>
+        <span class="n">name</span><span class="o">=</span><span class="s2">&quot;sfadamw&quot;</span><span class="p">,</span>
+        <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">,</span>
+    <span class="p">)</span>
     <span class="n">store</span><span class="p">(</span><span class="n">pfs_builds</span><span class="p">(</span><span class="n">SGD</span><span class="p">),</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;sgd&quot;</span><span class="p">,</span> <span class="n">group</span><span class="o">=</span><span class="s2">&quot;litmodule/optimizer&quot;</span><span class="p">)</span></div>