Merge pull request #1763 from Azure/release_update/Release-144

update samples from Release-144 as a part of  SDK release
This commit is contained in:
Harneet Virk
2022-06-03 11:04:13 -07:00
committed by GitHub
4 changed files with 16 additions and 10 deletions

View File

@@ -17,6 +17,7 @@ dependencies:
- notebook - notebook
- pywin32==227 - pywin32==227
- PySocks==1.7.1 - PySocks==1.7.1
- jsonschema==4.5.1
- conda-forge::pyqt==5.12.3 - conda-forge::pyqt==5.12.3
- pip: - pip:

View File

@@ -361,7 +361,7 @@
"\n", "\n",
"batch_conda_deps = CondaDependencies.create(python_version=\"3.7\",\n", "batch_conda_deps = CondaDependencies.create(python_version=\"3.7\",\n",
" conda_packages=['pip==20.2.4'],\n", " conda_packages=['pip==20.2.4'],\n",
" pip_packages=[\"tensorflow==1.15.2\", \"pillow\", \n", " pip_packages=[\"tensorflow==1.15.2\", \"pillow\", \"protobuf==3.20.1\",\n",
" \"azureml-core\", \"azureml-dataset-runtime[fuse]\"])\n", " \"azureml-core\", \"azureml-dataset-runtime[fuse]\"])\n",
"batch_env = Environment(name=\"batch_environment\")\n", "batch_env = Environment(name=\"batch_environment\")\n",
"batch_env.python.conda_dependencies = batch_conda_deps\n", "batch_env.python.conda_dependencies = batch_conda_deps\n",

View File

@@ -5,17 +5,19 @@ import os
import argparse import argparse
import datetime import datetime
import time import time
import tensorflow as tf import tensorflow.compat.v1 as tf
from math import ceil from math import ceil
import numpy as np import numpy as np
import sys
import shutil import shutil
from tensorflow.contrib.slim.python.slim.nets import inception_v3 import subprocess
import tf_slim
from azureml.core import Run from azureml.core import Run
from azureml.core.model import Model from azureml.core.model import Model
from azureml.core.dataset import Dataset from azureml.core.dataset import Dataset
slim = tf.contrib.slim slim = tf_slim
image_size = 299 image_size = 299
num_channel = 3 num_channel = 3
@@ -32,16 +34,18 @@ def get_class_label_dict(labels_dir):
def init(): def init():
global g_tf_sess, probabilities, label_dict, input_images global g_tf_sess, probabilities, label_dict, input_images
subprocess.run(["git", "clone", "https://github.com/tensorflow/models/"])
sys.path.append("./models/research/slim")
parser = argparse.ArgumentParser(description="Start a tensorflow model serving") parser = argparse.ArgumentParser(description="Start a tensorflow model serving")
parser.add_argument('--model_name', dest="model_name", required=True) parser.add_argument('--model_name', dest="model_name", required=True)
parser.add_argument('--labels_dir', dest="labels_dir", required=True) parser.add_argument('--labels_dir', dest="labels_dir", required=True)
args, _ = parser.parse_known_args() args, _ = parser.parse_known_args()
from nets import inception_v3, inception_utils
label_dict = get_class_label_dict(args.labels_dir) label_dict = get_class_label_dict(args.labels_dir)
classes_num = len(label_dict) classes_num = len(label_dict)
tf.disable_v2_behavior()
with slim.arg_scope(inception_v3.inception_v3_arg_scope()): with slim.arg_scope(inception_utils.inception_arg_scope()):
input_images = tf.placeholder(tf.float32, [1, image_size, image_size, num_channel]) input_images = tf.placeholder(tf.float32, [1, image_size, image_size, num_channel])
logits, _ = inception_v3.inception_v3(input_images, logits, _ = inception_v3.inception_v3(input_images,
num_classes=classes_num, num_classes=classes_num,

View File

@@ -247,7 +247,7 @@
" config = AmlCompute.provisioning_configuration(vm_size=\"STANDARD_NC6\",\n", " config = AmlCompute.provisioning_configuration(vm_size=\"STANDARD_NC6\",\n",
" vm_priority=\"lowpriority\", \n", " vm_priority=\"lowpriority\", \n",
" min_nodes=0, \n", " min_nodes=0, \n",
" max_nodes=1)\n", " max_nodes=2)\n",
"\n", "\n",
" compute_target = ComputeTarget.create(workspace=ws, name=compute_name, provisioning_configuration=config)\n", " compute_target = ComputeTarget.create(workspace=ws, name=compute_name, provisioning_configuration=config)\n",
" compute_target.wait_for_completion(show_output=True, min_node_count=None, timeout_in_minutes=20)" " compute_target.wait_for_completion(show_output=True, min_node_count=None, timeout_in_minutes=20)"
@@ -305,9 +305,10 @@
"from azureml.core.conda_dependencies import CondaDependencies\n", "from azureml.core.conda_dependencies import CondaDependencies\n",
"from azureml.core.runconfig import DEFAULT_GPU_IMAGE\n", "from azureml.core.runconfig import DEFAULT_GPU_IMAGE\n",
"\n", "\n",
"cd = CondaDependencies.create(python_version=\"3.7\",\n", "cd = CondaDependencies.create(python_version=\"3.8\",\n",
" conda_packages=['pip==20.2.4'],\n", " conda_packages=['pip==20.2.4'],\n",
" pip_packages=[\"tensorflow-gpu==1.15.2\",\n", " pip_packages=[\"tensorflow-gpu==2.3.0\",\n",
" \"tf_slim==1.1.0\", \"protobuf==3.20.1\",\n",
" \"azureml-core\", \"azureml-dataset-runtime[fuse]\"])\n", " \"azureml-core\", \"azureml-dataset-runtime[fuse]\"])\n",
"\n", "\n",
"env = Environment(name=\"parallelenv\")\n", "env = Environment(name=\"parallelenv\")\n",