neural-compressor icon indicating copy to clipboard operation
neural-compressor copied to clipboard

Update DyNAS interface

Open macsz opened this issue 3 years ago • 3 comments

Signed-off-by: Maciej Szankin [email protected]

macsz avatar Dec 08 '22 19:12 macsz

thanks Maciej, yes, this is what we prefer. thanks for the quick update

ftian1 avatar Dec 09 '22 10:12 ftian1

We still have to work on the NASConfig class, as it currently does not have an option to pass Dynas/NASBase objects to it, but given the timeline I think we can work on it for 2.1.

dynas_setting = Dynas(supernet='transformer_lt_wmt_en_de', metrics=..., population=..., ...)
config = NASConfig(approach='dynas', search_algorithm='nsga2', seed=42, setting=dynas_setting)
agent = NAS(config)
results = agent.search()

macsz avatar Dec 09 '22 10:12 macsz

We still have to work on the NASConfig class, as it currently does not have an option to pass Dynas/NASBase objects to it, but given the timeline I think we can work on it for 2.1.

dynas_setting = Dynas(supernet='transformer_lt_wmt_en_de', metrics=..., population=..., ...)
config = NASConfig(approach='dynas', search_algorithm='nsga2', seed=42, setting=dynas_setting)
agent = NAS(config)
results = agent.search()

totally agree. thanks for your effort

ftian1 avatar Dec 09 '22 12:12 ftian1

As discussed with @ftian1 today this should be postponed to next release (or release after).

macsz avatar Mar 01 '23 01:03 macsz