Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enable fixed_arch on Retiarii #3972

Merged
merged 6 commits into from Jul 27, 2021
Merged

Conversation

ultmaster
Copy link
Contributor

@ultmaster ultmaster commented Jul 22, 2021

Fix #3969.

Added an API called fixed_arch to enable users to quickly use the exported architecture.

I haven't tested on other trainers because their retraining code looks over complex. I'll do that as a follow-up task.

@ultmaster ultmaster marked this pull request as ready for review July 22, 2021 08:44
@@ -19,7 +20,7 @@ class DartsLayerChoice(nn.Module):
def __init__(self, layer_choice):
super(DartsLayerChoice, self).__init__()
self.name = layer_choice.key
Copy link
Contributor

Choose a reason for hiding this comment

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

should be layer_choice.label, otherwise there would be warning. DartsInputChoice should also be modified.

@QuanluZhang QuanluZhang merged commit a784613 into microsoft:master Jul 27, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

How to print and save the best model when using one-shot nas
6 participants