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

Add unit tests for NAS #1954

Merged
merged 18 commits into from
Feb 7, 2020
Merged

Add unit tests for NAS #1954

merged 18 commits into from
Feb 7, 2020

Conversation

ultmaster
Copy link
Contributor

No description provided.

selected_multihot = []
for i, src_key in enumerate(mutable.choose_from):
if i not in topk_edge_indices and src_key in result:
result[src_key] = torch.zeros_like(result[src_key]) # clear this choice to optimize calc graph
Copy link
Contributor

Choose a reason for hiding this comment

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

what is the meaning of this line?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

If an edge is never selected, there is no need to calculate any op on this edge. This is to eliminate redundant calculation. If there is any problem with this, open an issue.

Copy link
Contributor

Choose a reason for hiding this comment

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

please add comments here to explain.

@QuanluZhang QuanluZhang merged commit b49b38f into master Feb 7, 2020
@liuzhe-lz liuzhe-lz deleted the add-nas-test-for-it branch July 17, 2020 11:15
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