Promote Retiarii to NAS (step 2) - update imports (#5025)

nni/retiarii/hub/pytorch/proxylessnas.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.hub.pytorch.proxylessnas import *

nni/retiarii/hub/pytorch/shufflenet.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.hub.pytorch.shufflenet import *

nni/retiarii/hub/pytorch/utils/fixed.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.hub.pytorch.utils.fixed import *

nni/retiarii/hub/pytorch/utils/pretrained.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.hub.pytorch.utils.pretrained import *

nni/retiarii/integration.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.execution.common.integration import *

nni/retiarii/integration_api.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.execution.common.integration_api import *

nni/retiarii/mutator.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.mutable.mutator import *

nni/retiarii/nn/pytorch/api.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.nn.pytorch.choice import *

nni/retiarii/nn/pytorch/cell.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.nn.pytorch.cell import *

nni/retiarii/nn/pytorch/component.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import,unused-import
+
+from nni.nas.nn.pytorch.repeat import Repeat
+from nni.nas.nn.pytorch.cell import Cell
+from nni.nas.hub.pytorch.modules import NasBench101Cell, NasBench101Mutator, NasBench201Cell

nni/retiarii/nn/pytorch/hypermodule.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import,unused-import
+
+from nni.nas.hub.pytorch.modules import AutoActivation

nni/retiarii/nn/pytorch/mutation_utils.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.nn.pytorch.mutation_utils import *

nni/retiarii/nn/pytorch/mutator.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.nn.pytorch.mutator import *

nni/retiarii/nn/pytorch/nasbench101.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.hub.pytorch.modules.nasbench101 import *

nni/retiarii/nn/pytorch/nn.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.nn.pytorch.layers import *

nni/retiarii/nn/tensorflow/api.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.nn.tensorflow.api import *

nni/retiarii/oneshot/pytorch/base_lightning.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.oneshot.pytorch.base_lightning import *

nni/retiarii/oneshot/pytorch/dataloader.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.oneshot.pytorch.dataloader import *

nni/retiarii/oneshot/pytorch/differentiable.py

+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+# pylint: disable=wildcard-import,unused-wildcard-import
+
+from nni.nas.oneshot.pytorch.differentiable import *

nni/retiarii/oneshot/pytorch/enas.py

@@ -3,14 +3,14 @@
 
 import logging
 import warnings
-from typing import cast
 
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
 import torch.optim as optim
 from torch.utils.data import SubsetRandomSampler, DataLoader
 
+from nni.nas.oneshot.pytorch.enas import ReinforceController, ReinforceField
+
 from ..interface import BaseOneShotTrainer
 from .random import PathSamplingLayerChoice, PathSamplingInputChoice
 from .utils import AverageMeterGroup, replace_layer_choice, replace_input_choice, to_device