{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# DeepWeeds on Vertex AI \n",
    "from [GCP codelab](https://codelabs.developers.google.com/vertex_notebook_executor#4)\n",
    "\n",
    "\n",
    "The DeepWeeds dataset consists of 17,509 images capturing eight different weed species native to Australia. In this section, you'll write the code to preprocess the DeepWeeds dataset and build and train an image classification model using feature vectors downloaded from TensorFlow Hub."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "import tensorflow as tf\n",
    "import tensorflow_datasets as tfds\n",
    "import tensorflow_hub as hub"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "2022-01-12 16:45:46.873546: W tensorflow/core/platform/cloud/google_auth_provider.cc:184] All attempts to get a Google authentication bearer token failed, returning an empty token. Retrieving token from files failed with \"Not found: Could not locate the credentials file.\". Retrieving token from GCE failed with \"Failed precondition: Error executing an HTTP request: libcurl code 6 meaning 'Couldn't resolve host name', error details: Could not resolve host: metadata\".\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1mDownloading and preparing dataset 469.32 MiB (download: 469.32 MiB, generated: 469.99 MiB, total: 939.31 MiB) to /Users/johnnydevriese/tensorflow_datasets/deep_weeds/3.0.0...\u001b[0m\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Dl Completed...: 0 url [00:00, ? url/s]\n",
      "Dl Completed...:   0%|          | 0/1 [00:00<?, ? url/s]\n",
      "Dl Completed...:   0%|          | 0/2 [00:00<?, ? url/s]\n",
      "Dl Completed...:   0%|          | 0/2 [00:00<?, ? url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:00<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:13<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:17<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:23<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:24<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:27<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:30<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:35<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:41<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:43<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:43<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:44<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:44<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:45<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:46<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:46<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:47<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:48<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:49<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:50<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:51<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:52<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:54<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:55<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:56<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:56<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:57<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:58<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:58<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [00:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:00<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:01<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:01<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:11<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:13<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:13<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:17<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:17<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:23<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:24<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:27<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:30<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:35<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:41<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:43<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:44<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:45<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:45<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:46<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:47<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:48<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:49<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:49<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:50<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:51<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:52<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:53<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:54<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:54<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:55<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:56<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:57<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:57<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:58<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [01:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:00<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:01<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:01<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:11<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:13<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:17<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:23<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:23<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:24<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:24<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:27<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:30<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:30<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:35<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:35<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:41<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:43<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:44<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:45<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:46<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:46<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:47<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:48<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:49<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:50<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:51<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:51<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:52<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:53<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:54<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:55<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:56<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:57<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:58<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [02:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:00<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:01<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:11<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:13<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:17<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:23<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:24<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:27<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:30<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:35<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:41<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:41<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:43<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:44<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:45<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:46<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:47<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:47<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:48<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:49<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:50<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:50<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:51<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:52<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:53<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:53<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:54<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:55<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:55<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:56<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:56<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:57<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:58<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [03:59<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:00<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:01<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:02<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:03<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:04<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:05<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:06<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:07<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:08<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:09<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:10<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:11<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:11<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:12<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:13<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:14<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:15<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:16<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:17<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:18<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:19<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:20<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:21<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:22<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:23<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:24<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:25<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:26<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:27<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:27<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:28<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:29<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:30<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:31<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:32<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:33<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:34<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:35<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:36<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:37<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:38<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:39<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:40<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:41<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:42<00:00,  1.41 url/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:44<00:00,  1.41 url/s]\n",
      "Extraction completed...: 0 file [04:44, ? file/s]\n",
      "Dl Size...:  84%|████████▍ | 392/468 [04:44<00:55,  1.38 MiB/s]\n",
      "Dl Completed...:  50%|█████     | 1/2 [04:44<04:44, 284.38s/ url]\n"
     ]
    },
    {
     "ename": "KeyboardInterrupt",
     "evalue": "",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
      "\u001b[0;32m/var/folders/4k/y4ljh2217c57vl68z1zkl0440000gn/T/ipykernel_82112/1007210419.py\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0mdata\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0minfo\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mtfds\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mload\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mname\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;34m'deep_weeds'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mas_supervised\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mTrue\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mwith_info\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mTrue\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      2\u001b[0m \u001b[0mNUM_CLASSES\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0minfo\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfeatures\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'label'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mnum_classes\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      3\u001b[0m \u001b[0mDATASET_SIZE\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0minfo\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0msplits\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'train'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mnum_examples\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/core/load.py\u001b[0m in \u001b[0;36mload\u001b[0;34m(name, split, data_dir, batch_size, shuffle_files, download, as_supervised, decoders, read_config, with_info, builder_kwargs, download_and_prepare_kwargs, as_dataset_kwargs, try_gcs)\u001b[0m\n\u001b[1;32m    331\u001b[0m   \u001b[0;32mif\u001b[0m \u001b[0mdownload\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    332\u001b[0m     \u001b[0mdownload_and_prepare_kwargs\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mdownload_and_prepare_kwargs\u001b[0m \u001b[0;32mor\u001b[0m \u001b[0;34m{\u001b[0m\u001b[0;34m}\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 333\u001b[0;31m     \u001b[0mdbuilder\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mdownload_and_prepare\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m**\u001b[0m\u001b[0mdownload_and_prepare_kwargs\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    334\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    335\u001b[0m   \u001b[0;32mif\u001b[0m \u001b[0mas_dataset_kwargs\u001b[0m \u001b[0;32mis\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/core/dataset_builder.py\u001b[0m in \u001b[0;36mdownload_and_prepare\u001b[0;34m(self, download_dir, download_config)\u001b[0m\n\u001b[1;32m    437\u001b[0m           \u001b[0;31m# Old version of TF are not os.PathLike compatible\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    438\u001b[0m           \u001b[0;32mwith\u001b[0m \u001b[0mtf_compat\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmock_gfile_pathlike\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 439\u001b[0;31m             self._download_and_prepare(\n\u001b[0m\u001b[1;32m    440\u001b[0m                 \u001b[0mdl_manager\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mdl_manager\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    441\u001b[0m                 \u001b[0mdownload_config\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mdownload_config\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/core/dataset_builder.py\u001b[0m in \u001b[0;36m_download_and_prepare\u001b[0;34m(self, dl_manager, download_config)\u001b[0m\n\u001b[1;32m   1131\u001b[0m       \u001b[0;32melse\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m   1132\u001b[0m         \u001b[0moptional_pipeline_kwargs\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34m{\u001b[0m\u001b[0;34m}\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m-> 1133\u001b[0;31m       split_generators = self._split_generators(  # pylint: disable=unexpected-keyword-arg\n\u001b[0m\u001b[1;32m   1134\u001b[0m           dl_manager, **optional_pipeline_kwargs)\n\u001b[1;32m   1135\u001b[0m       \u001b[0;31m# TODO(tfds): Could be removed once all datasets are migrated.\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/image_classification/deep_weeds.py\u001b[0m in \u001b[0;36m_split_generators\u001b[0;34m(self, dl_manager)\u001b[0m\n\u001b[1;32m     89\u001b[0m   \u001b[0;32mdef\u001b[0m \u001b[0m_split_generators\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mdl_manager\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     90\u001b[0m     \u001b[0;34m\"\"\"Define Splits.\"\"\"\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 91\u001b[0;31m     paths = dl_manager.download_and_extract({\n\u001b[0m\u001b[1;32m     92\u001b[0m         \u001b[0;34m\"image\"\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0m_URL\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     93\u001b[0m         \"label\": _URL_LABELS})\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/core/download/download_manager.py\u001b[0m in \u001b[0;36mdownload_and_extract\u001b[0;34m(self, url_or_urls)\u001b[0m\n\u001b[1;32m    635\u001b[0m     \u001b[0;32mwith\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_downloader\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mtqdm\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    636\u001b[0m       \u001b[0;32mwith\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_extractor\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mtqdm\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 637\u001b[0;31m         \u001b[0;32mreturn\u001b[0m \u001b[0m_map_promise\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_download_extract\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0murl_or_urls\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    638\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    639\u001b[0m   \u001b[0;34m@\u001b[0m\u001b[0mproperty\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/core/download/download_manager.py\u001b[0m in \u001b[0;36m_map_promise\u001b[0;34m(map_fn, all_inputs)\u001b[0m\n\u001b[1;32m    776\u001b[0m   \u001b[0;34m\"\"\"Map the function into each element and resolve the promise.\"\"\"\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    777\u001b[0m   \u001b[0mall_promises\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mtf\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mnest\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmap_structure\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mmap_fn\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mall_inputs\u001b[0m\u001b[0;34m)\u001b[0m  \u001b[0;31m# Apply the function\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 778\u001b[0;31m   \u001b[0mres\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mtf\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mnest\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmap_structure\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;32mlambda\u001b[0m \u001b[0mp\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mp\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mget\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mall_promises\u001b[0m\u001b[0;34m)\u001b[0m  \u001b[0;31m# Wait promises\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    779\u001b[0m   \u001b[0;32mreturn\u001b[0m \u001b[0mres\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow/python/util/nest.py\u001b[0m in \u001b[0;36mmap_structure\u001b[0;34m(func, *structure, **kwargs)\u001b[0m\n\u001b[1;32m    867\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    868\u001b[0m   return pack_sequence_as(\n\u001b[0;32m--> 869\u001b[0;31m       \u001b[0mstructure\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;36m0\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m[\u001b[0m\u001b[0mfunc\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m*\u001b[0m\u001b[0mx\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mfor\u001b[0m \u001b[0mx\u001b[0m \u001b[0;32min\u001b[0m \u001b[0mentries\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    870\u001b[0m       expand_composites=expand_composites)\n\u001b[1;32m    871\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow/python/util/nest.py\u001b[0m in \u001b[0;36m<listcomp>\u001b[0;34m(.0)\u001b[0m\n\u001b[1;32m    867\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    868\u001b[0m   return pack_sequence_as(\n\u001b[0;32m--> 869\u001b[0;31m       \u001b[0mstructure\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;36m0\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m[\u001b[0m\u001b[0mfunc\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m*\u001b[0m\u001b[0mx\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mfor\u001b[0m \u001b[0mx\u001b[0m \u001b[0;32min\u001b[0m \u001b[0mentries\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    870\u001b[0m       expand_composites=expand_composites)\n\u001b[1;32m    871\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/tensorflow_datasets/core/download/download_manager.py\u001b[0m in \u001b[0;36m<lambda>\u001b[0;34m(p)\u001b[0m\n\u001b[1;32m    776\u001b[0m   \u001b[0;34m\"\"\"Map the function into each element and resolve the promise.\"\"\"\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    777\u001b[0m   \u001b[0mall_promises\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mtf\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mnest\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmap_structure\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mmap_fn\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mall_inputs\u001b[0m\u001b[0;34m)\u001b[0m  \u001b[0;31m# Apply the function\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 778\u001b[0;31m   \u001b[0mres\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mtf\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mnest\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmap_structure\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;32mlambda\u001b[0m \u001b[0mp\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mp\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mget\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mall_promises\u001b[0m\u001b[0;34m)\u001b[0m  \u001b[0;31m# Wait promises\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    779\u001b[0m   \u001b[0;32mreturn\u001b[0m \u001b[0mres\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/promise/promise.py\u001b[0m in \u001b[0;36mget\u001b[0;34m(self, timeout)\u001b[0m\n\u001b[1;32m    509\u001b[0m         \u001b[0;31m# type: (Optional[float]) -> T\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    510\u001b[0m         \u001b[0mtarget\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_target\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 511\u001b[0;31m         \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_wait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mtimeout\u001b[0m \u001b[0;32mor\u001b[0m \u001b[0mDEFAULT_TIMEOUT\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    512\u001b[0m         \u001b[0;32mreturn\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_target_settled_value\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0m_raise\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mTrue\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    513\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/promise/promise.py\u001b[0m in \u001b[0;36m_wait\u001b[0;34m(self, timeout)\u001b[0m\n\u001b[1;32m    504\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m_wait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mNone\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    505\u001b[0m         \u001b[0;31m# type: (Optional[float]) -> None\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 506\u001b[0;31m         \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    507\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    508\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0mget\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mNone\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/promise/promise.py\u001b[0m in \u001b[0;36mwait\u001b[0;34m(cls, promise, timeout)\u001b[0m\n\u001b[1;32m    500\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0mwait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mcls\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mpromise\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mNone\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    501\u001b[0m         \u001b[0;31m# type: (Promise, Optional[float]) -> None\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 502\u001b[0;31m         \u001b[0masync_instance\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mpromise\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    503\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    504\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m_wait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mNone\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/promise/async_.py\u001b[0m in \u001b[0;36mwait\u001b[0;34m(self, promise, timeout)\u001b[0m\n\u001b[1;32m    115\u001b[0m                 \u001b[0;31m# fulfilled or rejected\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    116\u001b[0m                 \u001b[0;32mreturn\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 117\u001b[0;31m         \u001b[0mtarget\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mscheduler\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mtarget\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtimeout\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    118\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    119\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0mdrain_queues\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/site-packages/promise/schedulers/immediate.py\u001b[0m in \u001b[0;36mwait\u001b[0;34m(self, promise, timeout)\u001b[0m\n\u001b[1;32m     23\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     24\u001b[0m         \u001b[0mpromise\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_then\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mon_resolve_or_reject\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mon_resolve_or_reject\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 25\u001b[0;31m         \u001b[0mwaited\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0me\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mtimeout\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     26\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0mwaited\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     27\u001b[0m             \u001b[0;32mraise\u001b[0m \u001b[0mException\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m\"Timeout\"\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/threading.py\u001b[0m in \u001b[0;36mwait\u001b[0;34m(self, timeout)\u001b[0m\n\u001b[1;32m    556\u001b[0m             \u001b[0msignaled\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_flag\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    557\u001b[0m             \u001b[0;32mif\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0msignaled\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 558\u001b[0;31m                 \u001b[0msignaled\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_cond\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwait\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mtimeout\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    559\u001b[0m             \u001b[0;32mreturn\u001b[0m \u001b[0msignaled\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    560\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m~/miniforge3/envs/pytorch_m1/lib/python3.8/threading.py\u001b[0m in \u001b[0;36mwait\u001b[0;34m(self, timeout)\u001b[0m\n\u001b[1;32m    300\u001b[0m         \u001b[0;32mtry\u001b[0m\u001b[0;34m:\u001b[0m    \u001b[0;31m# restore state no matter what (e.g., KeyboardInterrupt)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    301\u001b[0m             \u001b[0;32mif\u001b[0m \u001b[0mtimeout\u001b[0m \u001b[0;32mis\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 302\u001b[0;31m                 \u001b[0mwaiter\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0macquire\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    303\u001b[0m                 \u001b[0mgotit\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mTrue\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    304\u001b[0m             \u001b[0;32melse\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mKeyboardInterrupt\u001b[0m: "
     ]
    }
   ],
   "source": [
    "data, info = tfds.load(name='deep_weeds', as_supervised=True, with_info=True)\n",
    "NUM_CLASSES = info.features['label'].num_classes\n",
    "DATASET_SIZE = info.splits['train'].num_examples"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def preprocess_data(image, label):\n",
    "  image = tf.image.resize(image, (300,300))\n",
    "  return tf.cast(image, tf.float32) / 255., label"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Create train/validation splits\n",
    "\n",
    "# Shuffle dataset\n",
    "dataset = data['train'].shuffle(1000)\n",
    "\n",
    "train_split = 0.8\n",
    "val_split = 0.2\n",
    "train_size = int(train_split * DATASET_SIZE)\n",
    "val_size = int(val_split * DATASET_SIZE)\n",
    "\n",
    "train_data = dataset.take(train_size)\n",
    "train_data  = train_data.map(preprocess_data)\n",
    "train_data  = train_data.batch(64)\n",
    "\n",
    "validation_data = dataset.skip(train_size)\n",
    "validation_data  = validation_data.map(preprocess_data)\n",
    "validation_data  = validation_data.batch(64)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "feature_extractor_model = \"inception_v3\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "tf_hub_uri = f\"https://tfhub.dev/google/imagenet/{feature_extractor_model}/feature_vector/5\"\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "feature_extractor_layer = hub.KerasLayer(\n",
    "    tf_hub_uri,\n",
    "    trainable=False)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "model = tf.keras.Sequential([\n",
    "  feature_extractor_layer,\n",
    "  tf.keras.layers.Dense(units=NUM_CLASSES)\n",
    "])\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "model.compile(\n",
    "  optimizer=tf.keras.optimizers.Adam(),\n",
    "  loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),\n",
    "  metrics=['acc'])\n",
    "\n",
    "model.fit(train_data, validation_data=validation_data, epochs=20)\n"
   ]
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "b7e818f66e33c31ac0526ee7f8556503ff93918b8b22809241939dc19e90de0b"
  },
  "kernelspec": {
   "display_name": "Python 3.8.12 64-bit ('pytorch_m1': conda)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.12"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}