From 009722a67ba034734ca4f488f537f933788fd910 Mon Sep 17 00:00:00 2001 From: Yuge Zhang Date: Mon, 21 Jun 2021 17:32:06 +0800 Subject: [PATCH] Revert notebook example to base execution engine (#3852) --- examples/notebooks/Retiarii_example_multi-trial_NAS.ipynb | 7 +++---- .../notebooks/tabular_data_classification_in_AML.ipynb | 3 +-- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/examples/notebooks/Retiarii_example_multi-trial_NAS.ipynb b/examples/notebooks/Retiarii_example_multi-trial_NAS.ipynb index 9394e54251..3c44648082 100644 --- a/examples/notebooks/Retiarii_example_multi-trial_NAS.ipynb +++ b/examples/notebooks/Retiarii_example_multi-trial_NAS.ipynb @@ -55,7 +55,7 @@ "\n", "class Net(nn.Module):\n", " def __init__(self):\n", - " super(Net, self).__init__()\n", + " super().__init__()\n", " self.conv1 = nn.Conv2d(3, 6, 3, padding=1)\n", " self.pool = nn.MaxPool2d(2, 2)\n", " self.conv2 = nn.Conv2d(6, 16, 3, padding=1)\n", @@ -109,12 +109,10 @@ "source": [ "import torch.nn.functional as F\n", "import nni.retiarii.nn.pytorch as nn\n", - "from nni.retiarii import model_wrapper\n", "\n", - "@model_wrapper\n", "class Net(nn.Module):\n", " def __init__(self):\n", - " super(Net, self).__init__()\n", + " super().__init__()\n", " # self.conv1 = nn.Conv2d(3, 6, 3, padding=1)\n", " self.conv1 = nn.LayerChoice([nn.Conv2d(3, 6, 3, padding=1), nn.Conv2d(3, 6, 5, padding=2)])\n", " self.pool = nn.MaxPool2d(2, 2)\n", @@ -304,6 +302,7 @@ "exp_config.max_trial_number = 10\n", "exp_config.trial_gpu_number = 2\n", "exp_config.max_experiment_duration = '5m'\n", + "exp_config.execution_engine = 'base'\n", "exp_config.training_service.use_active_gpu = True" ] }, diff --git a/examples/notebooks/tabular_data_classification_in_AML.ipynb b/examples/notebooks/tabular_data_classification_in_AML.ipynb index 5526639134..5d9127994f 100644 --- a/examples/notebooks/tabular_data_classification_in_AML.ipynb +++ b/examples/notebooks/tabular_data_classification_in_AML.ipynb @@ -127,9 +127,7 @@ "source": [ "import nni.retiarii.nn.pytorch as nn\n", "import torch.nn.functional as F\n", - "from nni.retiarii import model_wrapper\n", "\n", - "@model_wrapper\n", "class Net(nn.Module):\n", "\n", " def __init__(self, input_size):\n", @@ -287,6 +285,7 @@ "exp_config.trial_concurrency = 2\n", "exp_config.max_trial_number = 20\n", "exp_config.max_experiment_duration = '2h'\n", + "exp_config.execution_engine = 'base'\n", "exp_config.nni_manager_ip = '' # your nni_manager_ip" ] },