Skip to content

GitLab

Unverified Commit 12a5e3de authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Refactor HPO tuner code hierarchy (#3187) 


Co-authored-by: default avatarliuzhe <liuzhe.cs@pku.edu.cn>
parent 872554f1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/ut/sdk/test_hyperopt_tuner.py
View file @ 12a5e3de
...@@ -9,7 +9,7 @@ from unittest import TestCase, main ...@@ -9,7 +9,7 @@ from unittest import TestCase, main
import hyperopt as hp import hyperopt as hp
from nni.algorithms.hpo.hyperopt_tuner.hyperopt_tuner import json2space, json2parameter, json2vals, HyperoptTuner from nni.algorithms.hpo.hyperopt_tuner import json2space, json2parameter, json2vals, HyperoptTuner
class HyperoptTunerTestCase(TestCase): class HyperoptTunerTestCase(TestCase):
......
test/ut/sdk/test_networkmorphism_tuner.py
View file @ 12a5e3de
...@@ -6,6 +6,7 @@ from unittest import TestCase, main ...@@ -6,6 +6,7 @@ from unittest import TestCase, main
from copy import deepcopy from copy import deepcopy
import torch import torch
from nni.algorithms.hpo.networkmorphism_tuner import NetworkMorphismTuner
from nni.algorithms.hpo.networkmorphism_tuner.graph import graph_to_json, json_to_graph from nni.algorithms.hpo.networkmorphism_tuner.graph import graph_to_json, json_to_graph
from nni.algorithms.hpo.networkmorphism_tuner.graph_transformer import ( from nni.algorithms.hpo.networkmorphism_tuner.graph_transformer import (
to_deeper_graph, to_deeper_graph,
...@@ -13,7 +14,6 @@ from nni.algorithms.hpo.networkmorphism_tuner.graph_transformer import ( ...@@ -13,7 +14,6 @@ from nni.algorithms.hpo.networkmorphism_tuner.graph_transformer import (
to_wider_graph, to_wider_graph,
) )
from nni.algorithms.hpo.networkmorphism_tuner.layers import layer_description_extractor from nni.algorithms.hpo.networkmorphism_tuner.layers import layer_description_extractor
from nni.algorithms.hpo.networkmorphism_tuner.networkmorphism_tuner import NetworkMorphismTuner
from nni.algorithms.hpo.networkmorphism_tuner.nn import CnnGenerator from nni.algorithms.hpo.networkmorphism_tuner.nn import CnnGenerator
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment