98
98
< ul >
99
99
< li class ="toctree-l1 "> < a class ="reference internal " href ="../../../../quicktour.html "> Quick Tour</ a > </ li >
100
100
< 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 >
102
102
</ ul >
103
103
< p class ="caption "> < span class ="caption-text "> API</ span > </ p >
104
104
< ul >
105
105
< li class ="toctree-l1 "> < a class ="reference internal " href ="../../../../api/sparseml.html "> sparseml package</ a > </ li >
106
106
</ 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 >
108
108
< ul >
109
109
< li class ="toctree-l1 "> < a class ="reference external " href ="https://github.com/neuralmagic/sparseml/issues "> Bugs, Feature Requests</ a > </ li >
110
110
< li class ="toctree-l1 "> < a class ="reference external " href ="https://github.com/neuralmagic/sparseml/discussions "> Support, General Q&A</ a > </ li >
111
+ < li class ="toctree-l1 "> < a class ="reference external " href ="https://docs.neuralmagic.com "> Neural Magic Docs</ a > </ li >
111
112
</ ul >
112
113
113
114
@@ -202,6 +203,7 @@ <h1>Source code for sparseml.keras.optim.manager</h1><div class="highlight"><pre
202
203
< span class ="kn "> from</ span > < span class ="nn "> sparseml.keras.utils.logger</ span > < span class ="kn "> import</ span > < span class ="n "> KerasLogger</ span >
203
204
< span class ="kn "> from</ span > < span class ="nn "> sparseml.optim</ span > < span class ="kn "> import</ span > < span class ="n "> BaseManager</ span >
204
205
< 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 >
205
207
206
208
207
209
< span class ="n "> __all__</ span > < span class ="o "> =</ span > < span class ="p "> [</ span > < span class ="s2 "> "ScheduledModifierManager"</ span > < span class ="p "> ]</ span >
@@ -213,15 +215,23 @@ <h1>Source code for sparseml.keras.optim.manager</h1><div class="highlight"><pre
213
215
< span class ="sd "> """</ span >
214
216
215
217
< 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 >
217
222
< span class ="sd "> """</ 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 'zoo:', and can contain an optional</ span >
229
+ < span class ="sd "> '?recipe_type=<type>' parameter. Can also be a SparseZoo OptimizationRecipe</ span >
230
+ < span class ="sd "> object. i.e. '/path/to/local/recipe.yaml', 'zoo:model/stub/path',</ span >
231
+ < span class ="sd "> 'zoo:model/stub/path?recipe_type=transfer'</ span >
222
232
< 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 >
225
235
< span class ="sd "> """</ span >
226
236
< 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 >
227
237
< 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