Update Text Classification using PyTorch runtime version
"source": [
"MODEL_NAME='torch_text_classification'\n",
"VERSION_NAME='v201903'\n",
- "RUNTIME_VERSION='1.13'\n",
+ "RUNTIME_VERSION='1.12'\n",
"REGION='us-central1'"
]
},
"source": [
"!gcloud beta ai-platform versions create {VERSION_NAME} --model {MODEL_NAME} \\\n",
"--origin=gs://{BUCKET}/{MODEL_DIR}/ \\\n",
- "--python-version=3.5 \\\n",
+ "--python-version=2.7 \\\n",
"--runtime-version={RUNTIME_VERSION} \\\n",
"--framework='SCIKIT_LEARN' \\\n",
"--package-uris=gs://{BUCKET}/{PACKAGES_DIR}/my_package-0.1.tar.gz \\\n",
