-
Notifications
You must be signed in to change notification settings - Fork 1.8k
Conversation
# Conflicts: # README.md # docs/en_US/NAS/Overview.md
|
||
## Introduction | ||
CDARTS builds a cyclic feedback mechanism between the search and evaluation networks. First, the search network generates an initial topology for evaluation, so that the weights of the evaluation network can be optimized. Second, the architecture topology in the search network is further optimized by the label supervision in classification, as well as the regularization from the evaluation network through feature distillation. Repeating the above cycle results in a joint optimization of the search and evaluation networks, and thus enables the evolution of the topology to fit the final evaluation network. | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
from nni.nas.pytorch.mutator import Mutator # pylint: disable=wrong-import-order | ||
|
||
|
||
class RegularizedDartsMutator(DartsMutator): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
it would be great if you add docstring for these mutators.
examples/nas/cdarts/README.md
Outdated
@@ -0,0 +1,110 @@ | |||
# Cyclic Differentiable Architecture Search | |||
This is CDARTS based on the NNI platform, which currently supports CIFAR10 search and retrain. ImageNet search and retrain should also be supported, and we provide corresponding interfaces. Our reproduced results on NNI are slightly lower than the paper, but much higher than the original DARTS. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
" Our reproduced results on NNI are slightly lower than the paper", I wonder why.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@penghouwen Can you elaborate?
examples/nas/cdarts/README.md
Outdated
|
||
## Search and Retrain | ||
### Search | ||
* Main python file is ${ROOT}/search.py |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
where is ${ROOT}
, seems don't mention before?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I can't see this file. Are you commenting on an old commit?
DON'T MERGE. Still pending review from @xuehui1991 . |
Can anyone provide the paper related to CDARTS? I can't find any details about this method。 |
cc @penghouwen |
No description provided.