From 311a0784a290443e26a0d68a2594dba9630c386a Mon Sep 17 00:00:00 2001 From: or-toledano <49984106+or-toledano@users.noreply.github.com> Date: Mon, 9 Aug 2021 08:26:41 +0300 Subject: [PATCH] Reduce size of diff #2 --- .../Prompt_Engineering_for_ImageNet.ipynb | 60 +++++++++---------- 1 file changed, 30 insertions(+), 30 deletions(-) diff --git a/notebooks/Prompt_Engineering_for_ImageNet.ipynb b/notebooks/Prompt_Engineering_for_ImageNet.ipynb index d30ccfe..86eae81 100644 --- a/notebooks/Prompt_Engineering_for_ImageNet.ipynb +++ b/notebooks/Prompt_Engineering_for_ImageNet.ipynb @@ -713,7 +713,7 @@ "output_type": "execute_result", "data": { "text/plain": [ - "['RN50', 'ViT-f']" + "['RN50', 'ViT-B/32']" ] }, "metadata": { @@ -733,43 +733,43 @@ "outputId": "58e644d4-6e23-43b5-964e-1e9e8540d22e" }, "source": [ - "jit = True\n", - "model, preprocess = clip.load(\"ViT-B/32\", jit=jit)" + "jit = True\n", + "model, preprocess = clip.load(\"ViT-B/32\", jit=jit)" ], "execution_count": 6, "outputs": [ - { - "output_type": "stream", - "text": [ - "100%|██████████████████████| 353976522/353976522 [00:01<00:00, 188872424.30it/s]\n" - ], - "name": "stderr" - } + { + "output_type": "stream", + "text": [ + "100%|██████████████████████| 353976522/353976522 [00:01<00:00, 188872424.30it/s]\n" + ], + "name": "stderr" + } ] - }, - { + }, + { "cell_type": "code", "metadata": { - "colab": { - "base_uri": "https://localhost:8080/" - }, - "id": "IBRVTY9lbGm8", - "outputId": "58641dc2-919d-40ae-b71a-7b7b47830f77" + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "IBRVTY9lbGm8", + "outputId": "58641dc2-919d-40ae-b71a-7b7b47830f77" }, "source": [ - "if jit:\n", - " input_resolution = model.input_resolution.item()\n", - " context_length = model.context_length.item()\n", - " vocab_size = model.vocab_size.item()\n", - "else:\n", - " input_resolution = model.visual.input_resolution\n", - " context_length = model.context_length\n", - " vocab_size = model.vocab_size\n", - "\n", - "print(\"Model parameters:\", f\"{np.sum([int(np.prod(p.shape)) for p in model.parameters()]):,}\")\n", - "print(\"Input resolution:\", input_resolution)\n", - "print(\"Context length:\", context_length)\n", - "print(\"Vocab size:\", vocab_size)" + "if jit:\n", + " input_resolution = model.input_resolution.item()\n", + " context_length = model.context_length.item()\n", + " vocab_size = model.vocab_size.item()\n", + "else:\n", + " input_resolution = model.visual.input_resolution\n", + " context_length = model.context_length\n", + " vocab_size = model.vocab_size\n", + "\n", + "print(\"Model parameters:\", f\"{np.sum([int(np.prod(p.shape)) for p in model.parameters()]):,}\")\n", + "print(\"Input resolution:\", input_resolution)\n", + "print(\"Context length:\", context_length)\n", + "print(\"Vocab size:\", vocab_size)" ], "execution_count": 7, "outputs": [