|
202 | 202 | "# Download config files.\n", |
203 | 203 | "CONFIG_DIR = os.path.join(BUCKET_URI, \"config\")\n", |
204 | 204 | "! wget https://raw.githubusercontent.com/tensorflow/models/master/official/vision/configs/experiments/retinanet/coco_spinenet49_gpu_multiworker_mirrored.yaml\n", |
205 | | - "! gcloud storage cp coco_spinenet49_gpu_multiworker_mirrored.yaml $CONFIG_DIR/\n", |
| 205 | + "! gsutil cp coco_spinenet49_gpu_multiworker_mirrored.yaml $CONFIG_DIR/\n", |
206 | 206 | "\n", |
207 | 207 | "! wget https://raw.githubusercontent.com/tensorflow/models/master/official/vision/configs/experiments/retinanet/coco_spinenet96_gpu_multiworker_mirrored.yaml\n", |
208 | | - "! gcloud storage cp coco_spinenet96_gpu_multiworker_mirrored.yaml $CONFIG_DIR/\n", |
| 208 | + "! gsutil cp coco_spinenet96_gpu_multiworker_mirrored.yaml $CONFIG_DIR/\n", |
209 | 209 | "\n", |
210 | 210 | "! wget https://raw.githubusercontent.com/tensorflow/models/master/official/vision/configs/experiments/retinanet/coco_spinenet143_gpu_multiworker_mirrored.yaml\n", |
211 | | - "! gcloud storage cp coco_spinenet143_gpu_multiworker_mirrored.yaml $CONFIG_DIR/\n", |
| 211 | + "! gsutil cp coco_spinenet143_gpu_multiworker_mirrored.yaml $CONFIG_DIR/\n", |
212 | 212 | "\n", |
213 | 213 | "! wget https://raw.githubusercontent.com/tensorflow/models/master/official/projects/yolo/configs/experiments/yolov4/detection/scaled_yolov4_1280_gpu.yaml\n", |
214 | | - "! gcloud storage cp scaled_yolov4_1280_gpu.yaml $CONFIG_DIR/\n", |
| 214 | + "! gsutil cp scaled_yolov4_1280_gpu.yaml $CONFIG_DIR/\n", |
215 | 215 | "\n", |
216 | 216 | "! wget https://raw.githubusercontent.com/tensorflow/models/master/official/projects/yolo/configs/experiments/yolov7/detection/yolov7_gpu.yaml\n", |
217 | | - "! gcloud storage cp yolov7_gpu.yaml $CONFIG_DIR/" |
| 217 | + "! gsutil cp yolov7_gpu.yaml $CONFIG_DIR/" |
218 | 218 | ] |
219 | 219 | }, |
220 | 220 | { |
|
499 | 499 | " checkpoint_path = os.path.relpath(checkpoint_path, checkpoint_name)\n", |
500 | 500 | " break\n", |
501 | 501 | "\n", |
502 | | - " ! gcloud storage cp --recursive $checkpoint_name $CHECKPOINT_BUCKET/\n", |
| 502 | + " ! gsutil cp -r $checkpoint_name $CHECKPOINT_BUCKET/\n", |
503 | 503 | " checkpoint_uri = os.path.join(CHECKPOINT_BUCKET, checkpoint_name, checkpoint_path)\n", |
504 | 504 | " print(\"Checkpoint uploaded to\", checkpoint_uri)\n", |
505 | 505 | " return checkpoint_uri" |
|
932 | 932 | " serving_container_image_uri=PREDICTION_CONTAINER_URI,\n", |
933 | 933 | " serving_container_args=SERVING_CONTAINER_ARGS,\n", |
934 | 934 | " serving_container_environment_variables=serving_env,\n", |
935 | | - " model_garden_source_model_name=(f\"publishers/google/models/{publisher_model_id}\",),\n", |
| 935 | + " model_garden_source_model_name=(\n", |
| 936 | + " f\"publishers/google/models/{publisher_model_id}\",\n", |
| 937 | + " ),\n", |
936 | 938 | ")\n", |
937 | 939 | "\n", |
938 | 940 | "model.wait()\n", |
|
0 commit comments