More progress on evolutionary search pareto frontier generation.

This commit is contained in:
Debadeepta Dey 2022-04-18 20:20:56 -07:00 коммит произвёл Gustavo Rosa
Родитель d6d8055abf
Коммит a8d01da027
4 изменённых файлов: 10 добавлений и 2 удалений

8
.vscode/launch.json поставляемый
Просмотреть файл

@ -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):

Просмотреть файл