neural-compressor
neural-compressor copied to clipboard
Update DyNAS interface
thanks Maciej, yes, this is what we prefer. thanks for the quick update
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()
We still have to work on the
NASConfigclass, 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
As discussed with @ftian1 today this should be postponed to next release (or release after).