зеркало из https://github.com/microsoft/archai.git
More progress on evolutionary search pareto frontier generation.
This commit is contained in:
Родитель
d6d8055abf
Коммит
a8d01da027
|
@ -612,6 +612,14 @@
|
|||
"console": "integratedTerminal",
|
||||
"args": ["--full", "--algos", "local_search_natsbench_tss_reg", "--datasets", "cifar10", "--no-eval"]
|
||||
},
|
||||
{
|
||||
"name": "Evolution Pareto Search Natsbench Tss",
|
||||
"type": "python",
|
||||
"request": "launch",
|
||||
"program": "${cwd}/scripts/main.py",
|
||||
"console": "integratedTerminal",
|
||||
"args": ["--full", "--algos", "evolution_search_natsbench_tss", "--datasets", "cifar10", "--no-eval"]
|
||||
},
|
||||
{
|
||||
"name": "Local Search DARTS Reg",
|
||||
"type": "python",
|
||||
|
|
|
@ -14,8 +14,8 @@ from archai.nas.arch_meta import ArchWithMetaData
|
|||
from archai.common import utils
|
||||
from archai.search_spaces.discrete_search_spaces.natsbench_tss_search_spaces.discrete_search_space_natsbench_tss import DiscreteSearchSpaceNatsbenchTSS
|
||||
|
||||
from archai.nas.nas_utils.torch_constraints import measure_torch_inference_latency, measure_torch_peak_memory
|
||||
from archai.nas.nas_utils.pareto_frontier import find_pareto_frontier_points
|
||||
from archai.nas.constraints.torch_constraints import measure_torch_inference_latency, measure_torch_peak_memory
|
||||
from archai.nas.constraints.pareto_frontier import find_pareto_frontier_points
|
||||
|
||||
class EvolutionParetoSearchNatsbenchTSS(EvolutionParetoSearch):
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче