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

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 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.

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