-
Notifications
You must be signed in to change notification settings - Fork 1.8k
PPO tuner for NAS, supports NNI's NAS interface #1380
PPO tuner for NAS, supports NNI's NAS interface #1380
Conversation
'PPOTuner': { | ||
'builtinTunerName': 'PPOTuner', | ||
'classArgs': { | ||
'optimize_mode': setChoice('optimize_mode', 'maximize', 'minimize'), |
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 think there the hyper-parameter in PPO may affect the performance of this tuner. such as "learning rate"...
> Built-in Tuner Name: **PPOTuner** | ||
|
||
Note that the only acceptable type of search space is `mutable_layer`. `optional_input_size` can only be 0, 1, or [0, 1]. | ||
|
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.
maybe the installation command should be mentioned here
@@ -0,0 +1,3 @@ | |||
enum34 |
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.
why is enum34 needed ?
""" | ||
|
||
return True | ||
|
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.
unused function ?
delivered to dev branch, will merge in Aug release. |
* PPO tuner for NAS, supports NNI's NAS interface (#1380)
This is a tuner generally for NNI's NAS interface, it uses ppo algorithm. The implementation inherits the main logic of the implementation here (i.e., ppo2 from openai), and is adapted for NAS scenario.
It could successfully tune the mnist-nas example, and has the following result:
result of tuning enas search space (limited epoch number)