diff --git a/Karpeev2024RiemannianGen/code/ckpt_s10.pth b/Karpeev2024RiemannianGen/code/ckpt_s10.pth new file mode 100644 index 0000000..3725628 Binary files /dev/null and b/Karpeev2024RiemannianGen/code/ckpt_s10.pth differ diff --git a/Karpeev2024RiemannianGen/code/gen score based models [latest].ipynb b/Karpeev2024RiemannianGen/code/gen score based models [latest].ipynb index 7995fa5..4c72eab 100644 --- a/Karpeev2024RiemannianGen/code/gen score based models [latest].ipynb +++ b/Karpeev2024RiemannianGen/code/gen score based models [latest].ipynb @@ -3,7 +3,8 @@ "nbformat_minor": 0, "metadata": { "colab": { - "provenance": [] + "provenance": [], + "gpuType": "T4" }, "kernelspec": { "name": "python3", @@ -14,7 +15,7 @@ }, "widgets": { "application/vnd.jupyter.widget-state+json": { - "a848b9e89e0b4c29bdae332651111aea": { + "fae5af4162754b12abcd3d1cb2909455": { "model_module": "@jupyter-widgets/controls", "model_name": "HBoxModel", "model_module_version": "1.5.0", @@ -29,14 +30,14 @@ "_view_name": "HBoxView", "box_style": "", "children": [ - "IPY_MODEL_8ed2bbfb61624d4fb720bded20e2a2dd", - "IPY_MODEL_a574d9608d5a4284b6b753da45f846f2", - "IPY_MODEL_e199e1db8b8e4b098e345b20086679fc" + "IPY_MODEL_92b70b00c2dd4681bb5651fc9b4dc703", + "IPY_MODEL_b2acb21ba4f94ccb97ad4da2d75b308b", + "IPY_MODEL_5afdcbb0b3184574bbabb8d96ad7d26c" ], - "layout": "IPY_MODEL_f857a567546c4ff6bd9615606b0ef0b6" + "layout": "IPY_MODEL_e526ea36fd244207b0e6fb3de9fa4bea" } }, - "8ed2bbfb61624d4fb720bded20e2a2dd": { + "92b70b00c2dd4681bb5651fc9b4dc703": { "model_module": "@jupyter-widgets/controls", "model_name": "HTMLModel", "model_module_version": "1.5.0", @@ -51,13 +52,13 @@ "_view_name": "HTMLView", "description": "", "description_tooltip": null, - "layout": "IPY_MODEL_a75bf085c02d44c2bc3f139196164acf", + "layout": "IPY_MODEL_98ca802b97af46ad9721005ea4a3a9e0", "placeholder": "​", - "style": "IPY_MODEL_5fe4b1d38c754f5cb079fe352042bc28", - "value": "" + "style": "IPY_MODEL_6f963dc2a27847229dfa24c1b0d95455", + "value": "Average Loss: 456.099613: 100%" } }, - "a574d9608d5a4284b6b753da45f846f2": { + "b2acb21ba4f94ccb97ad4da2d75b308b": { "model_module": "@jupyter-widgets/controls", "model_name": "FloatProgressModel", "model_module_version": "1.5.0", @@ -73,15 +74,15 @@ "bar_style": "success", "description": "", "description_tooltip": null, - "layout": "IPY_MODEL_d1c6c37c0df546c08701f85d7b162bb4", - "max": 1, + "layout": "IPY_MODEL_75a8beb9e0d14ad7bf2406389fd7feba", + "max": 50, "min": 0, "orientation": "horizontal", - "style": "IPY_MODEL_4fd1a760595440e38058fca2db883bb1", - "value": 0 + "style": "IPY_MODEL_048c10880e904eebbfbe321604a94d4a", + "value": 50 } }, - "e199e1db8b8e4b098e345b20086679fc": { + "5afdcbb0b3184574bbabb8d96ad7d26c": { "model_module": "@jupyter-widgets/controls", "model_name": "HTMLModel", "model_module_version": "1.5.0", @@ -96,13 +97,13 @@ "_view_name": "HTMLView", "description": "", "description_tooltip": null, - "layout": "IPY_MODEL_90dca69159fd4fc0a7a25c82a085182d", + "layout": "IPY_MODEL_6fd73c81237845c6b16dfcc4f6c1b818", "placeholder": "​", - "style": "IPY_MODEL_8e0e17f92021477f95337247af51357c", - "value": " 0/0 [00:00<?, ?it/s]" + "style": "IPY_MODEL_0c3c214f39464a0d909bd081c41dff76", + "value": " 50/50 [28:05<00:00, 33.69s/it]" } }, - "f857a567546c4ff6bd9615606b0ef0b6": { + "e526ea36fd244207b0e6fb3de9fa4bea": { "model_module": "@jupyter-widgets/base", "model_name": "LayoutModel", "model_module_version": "1.2.0", @@ -154,7 +155,7 @@ "width": null } }, - "a75bf085c02d44c2bc3f139196164acf": { + "98ca802b97af46ad9721005ea4a3a9e0": { "model_module": "@jupyter-widgets/base", "model_name": "LayoutModel", "model_module_version": "1.2.0", @@ -206,7 +207,7 @@ "width": null } }, - "5fe4b1d38c754f5cb079fe352042bc28": { + "6f963dc2a27847229dfa24c1b0d95455": { "model_module": "@jupyter-widgets/controls", "model_name": "DescriptionStyleModel", "model_module_version": "1.5.0", @@ -221,7 +222,7 @@ "description_width": "" } }, - "d1c6c37c0df546c08701f85d7b162bb4": { + "75a8beb9e0d14ad7bf2406389fd7feba": { "model_module": "@jupyter-widgets/base", "model_name": "LayoutModel", "model_module_version": "1.2.0", @@ -270,10 +271,10 @@ "right": null, "top": null, "visibility": null, - "width": "20px" + "width": null } }, - "4fd1a760595440e38058fca2db883bb1": { + "048c10880e904eebbfbe321604a94d4a": { "model_module": "@jupyter-widgets/controls", "model_name": "ProgressStyleModel", "model_module_version": "1.5.0", @@ -289,7 +290,7 @@ "description_width": "" } }, - "90dca69159fd4fc0a7a25c82a085182d": { + "6fd73c81237845c6b16dfcc4f6c1b818": { "model_module": "@jupyter-widgets/base", "model_name": "LayoutModel", "model_module_version": "1.2.0", @@ -341,7 +342,7 @@ "width": null } }, - "8e0e17f92021477f95337247af51357c": { + "0c3c214f39464a0d909bd081c41dff76": { "model_module": "@jupyter-widgets/controls", "model_name": "DescriptionStyleModel", "model_module_version": "1.5.0", @@ -357,7 +358,8 @@ } } } - } + }, + "accelerator": "GPU" }, "cells": [ { @@ -390,19 +392,19 @@ "colab": { "base_uri": "https://localhost:8080/" }, - "outputId": "b4f9d276-a58e-419b-8b90-c484caafd250" + "outputId": "58a706d8-e7ea-4b06-aaec-a63dca346c4a" }, - "execution_count": 10, + "execution_count": 1, "outputs": [ { "output_type": "execute_result", "data": { "text/plain": [ - "" + "" ] }, "metadata": {}, - "execution_count": 10 + "execution_count": 1 } ] }, @@ -417,7 +419,7 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": 2, "metadata": { "id": "wdd3K0E9DMzy" }, @@ -450,9 +452,9 @@ "height": 444 }, "id": "z25QSoBsDOEx", - "outputId": "38dd8fff-f66b-4dd8-9e35-351a04659934" + "outputId": "08c0ff35-5ddb-4e9e-937d-fa6929b2880a" }, - "execution_count": 12, + "execution_count": 3, "outputs": [ { "output_type": "execute_result", @@ -514,7 +516,7 @@ ], "text/html": [ "\n", - "
\n", + "
\n", "
\n", "