Skip to content

Commit

Permalink
Created using Colaboratory
Browse files Browse the repository at this point in the history
  • Loading branch information
@yiyixuxu
yiyixuxu committed Nov 1, 2022
1 parent c6aab5a commit d09e271
Showing 1 changed file with 38 additions and 39 deletions.
77 changes: 38 additions & 39 deletions ddpm_flax_oxford102_end_to_end.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
"provenance": [],
"collapsed_sections": [],
"toc_visible": true,
"authorship_tag": "ABX9TyN7bltIi79pLsB8LyJDsBKH",
"authorship_tag": "ABX9TyOSW8m9jKq/HKH1ZEsif0Ih",
"include_colab_link": true
},
"kernelspec": {
Expand Down Expand Up @@ -334,7 +334,7 @@
"metadata": {
"id": "x1kiybCWozzm"
},
"execution_count": 1,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -359,7 +359,7 @@
},
"outputId": "68cfbed0-f63f-4e05-cfdc-63b3a4982623"
},
"execution_count": 2,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
Expand All @@ -379,7 +379,7 @@
"metadata": {
"id": "WkKXVdDOsPAm"
},
"execution_count": 3,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -392,7 +392,7 @@
"metadata": {
"id": "iltT-y3nmP6u"
},
"execution_count": 4,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -415,7 +415,7 @@
"metadata": {
"id": "E-erXgLkURJn"
},
"execution_count": 5,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -428,7 +428,7 @@
"metadata": {
"id": "MtYNTW38NVy0"
},
"execution_count": 6,
"execution_count": null,
"outputs": []
},
{
Expand Down Expand Up @@ -489,7 +489,7 @@
"id": "S2qNLrIsQGdj",
"outputId": "9ab7185a-dc39-4cc5-dbdd-1ed84c589e60"
},
"execution_count": 7,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
Expand Down Expand Up @@ -538,7 +538,7 @@
"id": "q8ZhvAmgSm1a",
"outputId": "13e0f773-010a-4397-f256-27d31c3694c6"
},
"execution_count": 8,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
Expand All @@ -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 <font color=red>__TPU colab is NOT working correctly__ </font> 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"
],
Expand All @@ -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",
Expand Down Expand Up @@ -611,7 +610,7 @@
"metadata": {
"id": "IEBP9yb9TdV7"
},
"execution_count": 9,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -630,7 +629,7 @@
"metadata": {
"id": "24MSzHTyrkIU"
},
"execution_count": 10,
"execution_count": null,
"outputs": []
},
{
Expand Down Expand Up @@ -668,7 +667,7 @@
"metadata": {
"id": "Y7w-WXiLVZVI"
},
"execution_count": 47,
"execution_count": null,
"outputs": []
},
{
Expand Down Expand Up @@ -706,7 +705,7 @@
"id": "YkO8kIzkWf8r",
"outputId": "becde2e9-62ee-413c-e5b1-a1dc6c9b4ed7"
},
"execution_count": 44,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -746,7 +745,7 @@
},
"outputId": "15de9c94-3213-447c-dc20-b92ee1577c51"
},
"execution_count": 48,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -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",
Expand All @@ -808,7 +807,7 @@
"metadata": {
"id": "F-yjxOtPqFcx"
},
"execution_count": 50,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -834,7 +833,7 @@
"id": "hPhlgV3Vy4oi",
"outputId": "3955c584-fd5b-47a5-92ac-7869f1bf48ba"
},
"execution_count": 51,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand All @@ -861,7 +860,7 @@
"id": "nD-cp8RSqPam",
"outputId": "847b2750-171e-420a-8416-46e9fc28bcbd"
},
"execution_count": 52,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -906,7 +905,7 @@
"id": "o7VKVIWfqPd5",
"outputId": "5e900833-b2c1-4113-e49a-cc4b786b7122"
},
"execution_count": 53,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -974,7 +973,7 @@
"id": "Wp2XjgMCqPhJ",
"outputId": "06325663-93da-42b7-e608-f4c298520270"
},
"execution_count": 54,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand All @@ -1000,7 +999,7 @@
"id": "3UV4iIdkqPkw",
"outputId": "125ffd4b-3af5-4d10-fc02-1c8ee9a98413"
},
"execution_count": 55,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1031,7 +1030,7 @@
"metadata": {
"id": "fDr1zZ9BnImb"
},
"execution_count": 56,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -1048,7 +1047,7 @@
"id": "yarvSCsdqN_G",
"outputId": "6543fcd6-ad83-43c6-f89f-a6b931fa969c"
},
"execution_count": 57,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1100,7 +1099,7 @@
"id": "w7k2j3tOldpn",
"outputId": "4983eb1f-dd4a-4728-fa63-3f86e30f1023"
},
"execution_count": 58,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1508,7 +1507,7 @@
"metadata": {
"id": "bv3Z4sqYvxgo"
},
"execution_count": 11,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -1523,7 +1522,7 @@
"id": "zhMOAo23yfTf",
"outputId": "371fae67-6df9-4f1f-caa7-a98bb8b86fae"
},
"execution_count": 13,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1658,7 +1657,7 @@
},
"outputId": "2286a813-c83f-4658-b7e3-6680dbe58a7e"
},
"execution_count": 15,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1691,7 +1690,7 @@
},
"outputId": "6a0e9aa6-d58a-46a0-ca97-0f97888095f2"
},
"execution_count": 16,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1721,7 +1720,7 @@
"id": "StiVH49ZyexN",
"outputId": "378a9182-07f5-4309-9830-d8d01e8d7f5a"
},
"execution_count": 17,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1756,7 +1755,7 @@
"metadata": {
"id": "VUmIO_7Y7cs5"
},
"execution_count": 26,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -1771,7 +1770,7 @@
"metadata": {
"id": "wBjpDd444Mvi"
},
"execution_count": 34,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -1782,7 +1781,7 @@
"metadata": {
"id": "VrpQFK4I8xny"
},
"execution_count": 35,
"execution_count": null,
"outputs": []
},
{
Expand All @@ -1802,7 +1801,7 @@
"id": "IQb6Dxhy9-ge",
"outputId": "a008648d-32e1-4eda-bc91-814ff2f34d0d"
},
"execution_count": 36,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down Expand Up @@ -1854,7 +1853,7 @@
"id": "oUd9pD8s2-ly",
"outputId": "3b95da36-6286-4315-e985-652c633ee918"
},
"execution_count": 37,
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
Expand Down Expand Up @@ -2008,7 +2007,7 @@
"id": "FSpAuBWF_i1y",
"outputId": "c9d02b00-e906-49de-932c-be4f487bec71"
},
"execution_count": 41,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
Expand Down Expand Up @@ -2039,7 +2038,7 @@
"id": "UeRE3dijMrGc",
"outputId": "170bd4a6-6feb-4e3e-a133-1042f915d603"
},
"execution_count": 42,
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
Expand Down

0 comments on commit d09e271

Please sign in to comment.