Compare commits

...
Author SHA1 Message Date
Andrew Ferlitsch 66bbda6454 debug: verify works 2023-09-22 19:46:23 +00:00
@@ -519,9 +519,9 @@
"outputs": [],
"source": [
"if GPU:\n",
" DEPLOY_VERSION = \"tf2-gpu.2-5\"\n",
" DEPLOY_VERSION = \"tf2-gpu.2-9\"\n",
"else:\n",
" DEPLOY_VERSION = \"tf2-cpu.2-5\"\n",
" DEPLOY_VERSION = \"tf2-cpu.2-9\"\n",
"\n",
"DEPLOY_IMAGE = \"{}-docker.pkg.dev/vertex-ai/prediction/{}:latest\".format(\n",
" REGION.split(\"-\")[0], DEPLOY_VERSION\n",
@@ -558,16 +558,10 @@
},
"outputs": [],
"source": [
"MACHINE_TYPE = \"n1-standard\"\n",
"\n",
"VCPU = \"4\"\n",
"TRAIN_COMPUTE = MACHINE_TYPE + \"-\" + VCPU\n",
"TRAIN_COMPUTE = \"n1-standard-4\"\n",
"print(\"Train machine type\", TRAIN_COMPUTE)\n",
"\n",
"MACHINE_TYPE = \"n1-standard\"\n",
"\n",
"VCPU = \"4\"\n",
"DEPLOY_COMPUTE = MACHINE_TYPE + \"-\" + VCPU\n",
"DEPLOY_COMPUTE = \"n1-standard-4\"\n",
"print(\"Deploy machine type\", DEPLOY_COMPUTE)"
]
},