Unverified Commit e21a6984 authored by liuzhe-lz's avatar liuzhe-lz Committed by GitHub
Browse files

[v2.0] Refactor code hierarchy (part 2) (#2987)

parent f98ee672
...@@ -7,7 +7,7 @@ import numpy as np ...@@ -7,7 +7,7 @@ import numpy as np
import torch import torch
from torch import nn from torch import nn
from nni.nas.pytorch.darts import DartsMutator from nni.algorithms.nas.pytorch.darts import DartsMutator
from nni.nas.pytorch.mutables import LayerChoice from nni.nas.pytorch.mutables import LayerChoice
......
...@@ -5,7 +5,7 @@ import json ...@@ -5,7 +5,7 @@ import json
import logging import logging
from nni.nas.pytorch.callbacks import LRSchedulerCallback from nni.nas.pytorch.callbacks import LRSchedulerCallback
from nni.nas.pytorch.darts import DartsTrainer from nni.algorithms.nas.pytorch.darts import DartsTrainer
from nni.nas.pytorch.trainer import BaseTrainer, TorchTensorEncoder from nni.nas.pytorch.trainer import BaseTrainer, TorchTensorEncoder
from .mutator import PdartsMutator from .mutator import PdartsMutator
......
...@@ -9,7 +9,7 @@ from collections import deque ...@@ -9,7 +9,7 @@ from collections import deque
import numpy as np import numpy as np
from nni.tuner import Tuner from nni.tuner import Tuner
from nni.nas.pytorch.classic_nas.mutator import LAYER_CHOICE, INPUT_CHOICE from nni.algorithms.nas.pytorch.classic_nas.mutator import LAYER_CHOICE, INPUT_CHOICE
_logger = logging.getLogger(__name__) _logger = logging.getLogger(__name__)
......
...@@ -9,7 +9,7 @@ import sys ...@@ -9,7 +9,7 @@ import sys
import tensorflow as tf import tensorflow as tf
import nni import nni
from nni.env_vars import trial_env_vars from nni.runtime.env_vars import trial_env_vars
from nni.nas.tensorflow.mutables import LayerChoice, InputChoice, MutableScope from nni.nas.tensorflow.mutables import LayerChoice, InputChoice, MutableScope
from nni.nas.tensorflow.mutator import Mutator from nni.nas.tensorflow.mutator import Mutator
......
...@@ -50,8 +50,8 @@ class Assessor(Recoverable): ...@@ -50,8 +50,8 @@ class Assessor(Recoverable):
See Also See Also
-------- --------
Builtin assessors: Builtin assessors:
:class:`~nni.medianstop_assessor.MedianstopAssessor` :class:`~nni.algorithms.hpo.medianstop_assessor.MedianstopAssessor`
:class:`~nni.curvefitting_assessor.CurvefittingAssessor` :class:`~nni.algorithms.hpo.curvefitting_assessor.CurvefittingAssessor`
""" """
def assess_trial(self, trial_job_id, trial_history): def assess_trial(self, trial_job_id, trial_history):
......
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