From d09e2711ed77c734ed0f43004376314523bca6cc Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Tue, 1 Nov 2022 06:01:15 -1000 Subject: [PATCH] Created using Colaboratory --- ddpm_flax_oxford102_end_to_end.ipynb | 77 ++++++++++++++-------------- 1 file changed, 38 insertions(+), 39 deletions(-) diff --git a/ddpm_flax_oxford102_end_to_end.ipynb b/ddpm_flax_oxford102_end_to_end.ipynb index dbe46b7..8fe4e73 100644 --- a/ddpm_flax_oxford102_end_to_end.ipynb +++ b/ddpm_flax_oxford102_end_to_end.ipynb @@ -6,7 +6,7 @@ "provenance": [], "collapsed_sections": [], "toc_visible": true, - "authorship_tag": "ABX9TyN7bltIi79pLsB8LyJDsBKH", + "authorship_tag": "ABX9TyOSW8m9jKq/HKH1ZEsif0Ih", "include_colab_link": true }, "kernelspec": { @@ -334,7 +334,7 @@ "metadata": { "id": "x1kiybCWozzm" }, - "execution_count": 1, + "execution_count": null, "outputs": [] }, { @@ -359,7 +359,7 @@ }, "outputId": "68cfbed0-f63f-4e05-cfdc-63b3a4982623" }, - "execution_count": 2, + "execution_count": null, "outputs": [ { "output_type": "stream", @@ -379,7 +379,7 @@ "metadata": { "id": "WkKXVdDOsPAm" }, - "execution_count": 3, + "execution_count": null, "outputs": [] }, { @@ -392,7 +392,7 @@ "metadata": { "id": "iltT-y3nmP6u" }, - "execution_count": 4, + "execution_count": null, "outputs": [] }, { @@ -415,7 +415,7 @@ "metadata": { "id": "E-erXgLkURJn" }, - "execution_count": 5, + "execution_count": null, "outputs": [] }, { @@ -428,7 +428,7 @@ "metadata": { "id": "MtYNTW38NVy0" }, - "execution_count": 6, + "execution_count": null, "outputs": [] }, { @@ -489,7 +489,7 @@ "id": "S2qNLrIsQGdj", "outputId": "9ab7185a-dc39-4cc5-dbdd-1ed84c589e60" }, - "execution_count": 7, + "execution_count": null, "outputs": [ { "output_type": "stream", @@ -538,7 +538,7 @@ "id": "q8ZhvAmgSm1a", "outputId": "13e0f773-010a-4397-f256-27d31c3694c6" }, - "execution_count": 8, + "execution_count": null, "outputs": [ { "output_type": "stream", @@ -562,8 +562,6 @@ "cell_type": "markdown", "source": [ "* If you wish to use TPU, please change your runtime type to TPU and check the box below \n", - " * Note that __TPU colab is NOT working correctly__ at the moment, see issue here https://github.com/googlecolab/colabtools/issues/3009\n", - " * *(I will remove this comment once the issue is resolved)*\n", "\n", "* If you don't use TPU, please change your runtime type to GPU - don't use CPU to train the model!\n" ], @@ -579,7 +577,8 @@ "use_tpu = False #@param {type:\"boolean\"}\n", "if use_tpu:\n", " import jax.tools.colab_tpu\n", - " jax.tools.colab_tpu.setup_tpu()" + " jax.tools.colab_tpu.setup_tpu()\n", + " print(f\"{jax.device_count()} devices: {jax.devices()}\")" ], "metadata": { "cellView": "form", @@ -611,7 +610,7 @@ "metadata": { "id": "IEBP9yb9TdV7" }, - "execution_count": 9, + "execution_count": null, "outputs": [] }, { @@ -630,7 +629,7 @@ "metadata": { "id": "24MSzHTyrkIU" }, - "execution_count": 10, + "execution_count": null, "outputs": [] }, { @@ -668,7 +667,7 @@ "metadata": { "id": "Y7w-WXiLVZVI" }, - "execution_count": 47, + "execution_count": null, "outputs": [] }, { @@ -706,7 +705,7 @@ "id": "YkO8kIzkWf8r", "outputId": "becde2e9-62ee-413c-e5b1-a1dc6c9b4ed7" }, - "execution_count": 44, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -746,7 +745,7 @@ }, "outputId": "15de9c94-3213-447c-dc20-b92ee1577c51" }, - "execution_count": 48, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -784,7 +783,7 @@ "id": "nIqI8-y1Aw-S", "outputId": "e1c219b2-c0d1-4e70-cff1-77a5d5c2ef8b" }, - "execution_count": 49, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -808,7 +807,7 @@ "metadata": { "id": "F-yjxOtPqFcx" }, - "execution_count": 50, + "execution_count": null, "outputs": [] }, { @@ -834,7 +833,7 @@ "id": "hPhlgV3Vy4oi", "outputId": "3955c584-fd5b-47a5-92ac-7869f1bf48ba" }, - "execution_count": 51, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -861,7 +860,7 @@ "id": "nD-cp8RSqPam", "outputId": "847b2750-171e-420a-8416-46e9fc28bcbd" }, - "execution_count": 52, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -906,7 +905,7 @@ "id": "o7VKVIWfqPd5", "outputId": "5e900833-b2c1-4113-e49a-cc4b786b7122" }, - "execution_count": 53, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -974,7 +973,7 @@ "id": "Wp2XjgMCqPhJ", "outputId": "06325663-93da-42b7-e608-f4c298520270" }, - "execution_count": 54, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1000,7 +999,7 @@ "id": "3UV4iIdkqPkw", "outputId": "125ffd4b-3af5-4d10-fc02-1c8ee9a98413" }, - "execution_count": 55, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1031,7 +1030,7 @@ "metadata": { "id": "fDr1zZ9BnImb" }, - "execution_count": 56, + "execution_count": null, "outputs": [] }, { @@ -1048,7 +1047,7 @@ "id": "yarvSCsdqN_G", "outputId": "6543fcd6-ad83-43c6-f89f-a6b931fa969c" }, - "execution_count": 57, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1100,7 +1099,7 @@ "id": "w7k2j3tOldpn", "outputId": "4983eb1f-dd4a-4728-fa63-3f86e30f1023" }, - "execution_count": 58, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1508,7 +1507,7 @@ "metadata": { "id": "bv3Z4sqYvxgo" }, - "execution_count": 11, + "execution_count": null, "outputs": [] }, { @@ -1523,7 +1522,7 @@ "id": "zhMOAo23yfTf", "outputId": "371fae67-6df9-4f1f-caa7-a98bb8b86fae" }, - "execution_count": 13, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1658,7 +1657,7 @@ }, "outputId": "2286a813-c83f-4658-b7e3-6680dbe58a7e" }, - "execution_count": 15, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1691,7 +1690,7 @@ }, "outputId": "6a0e9aa6-d58a-46a0-ca97-0f97888095f2" }, - "execution_count": 16, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1721,7 +1720,7 @@ "id": "StiVH49ZyexN", "outputId": "378a9182-07f5-4309-9830-d8d01e8d7f5a" }, - "execution_count": 17, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1756,7 +1755,7 @@ "metadata": { "id": "VUmIO_7Y7cs5" }, - "execution_count": 26, + "execution_count": null, "outputs": [] }, { @@ -1771,7 +1770,7 @@ "metadata": { "id": "wBjpDd444Mvi" }, - "execution_count": 34, + "execution_count": null, "outputs": [] }, { @@ -1782,7 +1781,7 @@ "metadata": { "id": "VrpQFK4I8xny" }, - "execution_count": 35, + "execution_count": null, "outputs": [] }, { @@ -1802,7 +1801,7 @@ "id": "IQb6Dxhy9-ge", "outputId": "a008648d-32e1-4eda-bc91-814ff2f34d0d" }, - "execution_count": 36, + "execution_count": null, "outputs": [ { "output_type": "execute_result", @@ -1854,7 +1853,7 @@ "id": "oUd9pD8s2-ly", "outputId": "3b95da36-6286-4315-e985-652c633ee918" }, - "execution_count": 37, + "execution_count": null, "outputs": [ { "output_type": "display_data", @@ -2008,7 +2007,7 @@ "id": "FSpAuBWF_i1y", "outputId": "c9d02b00-e906-49de-932c-be4f487bec71" }, - "execution_count": 41, + "execution_count": null, "outputs": [ { "output_type": "stream", @@ -2039,7 +2038,7 @@ "id": "UeRE3dijMrGc", "outputId": "170bd4a6-6feb-4e3e-a133-1042f915d603" }, - "execution_count": 42, + "execution_count": null, "outputs": [ { "output_type": "execute_result",