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
import torch
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
......
......@@ -5,7 +5,7 @@ import json
import logging
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 .mutator import PdartsMutator
......
......@@ -9,7 +9,7 @@ from collections import deque
import numpy as np
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__)
......
......@@ -9,7 +9,7 @@ import sys
import tensorflow as tf
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.mutator import Mutator
......
......@@ -50,8 +50,8 @@ class Assessor(Recoverable):
See Also
--------
Builtin assessors:
:class:`~nni.medianstop_assessor.MedianstopAssessor`
:class:`~nni.curvefitting_assessor.CurvefittingAssessor`
:class:`~nni.algorithms.hpo.medianstop_assessor.MedianstopAssessor`
:class:`~nni.algorithms.hpo.curvefitting_assessor.CurvefittingAssessor`
"""
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