Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Commit 13b1839

Browse files
committed
Merge branch 'main' of github.com:neuralmagic/docs into main
2 parents 29c9ade + 3074bb8 commit 13b1839

File tree

382 files changed

+3731
-3652
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

382 files changed

+3731
-3652
lines changed

docs/source/sparsification/flow-sparsification-model_recipe_sparsezoo-sparseml_transfer_learning.svg

Lines changed: 255 additions & 0 deletions
Loading

sparseml/_modules/index.html

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,16 +98,17 @@
9898
<ul>
9999
<li class="toctree-l1"><a class="reference internal" href="../quicktour.html">Quick Tour</a></li>
100100
<li class="toctree-l1"><a class="reference internal" href="../installation.html">Installation</a></li>
101-
<li class="toctree-l1"><a class="reference internal" href="../recipes.html">Optimization Recipes</a></li>
101+
<li class="toctree-l1"><a class="reference internal" href="../recipes.html">Sparsification Recipes</a></li>
102102
</ul>
103103
<p class="caption"><span class="caption-text">API</span></p>
104104
<ul>
105105
<li class="toctree-l1"><a class="reference internal" href="../api/sparseml.html">sparseml package</a></li>
106106
</ul>
107-
<p class="caption"><span class="caption-text">Help and Support</span></p>
107+
<p class="caption"><span class="caption-text">Help</span></p>
108108
<ul>
109109
<li class="toctree-l1"><a class="reference external" href="https://github.com/neuralmagic/sparseml/issues">Bugs, Feature Requests</a></li>
110110
<li class="toctree-l1"><a class="reference external" href="https://github.com/neuralmagic/sparseml/discussions">Support, General Q&amp;A</a></li>
111+
<li class="toctree-l1"><a class="reference external" href="https://docs.neuralmagic.com">Neural Magic Docs</a></li>
111112
</ul>
112113

113114

sparseml/_modules/sparseml/keras/optim/manager.html

Lines changed: 19 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -98,16 +98,17 @@
9898
<ul>
9999
<li class="toctree-l1"><a class="reference internal" href="../../../../quicktour.html">Quick Tour</a></li>
100100
<li class="toctree-l1"><a class="reference internal" href="../../../../installation.html">Installation</a></li>
101-
<li class="toctree-l1"><a class="reference internal" href="../../../../recipes.html">Optimization Recipes</a></li>
101+
<li class="toctree-l1"><a class="reference internal" href="../../../../recipes.html">Sparsification Recipes</a></li>
102102
</ul>
103103
<p class="caption"><span class="caption-text">API</span></p>
104104
<ul>
105105
<li class="toctree-l1"><a class="reference internal" href="../../../../api/sparseml.html">sparseml package</a></li>
106106
</ul>
107-
<p class="caption"><span class="caption-text">Help and Support</span></p>
107+
<p class="caption"><span class="caption-text">Help</span></p>
108108
<ul>
109109
<li class="toctree-l1"><a class="reference external" href="https://github.com/neuralmagic/sparseml/issues">Bugs, Feature Requests</a></li>
110110
<li class="toctree-l1"><a class="reference external" href="https://github.com/neuralmagic/sparseml/discussions">Support, General Q&amp;A</a></li>
111+
<li class="toctree-l1"><a class="reference external" href="https://docs.neuralmagic.com">Neural Magic Docs</a></li>
111112
</ul>
112113

113114

@@ -202,6 +203,7 @@ <h1>Source code for sparseml.keras.optim.manager</h1><div class="highlight"><pre
202203
<span class="kn">from</span> <span class="nn">sparseml.keras.utils.logger</span> <span class="kn">import</span> <span class="n">KerasLogger</span>
203204
<span class="kn">from</span> <span class="nn">sparseml.optim</span> <span class="kn">import</span> <span class="n">BaseManager</span>
204205
<span class="kn">from</span> <span class="nn">sparseml.utils</span> <span class="kn">import</span> <span class="n">load_recipe_yaml_str</span>
206+
<span class="kn">from</span> <span class="nn">sparsezoo.objects</span> <span class="kn">import</span> <span class="n">OptimizationRecipe</span>
205207

206208

207209
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s2">&quot;ScheduledModifierManager&quot;</span><span class="p">]</span>
@@ -213,15 +215,23 @@ <h1>Source code for sparseml.keras.optim.manager</h1><div class="highlight"><pre
213215
<span class="sd"> &quot;&quot;&quot;</span>
214216

215217
<div class="viewcode-block" id="ScheduledModifierManager.from_yaml"><a class="viewcode-back" href="../../../../api/sparseml.keras.optim.html#sparseml.keras.optim.manager.ScheduledModifierManager.from_yaml">[docs]</a> <span class="nd">@staticmethod</span>
216-
<span class="k">def</span> <span class="nf">from_yaml</span><span class="p">(</span><span class="n">file_path</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">add_modifiers</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Modifier</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">):</span>
218+
<span class="k">def</span> <span class="nf">from_yaml</span><span class="p">(</span>
219+
<span class="n">file_path</span><span class="p">:</span> <span class="n">Union</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="n">OptimizationRecipe</span><span class="p">],</span>
220+
<span class="n">add_modifiers</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Modifier</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
221+
<span class="p">):</span>
217222
<span class="sd">&quot;&quot;&quot;</span>
218-
<span class="sd"> Convenience function used to create the manager of multiple modifiers</span>
219-
<span class="sd"> from a yaml file.</span>
220-
221-
<span class="sd"> :param file_path: the path to the yaml file to load the modifier from</span>
223+
<span class="sd"> Convenience function used to create the manager of multiple modifiers from a</span>
224+
<span class="sd"> recipe file.</span>
225+
226+
<span class="sd"> :param file_path: the path to the recipe file to load the modifier from, or</span>
227+
<span class="sd"> a SparseZoo model stub to load a recipe for a model stored in SparseZoo.</span>
228+
<span class="sd"> SparseZoo stubs should be preceded by &#39;zoo:&#39;, and can contain an optional</span>
229+
<span class="sd"> &#39;?recipe_type=&lt;type&gt;&#39; parameter. Can also be a SparseZoo OptimizationRecipe</span>
230+
<span class="sd"> object. i.e. &#39;/path/to/local/recipe.yaml&#39;, &#39;zoo:model/stub/path&#39;,</span>
231+
<span class="sd"> &#39;zoo:model/stub/path?recipe_type=transfer&#39;</span>
222232
<span class="sd"> :param add_modifiers: additional modifiers that should be added to the</span>
223-
<span class="sd"> returned manager alongside the ones loaded from the yaml file</span>
224-
<span class="sd"> :return: ScheduledModifierManager() created from the yaml file</span>
233+
<span class="sd"> returned manager alongside the ones loaded from the recipe file</span>
234+
<span class="sd"> :return: ScheduledModifierManager() created from the recipe file</span>
225235
<span class="sd"> &quot;&quot;&quot;</span>
226236
<span class="n">yaml_str</span> <span class="o">=</span> <span class="n">load_recipe_yaml_str</span><span class="p">(</span><span class="n">file_path</span><span class="p">)</span>
227237
<span class="n">modifiers</span> <span class="o">=</span> <span class="n">Modifier</span><span class="o">.</span><span class="n">load_list</span><span class="p">(</span><span class="n">yaml_str</span><span class="p">)</span>

0 commit comments

Comments
 (0)