-
Notifications
You must be signed in to change notification settings - Fork 5.9k
Add Light-NAS for PaddleSlim #17679
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
Add Light-NAS for PaddleSlim #17679
Conversation
1. Fix compressor. 2. Enhence graph executor. 3. Add SAController 4. Add auto pruning strategy. 5. Add unitest for auto pruning strategy. test=develop
test=develop
test=develop
… slim_ligth_nas
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.
请参考提示进行review
test=develop
20ec490 to
2e53161
Compare
1. Fix unitest on windows. 2. Fix package importing in tests directory.
bb2acb3 to
2f4058d
Compare
2. Expose eval_epoch option. 3. Remove unused function in search_agent. 4. Expose max_client_num to yaml file. 5. Move flops constraint to on_epoch_begin function test=develop
2f4058d to
d5c4eaa
Compare
test=develop
612c3ae to
ea54e74
Compare
test=develop
57417dc to
734878f
Compare
2. Remove unitest of light-nas for windows. test=develop
734878f to
a6c428c
Compare
Enhence controller test=develop
… slim_ligth_nas
|
|
||
| def update(self, tokens, reward): | ||
| """ | ||
| Update the controller according to latest tokens and reward. |
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.
reward->award?不太清楚reward是想表述授予吗?如果是的话,相关的都需要修改
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.
这里就是需要reward,表示当前搜索出来的网络结构获得的奖励。
python/paddle/fluid/contrib/slim/tests/light_nas/light_nasnet.py
Outdated
Show resolved
Hide resolved
python/paddle/fluid/contrib/slim/tests/light_nas/light_nasnet.py
Outdated
Show resolved
Hide resolved
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.
请参考对应地方的review意见
b4d9374 to
ce50a96
Compare
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.
LG API
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.
LGTM
test=develop
ce50a96 to
adfdbfd
Compare
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.
LG API
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.
LGTM
Uh oh!
There was an error while loading. Please reload this page.