Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

Fix Error in nas SPOS trainer, apply_fixed_architecture #3051

Merged
merged 2 commits into from
Nov 4, 2020

Conversation

HeekangPark
Copy link
Contributor

Related to Issue #3047("Several issues in nni.nas.pytorch") by Heekang Park

@ghost
Copy link

ghost commented Oct 31, 2020

CLA assistant check
All CLA requirements met.

@@ -26,9 +26,10 @@ class FixedArchitecture(Mutator):
Force everything that appears in ``fixed_arc`` to be used at least once.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please update the docstring accordingly.

@@ -113,7 +115,7 @@ def replace_layer_choice(self, module=None, prefix=""):
self.replace_layer_choice(mutable, global_name)


def apply_fixed_architecture(model, fixed_arc):
def apply_fixed_architecture(model, fixed_arc, verbose=True):
"""
Load architecture from `fixed_arc` and apply to model.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same here

Add docstring
@liuzhe-lz liuzhe-lz merged commit b372abf into microsoft:master Nov 4, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants