diff --git a/docs/api/datasets.rst b/docs/api/datasets.rst index b02439d26..df3ff2164 100644 --- a/docs/api/datasets.rst +++ b/docs/api/datasets.rst @@ -238,6 +238,7 @@ Available Datasets datasets/pyhealth.datasets.BMDHSDataset datasets/pyhealth.datasets.COVID19CXRDataset datasets/pyhealth.datasets.ChestXray14Dataset + datasets/pyhealth.datasets.VQARADDataset datasets/pyhealth.datasets.TUABDataset datasets/pyhealth.datasets.TUEVDataset datasets/pyhealth.datasets.ClinVarDataset diff --git a/docs/api/datasets/pyhealth.datasets.VQARADDataset.rst b/docs/api/datasets/pyhealth.datasets.VQARADDataset.rst new file mode 100644 index 000000000..d38986dc5 --- /dev/null +++ b/docs/api/datasets/pyhealth.datasets.VQARADDataset.rst @@ -0,0 +1,11 @@ +pyhealth.datasets.VQARADDataset +=================================== + +The VQA-RAD dataset for medical visual question answering. The dataset loader +converts the public JSON annotations into a flat metadata CSV that PyHealth can +ingest, and its default task is :class:`~pyhealth.tasks.MedicalVQATask`. + +.. autoclass:: pyhealth.datasets.VQARADDataset + :members: + :undoc-members: + :show-inheritance: diff --git a/docs/api/models.rst b/docs/api/models.rst index 7368dec94..72ee1cb5c 100644 --- a/docs/api/models.rst +++ b/docs/api/models.rst @@ -48,6 +48,9 @@ routes each feature type automatically. * - :doc:`models/pyhealth.models.GraphCare` - You want to augment EHR codes with a medical knowledge graph - Combines code sequences with a :class:`~pyhealth.graph.KnowledgeGraph` + * - :doc:`models/pyhealth.models.MedFlamingo` + - You are solving multimodal medical tasks with images plus text prompts (for example, VQA-style radiology QA) + - Flamingo-style architecture with a frozen vision encoder + frozen language model connected by gated cross-attention layers How BaseModel Works -------------------- @@ -194,6 +197,7 @@ API Reference models/pyhealth.models.ConCare models/pyhealth.models.Agent models/pyhealth.models.GRASP + models/pyhealth.models.MedFlamingo models/pyhealth.models.MedLink models/pyhealth.models.TCN models/pyhealth.models.TFMTokenizer diff --git a/docs/api/models/pyhealth.models.MedFlamingo.rst b/docs/api/models/pyhealth.models.MedFlamingo.rst new file mode 100644 index 000000000..7f782d0e3 --- /dev/null +++ b/docs/api/models/pyhealth.models.MedFlamingo.rst @@ -0,0 +1,24 @@ +pyhealth.models.MedFlamingo +=================================== + +MedFlamingo: multimodal medical few-shot learner. + +The separate callable MedFlamingoLayer (gated cross-attention dense block) +and the complete MedFlamingo model. + +**Paper:** Moor et al. "Med-Flamingo: a Multimodal Medical Few-shot Learner" ML4H 2023. + +.. note:: + + This is a stub implementation. The class structure and signatures are + in place, but ``forward()`` and ``generate()`` raise ``NotImplementedError``. + +.. autoclass:: pyhealth.models.MedFlamingoLayer + :members: + :undoc-members: + :show-inheritance: + +.. autoclass:: pyhealth.models.MedFlamingo + :members: + :undoc-members: + :show-inheritance: diff --git a/docs/api/tasks.rst b/docs/api/tasks.rst index 399b8f1aa..b1aaf74fd 100644 --- a/docs/api/tasks.rst +++ b/docs/api/tasks.rst @@ -213,6 +213,7 @@ Available Tasks DKA Prediction (MIMIC-IV) Drug Recommendation Length of Stay Prediction + Medical VQA Medical Transcriptions Classification Mortality Prediction (Next Visit) Mortality Prediction (StageNet MIMIC-IV) diff --git a/docs/api/tasks/pyhealth.tasks.MedicalVQATask.rst b/docs/api/tasks/pyhealth.tasks.MedicalVQATask.rst new file mode 100644 index 000000000..4221d6ab3 --- /dev/null +++ b/docs/api/tasks/pyhealth.tasks.MedicalVQATask.rst @@ -0,0 +1,12 @@ +pyhealth.tasks.MedicalVQATask +=================================== + +Medical visual question answering task for paired radiology images and +questions. This task treats VQA-RAD answers as a multiclass prediction target +so the resulting ``SampleDataset`` can be trained with the standard PyHealth +trainer loop. + +.. autoclass:: pyhealth.tasks.MedicalVQATask + :members: + :undoc-members: + :show-inheritance: diff --git a/examples/medflamingo_vqarad_demo.ipynb b/examples/medflamingo_vqarad_demo.ipynb new file mode 100644 index 000000000..39ee8a28d --- /dev/null +++ b/examples/medflamingo_vqarad_demo.ipynb @@ -0,0 +1,29029 @@ +{ + "cells": [ + { + "cell_type": "markdown", + "metadata": { + "id": "title" + }, + "source": [ + "# Med-Flamingo on VQA-RAD: Colab Evaluation Notebook\n", + "\n", + "This notebook is designed to get **presentation-worthy, grader-showable** results from real VQA-RAD runs.\n", + "\n", + "It supports two tracks:\n", + "\n", + "1. **Fast, honest benchmark track**: leakage-aware VQA-RAD evaluation, majority baseline, question-only baseline, and a PyHealth `MedFlamingo` classifier sweep.\n", + "2. **Stretch track**: official `Med-Flamingo-9B` checkpoint generation on Colab GPU, scored with exact match, BERTScore, and yes/no accuracy.\n", + "\n", + "Important: the stretch track is still **not identical to the paper's clinician-rated protocol**. It is a stronger local reproduction path than the current PyHealth classifier, but it should still be presented honestly as a local generative benchmark." + ], + "id": "title" + }, + { + "cell_type": "code", + "execution_count": 10, + "metadata": { + "id": "setup" + }, + "outputs": [], + "source": [ + "%%capture\n", + "import os\n", + "import sys\n", + "import shutil\n", + "import subprocess\n", + "from pathlib import Path\n", + "\n", + "IN_COLAB = \"google.colab\" in sys.modules\n", + "if IN_COLAB:\n", + " from google.colab import drive\n", + " drive.mount(\"/content/drive\", force_remount=False)\n", + "\n", + "REPO_URL = \"https://github.com/sunlabuiuc/PyHealth.git\"\n", + "REPO_DIR = Path(\"/content/PyHealth\")\n", + "PR_REF = \"pull/954/head:upstream-pr-954\"\n", + "SENTINEL = Path(\"/content/.medflamingo_colab_setup_complete\")\n", + "Path(\"/content/PyHealth/output/colab_eval\").mkdir(parents=True, exist_ok=True)\n", + "\n", + "\n", + "os.chdir(\"/content\")\n", + "\n", + "if REPO_DIR.exists():\n", + " shutil.rmtree(REPO_DIR)\n", + "\n", + "subprocess.run([\"git\", \"clone\", REPO_URL, str(REPO_DIR)], check=True)\n", + "os.chdir(REPO_DIR)\n", + "\n", + "subprocess.run([\"git\", \"remote\", \"-v\"], check=True)\n", + "subprocess.run([\"git\", \"fetch\", \"origin\", PR_REF], check=True)\n", + "subprocess.run([\"git\", \"checkout\", \"-B\", \"upstream-pr-954\", \"upstream-pr-954\"], check=True)\n", + "\n", + "if not SENTINEL.exists():\n", + " subprocess.run([sys.executable, \"-m\", \"pip\", \"install\", \"-q\", \"-U\", \"pip\", \"setuptools\", \"wheel\"], check=True)\n", + " subprocess.run([\n", + " sys.executable, \"-m\", \"pip\", \"install\", \"-q\",\n", + " \"numpy~=2.2.0\",\n", + " \"scipy~=1.16.0\",\n", + " \"pandas~=2.3.1\",\n", + " \"scikit-learn~=1.7.0\",\n", + " \"bert-score\",\n", + " \"sentencepiece\",\n", + " \"huggingface_hub\",\n", + " \"accelerate\",\n", + " ], check=True)\n", + " subprocess.run([sys.executable, \"-m\", \"pip\", \"install\", \"-q\", \"-e\", \".\"], check=True)\n", + " SENTINEL.write_text(\"ok\")\n", + "\n", + "subprocess.run([sys.executable, \"-m\", \"pip\", \"check\"], check=False)\n", + "print(\"Checked out upstream PR 954 only\")\n" + ], + "id": "setup" + }, + { + "cell_type": "code", + "execution_count": 1, + "metadata": { + "id": "config", + "colab": { + "base_uri": "https://localhost:8080/" + }, + "outputId": "d99a9e2a-dc11-4f09-b819-74d9c4707a2b" + }, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "repo: /content\n", + "vqarad_root exists: True\n", + "device: cuda\n", + "torch: 2.7.1+cu126\n" + ] + } + ], + "source": [ + "\n", + "from pathlib import Path\n", + "import json\n", + "import os\n", + "import sys\n", + "import subprocess\n", + "\n", + "import pandas as pd\n", + "import torch\n", + "\n", + "VQARAD_ROOT = Path(\"/content/drive/MyDrive/OSF Storage Archive\")\n", + "RESULTS_DIR = Path(\"/content/PyHealth/output/colab_eval\")\n", + "RESULTS_DIR.mkdir(parents=True, exist_ok=True)\n", + "\n", + "SUBSET = \"yesno\"\n", + "SPLIT_MODE = \"leakage_union\"\n", + "SEED = 42\n", + "\n", + "RUN_STRONG_BASELINES = True\n", + "RUN_PYHEALTH_SWEEP = True\n", + "RUN_OFFICIAL_MEDFLAMINGO = True\n", + "\n", + "# Start with 100 for a quick pilot, then scale up on A100/L4 once the pipeline works.\n", + "OFFICIAL_EVAL_LIMIT = 250\n", + "FEW_SHOT_K = 6\n", + "OFFICIAL_MAX_NEW_TOKENS = 4\n", + "OFFICIAL_VOTE_PASSES = 3\n", + "\n", + "DEVICE = \"cuda\" if torch.cuda.is_available() else \"cpu\"\n", + "\n", + "print(f\"repo: {Path.cwd()}\")\n", + "print(f\"vqarad_root exists: {VQARAD_ROOT.exists()}\")\n", + "print(f\"device: {DEVICE}\")\n", + "print(f\"torch: {torch.__version__}\")\n", + "if torch.cuda.is_available():\n", + " subprocess.run([\"nvidia-smi\"], check=False)\n" + ], + "id": "config" + }, + { + "cell_type": "code", + "execution_count": 2, + "metadata": { + "id": "inline-eval-and-baselines", + "colab": { + "base_uri": "https://localhost:8080/", + "height": 1000, + "referenced_widgets": [ + "acd7b5ff0a9f4d63bf388a6b2cebf7a2", + "6ce4ca3ff29845708d4e160e2d85fdc1", + "e7cc05e116b14a8295aa2d4a6a3e1f71", + "51846529e85e4328861363b04f407960", + "84a9e53caa524c13bfdb20a1c3186df9", + "604a6b8efbb74826bd7d45be941838bb", + "f42d1d7b51884b7bbb4e5104b8798d7a", + "6bdc1e828cfc482aab413da3a8d7dfb2", + "ab1eb004b8714a25b1edfb6f76f3faa4", + "e958a0ce91d7432cb0dc0e89d9a70b89", + "d176747e04594af089dc09a824a9a43e", + "33a890b9ac7a44b69e5abdf5551db128", + "ed45b1fda48345ce8c9608f763a48aa5", + "2cc26f3bf8e0447eaacf48ab265f1389", + "e17f0eac5ea24bc586bb58fca8abd2ad", + "14ef024857914b20bbd6df5a31210300", + "72099a28efa445baa050627a973cb7b8", + "535461538b114038ab19a1c17809c1be", + "7e3f94ef50a040b68ef7bff8792db3e9", + "35a841d9e83643ed95e63b52b4e15c0b", + "15d2c16bc4b847aba7c776beaffdb905", + "70c468a5e51f44ecad363d3f85699ea5", + "3d48ab5e157041ae964922f5b8eb72df", + "e255481141fc46689caadd7fcd205e29", + "5a0cabdcdee348c295ea35b28a8fa5d4", + "adce6d44c61f4371841466da8baf6712", + "264141dd7200401b893a2d829db47b14", + "be9a8d0b703543b3bdebe4309e7ce82b", + "4557092565324dfa902e8ff4dbc26c68", + "d40c4075d42945aabbb1a55758fc0fcb", + "ff025ea9727441e589dd2957ffeb280e", + "0d2f7497712646d9a34f74c1b21eec6d", + "2bbffd286d814603a3eb1f31d8bca9f3", + "6df7f59495984977839277d0cc8d189a", + "9cadcb737429486fa6a89b8885af0985", + "4a61a2ae454e4fdb8f64a95f86645891", + "07d0a0d199cb4e0e91fb5a7c54cea000", + "007aa53b17aa4ca09b1cefa9230d2d15", + "f28787e3e77c4f4e8b2094a367d423d8", + "c1a31e283c9c4ada8bd2854ca9a91357", + "7812a101bd714c6fa6aadba6ceb5aff8", + "cc537b85722c4477a8df63c3e6e7caaf", + "cb1a6b1e367c4f74abea9fd73f2f6102", + "b56c07b88dd74ee9a6a93e91bf62ff65", + "6c2c78aa71a64f7697b6773553d2be58", + "ad71a53f38a741f8940205b98999f809", + "111dd33db66c4a249fc5a4d04b07629b", + "795539cbc9a24914832bf895a2aac7c9", + "4e74fd0dd3fd47258aa7cc1e38e2544d", + "ac11b99982db41cf92cc62eca6a263f0", + "57290ee4890b4331ae2660b097fd4991", + "e22bf4d7e1a94284a6a8941286ef9bbe", + "a0bbd056beb840b9805dfdc35a2192cc", + "5180a004be02425ebc69e844e6b98f59", + "aa2779d0654b466b8ce27f1b4db7d57d", + "df31724f148447ff81138992c0e814a8", + "448f69945f6846728ec99f3151c11d5f", + "bac56295e5404e2e8f4130dd43b8dc87", + "f88e40b41867487fb4f5ae3d2567d18b", + "a9ffa517041f45d7b9645b5778aff6b6", + "737a19e36d7849b3a6d54273058bb24f", + "a9695a8d5d794c17a530444cb9c3f8c7", + "9a327b232eab43cb89aa6d052a81a484", + "9f3dd0a3b96c4aff9cdce6897f79e99d", + "f1abec28112c43adabc687cc29369161", + "21ab6280076f46b6bfc9cf6acfde6cf6", + "de81098896474289a6d1d02fe347bef5", + "b36e1c7acb9d419baa38c1b181fde85e", + "bc63f48c3d844f8c92112fd37918604e", + "0e2fcd45270e4a858f4762e3bd550f1b", + "0fb6ae683fd1433cb6546f95b2fd2a04", + "1a4f12b78d9b42cba0c1cf174dbc595b", + "8ab706519d06433ebf8bab25f9390089", + "cdc86b05cf9b4919b0cc4947bc315dd3", + "efc6c70f89054c5ea2f8b9a6d1136e06", + "a605993ade3d4b5a8ad84d8f4a57f85b", + "9019fa49eb2a4f78a3921dbb7b6e8eea", + "40ad37b58fbd4d4aab2b9a2e315007c5", + "f173a0ce82c24009aa805aa7fd3fc921", + "25c658dba47749a3a4b801b54dbf3548", + "b92bc1018872457a9958937aa89e4fcb", + "3a0f366445b84d988e18c50d864e495b", + "a8bd363d80eb423891ee5e5e0f7dd864", + "e12cf301b1d24037b3e09ba5339c844f", + "510ea785201540fdafc7d432e028f4ea", + "9142ed175d47474b98bfd7bb9130d182", + "330a88de89704bd39052c9f2bbf60145", + "a1aaedeba6424a25b350cb841afc1afe", + "5c952ff026394314b7e00dc82a0f9ace", + "05b9bf390e844bc09cc85fac66bee422", + "185f78f0257541d2b415b347426e9142", + "7c49107925b84bda99ef04c0efedb186", + "5329cb3d6bf04370ae2a0c3bf560895f", + "cf8ab0c005bf41c581c0a9613070964d", + "2460c4d003784bcabc36ab5e7a5f0669", + "9ac18602e4df4c078ebdce876e42e56d", + "67a31b05d10d4b6c98e45f5697eddd19", + "8101ae386c89401d9b75a66a3de58e56", + "1e1c77c6cea74aeb8d7f45cbf29f3683", + "e102b122537443f891706bd41de97b4b", + "8e96694b501c4f7a8464b47acf253b8e", + "cd81cd4d272e415f81561d68be6b162f", + "ff42f063011946fe9d2b6e4522fecfc4", + "2690b31acc4b4d42b7b4862850cc3e88", + "06bd1e8d1d5642749398e18c54587815", + "b719b5dea4624ae883b127b94696fa85", + "82855b27296c4c03abde7160e0865a27", + "0b12db7d9ac6442999e3d890f2b7d5a8", + "cd0a9a11b1c4448580648b3f6ea4dab3", + "73bb2fee355b49929698f7d5c996c827", + "b0ba18baafe64e97b89862ed539a60c6", + "d3d4fc244fbe468bad7f6a95bc4675b1", + "4e4ea5e920be49a6a632b23546dd1fa6", + "69ced8f4c9974550bd52f4a24652cf7e", + "eedc6168a02842b9a83e0cdd460681dd", + "075e02f3c4b246179b546623b536db87", + "848f2f530934403eb7a28eeaec02ab26", + "e09a891456dc4055bcf284b1e8851534", + "c4bf481975ee42048291acadc69d6e92", + "7ad717da24b84648ae69628175bfb97a", + "0b72a826af984585ba0f940c723cd2a2", + "4e5d106e8df04c04afb93e2e05f1e5a1", + "62e662eb82a544eba23f8578efce062c", + "08395f2628dc47fdb78b587af045a3f2", + "b7c2c2fd01974ed7aa8a2eca8ef5eca2", + "c51a2a4822a0407d85b0a020d19daafd", + "2271ce8b6041454c8678dae859015a88", + "9eb6b05ccf9b41739b50f349e36ba236", + "21e18797a53b4cc38a1b1d84fa2e2a7e", + "ef374580a3894b72beaac1da7303ad74", + "57e07801abda45538ab02ab77fd629e2", + "0b906d06ba884f39bfc620c184262efa", + "0eeb71251ae74f7798e40fc402ec3427", + "b1850f5ba46747a6900488e407a8df81", + "bdafccd294e8413fb95244e0709b5842", + "852a157855904586b528183eec635bd9", + "bbefe3ed5e3b4097a7a266fcc606268f", + "b37c43fec7a44e0db34d84d0a496c688", + "3c44188c263e45859f046dd07432d62b", + "588ead88ef904af6abebd242bd800216", + "9e324017db60470880d8dc46f9bdda3b", + "afbf2f67e75149e39d8a5a8ddefd48d0", + "b95bf85d7d2f4961855001d74c766d28", + "8c13701c53964a22877473434a8e8844", + "771a173d67c64675a4bde9b588437eeb", + "b21efd6076d54af1a9befdd6577f1a4b", + "c487e8a64c12441393671b85b0ff060a", + "a5f1f050588c4fedae9e8ecebc60e7f4", + "33bfbc7cdd1b4416a1d9fde8a1f9a1d0", + "2fcedae3d8364a29b72b26f0b03fc4fe", + "a0b5e642d64c436b9eea9e6a484a4681", + "f51216bd565748edb00b4c57b03df263", + "ecde02d5df744d778d02289194a2863b", + "9afc1eb6281942d28ccfd1b63fec2c30" + ] + }, + "outputId": "a888874d-e55c-428b-dbe1-341688af4d77" + }, + "outputs": [ + { + "output_type": "stream", + "name": "stderr", + "text": [ + "/content/PyHealth/pyhealth/metrics/calibration.py:122: SyntaxWarning: invalid escape sequence '\\c'\n", + " accuracy of 1. Thus, the ECE is :math:`\\\\frac{1}{3} \\cdot 0.49 + \\\\frac{2}{3}\\cdot 0.3=0.3633`.\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "subset stats:\n", + "{\n", + " \"n_samples\": 1193,\n", + " \"n_unique_images\": 294,\n", + " \"n_unique_questions\": 1029,\n", + " \"n_unique_answers\": 2,\n", + " \"top_answers\": [\n", + " [\n", + " \"no\",\n", + " 606\n", + " ],\n", + " [\n", + " \"yes\",\n", + " 587\n", + " ]\n", + " ],\n", + " \"answer_type_counts\": {\n", + " \"closed\": 1193\n", + " },\n", + " \"question_type_top10\": [\n", + " [\n", + " \"pres\",\n", + " 613\n", + " ],\n", + " [\n", + " \"size\",\n", + " 154\n", + " ],\n", + " [\n", + " \"abn\",\n", + " 118\n", + " ],\n", + " [\n", + " \"modality\",\n", + " 72\n", + " ],\n", + " [\n", + " \"plane\",\n", + " 53\n", + " ],\n", + " [\n", + " \"other\",\n", + " 52\n", + " ],\n", + " [\n", + " \"attrib\",\n", + " 36\n", + " ],\n", + " [\n", + " \"color\",\n", + " 29\n", + " ],\n", + " [\n", + " \"organ\",\n", + " 17\n", + " ],\n", + " [\n", + " \"pos\",\n", + " 17\n", + " ]\n", + " ],\n", + " \"image_organ_counts\": {\n", + " \"head\": 308,\n", + " \"chest\": 477,\n", + " \"abd\": 408\n", + " }\n", + "}\n", + "split sizes: {'train': 886, 'val': 98, 'test': 209}\n", + "train majority: {'majority_answer': 'yes', 'accuracy': 0.45933014354066987, 'f1_macro': 0.31475409836065577}\n", + "question-only: {'accuracy': 0.6028708133971292, 'f1_macro': 0.6028708133971292, 'selected_config': {'analyzer': 'word', 'ngram_range': (1, 2), 'max_features': 20000, 'C': 0.5, 'class_weight': None}, 'val_f1_macro': 0.6530612244897959}\n", + "Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.processors.label_processor:Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "config.json: 0.00B [00:00, ?B/s]" + ], + "application/vnd.jupyter.widget-view+json": { + "version_major": 2, + "version_minor": 0, + "model_id": "acd7b5ff0a9f4d63bf388a6b2cebf7a2" + } + }, + "metadata": {} + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "pytorch_model.bin: 0%| | 0.00/605M [00:00\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer params: {'lr': 0.001}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer params: {'lr': 0.001}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Patience: 2\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Patience: 2\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Epoch 0 / 6: 0%| | 0/111 [00:00\n", + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
experimentaccuracyf1_macro
0Question-only baseline0.6028710.602871
1PyHealth MedFlamingo classifier0.4593300.314754
2Majority baseline0.4593300.314754
\n", + "
\n", + "
\n", + "\n", + "
\n", + " \n", + "\n", + " \n", + "\n", + " \n", + "
\n", + "\n", + "\n", + "
\n", + " \n" + ], + "application/vnd.google.colaboratory.intrinsic+json": { + "type": "dataframe", + "summary": "{\n \"name\": \" print('Skipping strong baseline run\",\n \"rows\": 3,\n \"fields\": [\n {\n \"column\": \"experiment\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 3,\n \"samples\": [\n \"Question-only baseline\",\n \"PyHealth MedFlamingo classifier\",\n \"Majority baseline\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"accuracy\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.082873244381286,\n \"min\": 0.45933014354066987,\n \"max\": 0.6028708133971292,\n \"num_unique_values\": 2,\n \"samples\": [\n 0.45933014354066987,\n 0.6028708133971292\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"f1_macro\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.16634426298433866,\n \"min\": 0.31475409836065577,\n \"max\": 0.6028708133971292,\n \"num_unique_values\": 2,\n \"samples\": [\n 0.31475409836065577,\n 0.6028708133971292\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n }\n ]\n}" + } + }, + "metadata": {} + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "saved to /content/PyHealth/output/colab_eval/yesno_leakage_union_colab.json\n", + "leakage checks: {'train_test_image_overlap': 75, 'train_test_question_overlap': 13}\n" + ] + } + ], + "source": [ + "import json\n", + "import random\n", + "import time\n", + "from collections import Counter\n", + "from pathlib import Path\n", + "\n", + "import numpy as np\n", + "import pandas as pd\n", + "import torch\n", + "from sklearn.feature_extraction.text import TfidfVectorizer\n", + "from sklearn.linear_model import LogisticRegression\n", + "from sklearn.metrics import accuracy_score, confusion_matrix, f1_score\n", + "\n", + "from pyhealth.datasets import create_sample_dataset, get_dataloader\n", + "from pyhealth.models.medflamingo import MedFlamingo\n", + "from pyhealth.trainer import Trainer\n", + "\n", + "def resolve_device(device: str = 'auto') -> str:\n", + " if device != 'auto':\n", + " return device\n", + " if torch.cuda.is_available():\n", + " return 'cuda'\n", + " if hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():\n", + " return 'mps'\n", + " return 'cpu'\n", + "\n", + "def load_rows(root: Path):\n", + " json_path = root / 'VQA_RAD Dataset Public.json'\n", + " if not json_path.exists():\n", + " raise FileNotFoundError(f'Missing VQA-RAD JSON: {json_path}')\n", + "\n", + " if (root / 'VQA_RAD Image Folder').is_dir():\n", + " image_root = root / 'VQA_RAD Image Folder'\n", + " elif (root / 'images').is_dir():\n", + " image_root = root / 'images'\n", + " else:\n", + " raise FileNotFoundError(f'Missing VQA-RAD image directory under {root}')\n", + "\n", + " raw_rows = json.loads(json_path.read_text())\n", + " rows = []\n", + " for idx, row in enumerate(raw_rows):\n", + " rows.append({\n", + " 'sample_id': idx,\n", + " 'patient_id': f'vqarad-{idx}',\n", + " 'visit_id': f'vqarad-{idx}',\n", + " 'image': str(image_root / row['image_name']),\n", + " 'image_name': row['image_name'],\n", + " 'question': row['question'].strip(),\n", + " 'answer': str(row['answer']).strip().lower(),\n", + " 'answer_type': str(row['answer_type']).strip().lower(),\n", + " 'question_type': str(row['question_type']).strip().lower(),\n", + " 'image_organ': str(row['image_organ']).strip().lower(),\n", + " })\n", + " return rows\n", + "\n", + "def filter_rows(rows, subset: str):\n", + " if subset == 'all':\n", + " return list(rows)\n", + " if subset == 'closed':\n", + " return [row for row in rows if row['answer_type'] == 'closed']\n", + " if subset == 'open':\n", + " return [row for row in rows if row['answer_type'] == 'open']\n", + " if subset == 'yesno':\n", + " return [row for row in rows if row['answer'] in {'yes', 'no'}]\n", + " raise ValueError(f'Unsupported subset: {subset}')\n", + "\n", + "def split_sample_indices(rows, seed: int):\n", + " indices = list(range(len(rows)))\n", + " rng = random.Random(seed)\n", + " rng.shuffle(indices)\n", + " train_end = int(len(indices) * 0.7)\n", + " val_end = int(len(indices) * 0.8)\n", + " return {'train': indices[:train_end], 'val': indices[train_end:val_end], 'test': indices[val_end:]}\n", + "\n", + "def split_leakage_union_indices(rows, seed: int):\n", + " rng = random.Random(seed)\n", + " unique_images = sorted({row['image_name'] for row in rows})\n", + " unique_questions = sorted({row['question'].strip().lower() for row in rows})\n", + " rng.shuffle(unique_images)\n", + " rng.shuffle(unique_questions)\n", + "\n", + " test_images = set(unique_images[:max(1, int(0.1 * len(unique_images)))])\n", + " test_questions = set(unique_questions[:max(1, int(0.1 * len(unique_questions)))])\n", + "\n", + " test = [\n", + " idx for idx, row in enumerate(rows)\n", + " if row['image_name'] in test_images or row['question'].strip().lower() in test_questions\n", + " ]\n", + " train_pool = [idx for idx in range(len(rows)) if idx not in set(test)]\n", + " rng.shuffle(train_pool)\n", + " val_size = max(1, int(0.1 * len(train_pool)))\n", + " val = train_pool[:val_size]\n", + " train = train_pool[val_size:]\n", + " return {'train': train, 'val': val, 'test': test}\n", + "\n", + "def subset_rows(rows, indices):\n", + " return [rows[idx] for idx in indices]\n", + "\n", + "def dataset_stats(rows):\n", + " answers = Counter(row['answer'] for row in rows)\n", + " answer_types = Counter(row['answer_type'] for row in rows)\n", + " question_types = Counter(row['question_type'] for row in rows)\n", + " organs = Counter(row['image_organ'] for row in rows)\n", + " return {\n", + " 'n_samples': len(rows),\n", + " 'n_unique_images': len({row['image_name'] for row in rows}),\n", + " 'n_unique_questions': len({row['question'] for row in rows}),\n", + " 'n_unique_answers': len(answers),\n", + " 'top_answers': answers.most_common(10),\n", + " 'answer_type_counts': dict(answer_types),\n", + " 'question_type_top10': question_types.most_common(10),\n", + " 'image_organ_counts': dict(organs),\n", + " }\n", + "\n", + "def make_dataset(rows, image_size: int = 224):\n", + " return create_sample_dataset(\n", + " samples=list(rows),\n", + " input_schema={\n", + " 'image': ('image', {'image_size': image_size, 'mode': 'RGB'}),\n", + " 'question': 'text',\n", + " },\n", + " output_schema={'answer': 'multiclass'},\n", + " dataset_name='vqarad_eval',\n", + " task_name='medical_vqa_eval',\n", + " in_memory=True,\n", + " )\n", + "\n", + "def majority_baseline(train_rows, test_rows):\n", + " majority_answer = Counter(row['answer'] for row in train_rows).most_common(1)[0][0]\n", + " y_true = [row['answer'] for row in test_rows]\n", + " y_pred = [majority_answer] * len(test_rows)\n", + " return {\n", + " 'majority_answer': majority_answer,\n", + " 'accuracy': accuracy_score(y_true, y_pred),\n", + " 'f1_macro': f1_score(y_true, y_pred, average='macro', zero_division=0),\n", + " }\n", + "\n", + "def question_only_baseline(train_rows, val_rows, test_rows):\n", + " search_space = [\n", + " {\n", + " 'analyzer': 'word',\n", + " 'ngram_range': (1, 2),\n", + " 'max_features': 20000,\n", + " 'C': 0.5,\n", + " 'class_weight': None,\n", + " },\n", + " {\n", + " 'analyzer': 'word',\n", + " 'ngram_range': (1, 2),\n", + " 'max_features': 30000,\n", + " 'C': 1.0,\n", + " 'class_weight': 'balanced',\n", + " },\n", + " {\n", + " 'analyzer': 'word',\n", + " 'ngram_range': (1, 3),\n", + " 'max_features': 30000,\n", + " 'C': 2.0,\n", + " 'class_weight': 'balanced',\n", + " },\n", + " {\n", + " 'analyzer': 'char_wb',\n", + " 'ngram_range': (3, 5),\n", + " 'max_features': 50000,\n", + " 'C': 2.0,\n", + " 'class_weight': 'balanced',\n", + " },\n", + " ]\n", + "\n", + " best_cfg = None\n", + " best_val_f1 = -1.0\n", + " for cfg in search_space:\n", + " vectorizer = TfidfVectorizer(\n", + " lowercase=True,\n", + " analyzer=cfg['analyzer'],\n", + " ngram_range=cfg['ngram_range'],\n", + " min_df=1,\n", + " max_features=cfg['max_features'],\n", + " )\n", + " clf = LogisticRegression(\n", + " max_iter=4000,\n", + " solver='lbfgs',\n", + " C=cfg['C'],\n", + " class_weight=cfg['class_weight'],\n", + " )\n", + " x_train = vectorizer.fit_transform([row['question'] for row in train_rows])\n", + " x_val = vectorizer.transform([row['question'] for row in val_rows])\n", + " y_train = [row['answer'] for row in train_rows]\n", + " y_val = [row['answer'] for row in val_rows]\n", + " clf.fit(x_train, y_train)\n", + " val_pred = clf.predict(x_val)\n", + " val_f1 = f1_score(y_val, val_pred, average='macro', zero_division=0)\n", + " if val_f1 > best_val_f1:\n", + " best_val_f1 = val_f1\n", + " best_cfg = dict(cfg)\n", + "\n", + " train_val_rows = list(train_rows) + list(val_rows)\n", + " vectorizer = TfidfVectorizer(\n", + " lowercase=True,\n", + " analyzer=best_cfg['analyzer'],\n", + " ngram_range=best_cfg['ngram_range'],\n", + " min_df=1,\n", + " max_features=best_cfg['max_features'],\n", + " )\n", + " clf = LogisticRegression(\n", + " max_iter=4000,\n", + " solver='lbfgs',\n", + " C=best_cfg['C'],\n", + " class_weight=best_cfg['class_weight'],\n", + " )\n", + " x_train_val = vectorizer.fit_transform([row['question'] for row in train_val_rows])\n", + " x_test = vectorizer.transform([row['question'] for row in test_rows])\n", + " y_train_val = [row['answer'] for row in train_val_rows]\n", + " y_test = [row['answer'] for row in test_rows]\n", + " clf.fit(x_train_val, y_train_val)\n", + " y_pred = clf.predict(x_test)\n", + " return {\n", + " 'accuracy': accuracy_score(y_test, y_pred),\n", + " 'f1_macro': f1_score(y_test, y_pred, average='macro', zero_division=0),\n", + " 'selected_config': best_cfg,\n", + " 'val_f1_macro': float(best_val_f1),\n", + " }\n", + "\n", + "def bootstrap_ci(y_true, y_pred, n_bootstrap: int, seed: int):\n", + " rng = np.random.default_rng(seed)\n", + " acc_scores = []\n", + " f1_scores = []\n", + " n = len(y_true)\n", + " for _ in range(n_bootstrap):\n", + " sample_idx = rng.integers(0, n, size=n)\n", + " ys = y_true[sample_idx]\n", + " ps = y_pred[sample_idx]\n", + " acc_scores.append(accuracy_score(ys, ps))\n", + " f1_scores.append(f1_score(ys, ps, average='macro', zero_division=0))\n", + "\n", + " def pct(values):\n", + " return [\n", + " float(np.percentile(values, 2.5)),\n", + " float(np.percentile(values, 50.0)),\n", + " float(np.percentile(values, 97.5)),\n", + " ]\n", + "\n", + " return {'accuracy': pct(acc_scores), 'f1_macro': pct(f1_scores)}\n", + "\n", + "def maybe_confusion_matrix(label_names, y_true_idx, y_pred_idx):\n", + " if len(label_names) > 10:\n", + " return None\n", + " matrix = confusion_matrix(y_true_idx, y_pred_idx, labels=list(range(len(label_names))))\n", + " return {'labels': list(label_names), 'matrix': matrix.tolist()}\n", + "\n", + "def run_model_eval(dataset, split_indices, *, vision_model_name, lang_model_name, batch_size, epochs, lr, weight_decay, patience, device, bootstrap_samples, seed):\n", + " train_dataset = dataset.subset(split_indices['train'])\n", + " val_dataset = dataset.subset(split_indices['val'])\n", + " test_dataset = dataset.subset(split_indices['test'])\n", + "\n", + " model = MedFlamingo(\n", + " dataset=train_dataset,\n", + " vision_model_name=vision_model_name,\n", + " lang_model_name=lang_model_name,\n", + " )\n", + " trainer = Trainer(model=model, metrics=['accuracy', 'f1_macro'], device=device, enable_logging=False)\n", + "\n", + " train_loader = get_dataloader(train_dataset, batch_size=batch_size, shuffle=True)\n", + " val_loader = get_dataloader(val_dataset, batch_size=batch_size, shuffle=False)\n", + " test_loader = get_dataloader(test_dataset, batch_size=batch_size, shuffle=False)\n", + "\n", + " start = time.time()\n", + " trainer.train(\n", + " train_dataloader=train_loader,\n", + " val_dataloader=val_loader,\n", + " epochs=epochs,\n", + " optimizer_params={'lr': lr},\n", + " weight_decay=weight_decay,\n", + " monitor='f1_macro',\n", + " monitor_criterion='max',\n", + " patience=patience,\n", + " )\n", + " train_seconds = time.time() - start\n", + "\n", + " metrics = trainer.evaluate(test_loader)\n", + " y_true, y_prob, loss_mean = trainer.inference(test_loader)\n", + " y_pred = y_prob.argmax(axis=1)\n", + "\n", + " answer_processor = train_dataset.output_processors['answer']\n", + " inverse_vocab = {idx: label for label, idx in answer_processor.label_vocab.items()}\n", + " label_names = [inverse_vocab[idx] for idx in range(answer_processor.size())]\n", + "\n", + " return {\n", + " 'metrics': {\n", + " 'accuracy': float(metrics['accuracy']),\n", + " 'f1_macro': float(metrics['f1_macro']),\n", + " 'loss': float(loss_mean),\n", + " },\n", + " 'bootstrap_ci_95': bootstrap_ci(y_true, y_pred, bootstrap_samples, seed),\n", + " 'train_seconds': train_seconds,\n", + " 'n_classes': int(answer_processor.size()),\n", + " 'confusion_matrix': maybe_confusion_matrix(label_names, y_true, y_pred),\n", + " }\n", + "\n", + "rows = filter_rows(load_rows(VQARAD_ROOT), SUBSET)\n", + "split_fn = split_leakage_union_indices if SPLIT_MODE == 'leakage_union' else split_sample_indices\n", + "split = split_fn(rows, SEED)\n", + "train_rows = subset_rows(rows, split['train'])\n", + "val_rows = subset_rows(rows, split['val'])\n", + "test_rows = subset_rows(rows, split['test'])\n", + "\n", + "majority_result = majority_baseline(train_rows, test_rows)\n", + "question_only_result = question_only_baseline(train_rows, val_rows, test_rows)\n", + "\n", + "print('subset stats:')\n", + "print(json.dumps(dataset_stats(rows), indent=2))\n", + "print('split sizes:', {k: len(v) for k, v in split.items()})\n", + "print('train majority:', majority_result)\n", + "print('question-only:', question_only_result)\n", + "\n", + "if RUN_STRONG_BASELINES:\n", + " device = resolve_device('auto')\n", + " output_json = RESULTS_DIR / 'yesno_leakage_union_colab.json'\n", + " dataset = make_dataset(rows, image_size=224)\n", + " results = {\n", + " 'config': {\n", + " 'subset': SUBSET,\n", + " 'split_mode': SPLIT_MODE,\n", + " 'seed': SEED,\n", + " 'device': device,\n", + " 'vision_model_name': 'openai/clip-vit-base-patch32',\n", + " 'lang_model_name': 'facebook/opt-125m',\n", + " 'epochs': 6,\n", + " 'batch_size': 8,\n", + " 'learning_rate': 1e-3,\n", + " 'weight_decay': 1e-4,\n", + " 'image_size': 224,\n", + " },\n", + " 'dataset_stats': dataset_stats(rows),\n", + " 'split_sizes': {k: len(v) for k, v in split.items()},\n", + " 'leakage_checks': {\n", + " 'train_test_image_overlap': len({row['image_name'] for row in train_rows} & {row['image_name'] for row in test_rows}),\n", + " 'train_test_question_overlap': len({row['question'].strip().lower() for row in train_rows} & {row['question'].strip().lower() for row in test_rows}),\n", + " },\n", + " 'baselines': {\n", + " 'majority': majority_result,\n", + " 'question_only_tfidf_logreg': question_only_result,\n", + " },\n", + " }\n", + "\n", + " results['medflamingo_classifier'] = run_model_eval(\n", + " dataset,\n", + " split,\n", + " vision_model_name='openai/clip-vit-base-patch32',\n", + " lang_model_name='facebook/opt-125m',\n", + " batch_size=8,\n", + " epochs=6,\n", + " lr=1e-3,\n", + " weight_decay=1e-4,\n", + " patience=2,\n", + " device=device,\n", + " bootstrap_samples=1000,\n", + " seed=SEED,\n", + " )\n", + "\n", + " output_json.write_text(json.dumps(results, indent=2))\n", + "\n", + " summary = pd.DataFrame([\n", + " {\n", + " 'experiment': 'PyHealth MedFlamingo classifier',\n", + " 'accuracy': results['medflamingo_classifier']['metrics']['accuracy'],\n", + " 'f1_macro': results['medflamingo_classifier']['metrics']['f1_macro'],\n", + " },\n", + " {\n", + " 'experiment': 'Majority baseline',\n", + " 'accuracy': results['baselines']['majority']['accuracy'],\n", + " 'f1_macro': results['baselines']['majority']['f1_macro'],\n", + " },\n", + " {\n", + " 'experiment': 'Question-only baseline',\n", + " 'accuracy': results['baselines']['question_only_tfidf_logreg']['accuracy'],\n", + " 'f1_macro': results['baselines']['question_only_tfidf_logreg']['f1_macro'],\n", + " },\n", + " ])\n", + " display(summary.sort_values(['accuracy', 'f1_macro'], ascending=False).reset_index(drop=True))\n", + " print('saved to', output_json)\n", + " print('leakage checks:', results['leakage_checks'])\n", + "else:\n", + " print('Skipping strong baseline run.')\n" + ], + "id": "inline-eval-and-baselines" + }, + { + "cell_type": "code", + "execution_count": 3, + "metadata": { + "id": "pyhealth-sweep", + "colab": { + "base_uri": "https://localhost:8080/", + "height": 1000, + "referenced_widgets": [ + "80841cd9a1254ed0921488602f6bc8ac", + "c688eb8ecea244d8a245e6e3be7683af", + "566507e388a848a38b1785242c3b9b2b", + "b01057a1da144b3681dbfafec8da839d", + "8efe4de11d9c459cb81c21c45a79ca0a", + "63b658ed08fc4d54940f4b7291b521d6", + "c47f0e121e8b4b6fad6f858b289cffc1", + "1d5097f6a7604a4da336025434cd1c62", + "11959cbae10d4c89818789ac4c50a4bf", + "d26b7c8e8e474342af8b1089a724963e", + "546b7e7a9dfb40799926f4119073c79a", + "ef2a8f30246043ae980116b11c539b1d", + "91fbed90a14f4b9687a156076b390a9e", + "ada4773fe1bd4b5cbe53bbc2c5c27132", + "03f9004190e444e3a21823a37d92a6ff", + "7b0aa6b967a442348757dbf9dd6f83ad", + "7a65709f790e47ff81a4c8b63922d640", + "6a66a27f072649a6845ab3abd90840e1", + "e02393dd2e1047b2a0ab9e83704fdacf", + "da6714b4927a4b7294a1e442352a7e3a", + "83797fbe29e0458199c77adaa0eabf0b", + "69304c74bd8148a8ae99dab82089196c", + "3cfc7bbf94ab40a892f58f08043bcabe", + "72ce6cdb5c8f4e72a9c03902294a316e", + "57194c49e2ef496abf2379890d123e48", + "e7e94a76ba09422782db85cd5926d51e", + "8d17b811e4fb4ce9b5b0f955b405212f", + "021efbd1c85c443abebc06c2d821d456", + "7400bd7f2d954f7eaff2c5db059b91fa", + "7098375b0e984f0d870efc564d47cca6", + "d204c069e2ca45b38e73d20cb47633a9", + "ea0ce3580cca4e94b494bcaf78bfffc4", + "c438ac99da5b4812ae6f533172d9cb2b", + "0a9df723378742c9b3ee6692e7c4011d", + "7596b808b94348eb821eb59952a8b626", + "f0309f365e8845bb92e199e036ffdd6c", + "1ef630ff54034d7fbfc3a00f7804390b", + "50203f5f18c649b3a7a16527da71beeb", + "33483e6d078b4038bef307c8124c7e06", + "f379fcc06939403fa7584ad92b2fc225", + "1b0602499e9b44b4be89e0cdcb14d4b5", + "a8ae55b1ee0b4220afce303cd998594f", + "aab89be66e5d46039d1821b3b12404d5", + "2b067b12b02547ccbc379492b409c0f0", + "75afd19177b2493084219a30b1ea80e9", + "a2d1baee530344d49ec19081e91f113a", + "ae8cb23915d14a1180d825e502084bd0", + "011a2e3962ad423cb2db2ecdab5a88ce", + "6005f7f4716344d3af218ce48c5f0a75", + "d0a86be8167a4ef0a7b7d0992489a1f3", + "ca5262b4ab044828afb48613b837352a", + "7e6d57fc41074dd19e0c2cabf17734ff", + "0b8c1d90464740ee874c8bd230734920", + "a0465d556fc44a8cbe2386ab9e730ac8", + "40ff142202c34dbab894478bd2a0c763", + "45aa5c19f3174cf4a87b79fa8461b8e9", + "935fd5c59e7840caa5bf9f14040dd606", + "f4eb7e61c5bd4359aeea893256cfff96", + "9b253f5fa36f4164a3f6b3a50279b0d4", + "24a7bd5f50ed489a985c64560ba92fdd", + "433065beb5494c72acda32eb5f3e3e4c", + "8f33b37f98d44b86a308deb2ee21f9e8", + "6140a2cfb0db4e219a2fa19b567e51bf", + "2fe3b150673a428da0d6c05df7ae02fc", + "74fc91c342204076a772d6e9d5a3b060", + "bca338b823bb4459ba3eb00498452cc5", + "1a19f0f8d5424461b0b1f02f7e9cad5b", + "298f8eb6997b482290cc15add49ca2b1", + "8c32ecde0b6f4535abc2b9cffe795d46", + "1e4b66d2b5b247c7b507cea16926e199", + "0587a9a7e9974ce38fb89a4afa2bc0f9", + "6c75063db68e4c68ae9e70f05b9fb1fe", + "6acae9f7715a4596aa5aac6680691864", + "1e0d83608f0648369ec93b617ffa5045", + "755446d0cbec41a99d3c5bb0fea917ee", + "9b677e467d6445f3a0c1ea363edf2b60", + "9c6276465c8f433e8985e27b8842fc72", + "152e4561c7e74ad0877a5160552309b4", + "72e2d3013ea14bb59d684051b1b3af10", + "1cd2ee93a65f4d4a81abc1c75d40c893", + "473c81b40a4a4750b791862c5334cbff", + "8ad962104312403eaf803d2bf4d996a1", + "bab012306a4b47ea9fcf46a0d7a7a6fa", + "166228d1351d4d2f8ee91c3c744345d1", + "9ae45a2f9f304fe1b0a2df60517a047b", + "c31e9412f7ac4f27999385a65596825a", + "05910225a05d4b3f8a00c7252fde722c", + "f8b7ca8a9f2f45d49c54db98eed9bcc4", + "1208473505b2422ab4d340f75fdf2624", + "eaf7dc090a2747b5877e57f4e425edf9", + "0f624dde85fe446092a9491797a1badc", + "ae52b5b14f254763b75e4820ed2b2a46", + "2481fee3c6f8401e86d2292b4468e631", + "1ec05e0fd379402299be8643e14af459", + "3837db24ce4f44c9acdde8e045c3b27c", + "56eacd3cfe8440a4a76805c72a8b2d6b", + "f1a528076c1846d8b62b61411a128510", + "454b1597345b4423bd7cfdb48135798f", + "aa13b7cdd8054693aef1544a6d5ebae9", + "f2172f4996634581895b73ee0fbde3b0", + "b95ec2a4fc4c4dfa8464ea5fbc8b44ba", + "535195a6b5f64e3890094ca90f8f2f8a", + "0e73f2a03d8a4eb3aaf0b82c3afd478c", + "45e10b07350d4aa8804720fa91de07dc", + "54540285fbe640d6b235079a1b7ec76b", + "ec3631996290457395f020b96f39059e", + "6a8cc3adc3b747cbba369de2f30e9755", + "ac22b083ff09461ba8d019a263b1ac7f", + "2dafaea27a0e4366b023c6ba95640c70", + "e8c5d749601d4087b26d5f3dc693a7fc", + "145d00cbda4e4fa19f0cb196a03e95a2", + "cd89abe8df764865aa5c11eab0a89edb", + "6647f50265294df1a9534b7c4c062329", + "40d8277ba9c94613bb52570b2b97a75b", + "8d4c3e7c6b554b0fba64463a031b5365", + "cd14a2c1a2ae45f48dca89ec4f84e45b", + "ae2d1da470b2401e8145d1e458c3937f", + "5b63eaf4f2b44638a73c6751e89ee78c", + "8c8e6bef4c0f4f069c0a67bbc9923270", + "2dad35f0ee3c4650a6993b272ea97604", + "c3628243bd6f421d8c3a4759c4af037e", + "8aa897cf6ab94954b05f025a70ac74fa", + "850516d144dd46d1aa2ad8dfa44ed356", + "21513d8d090c45a492e38d1b0a8d1b0b", + "bc8c53cacd4f4cf6a9ae68ec46d5599e", + "d6c219545379465b81a638a0bd0cf5ab", + "c956ee98682d4ffd989d13d11e4b76b1", + "59ac6c87be9840b1b139ecd8f1ebe4f0", + "f6edc0c3386549419ebdd1aa9629720e", + "d6fab987b0ee49e7bd81a0662d542f95", + "1b36ff4df62e415986e872a517090489", + "213556c80e074d62b9f0846a3d03c6c4", + "cc0975931770476d95b005c2563add64", + "940a7d31a7ec4658b643c1c42ef2ce7e", + "bb0290be80194bffad16464ee66a53ef", + "0d449cccd9ae476a8afeb48f8064bdbf", + "160daa533558466bbadb9deadf06f5fa", + "81ace1e936c844739907de1a9d505ad6", + "7fb2ba8001fb4b9088970b067689451b", + "5f352652b4304795ac2498ca48c38acd", + "7fc7a26fedb34b4e9c0e179f1488a518", + "a80f84c3148e4520885d34fad70a12f8", + "1ffbd742231947a2941d0fbdd226dba1", + "17ff4d9b65144ddb8fa8dd31c2d07123", + "9c1a1447733342b8864f9613bb9e0ece", + "75650414085e46058c2d6780c165bc15", + "e89739f7f6a746cfa7d886f12a0b85e0", + "02404160a8cf400b8847bed459f99adb", + "20ebb4063ffe4605a6b32b713e40872b", + "f667456051a1407f8fec350fb8308544", + "408284147a604a61aecb1e7987674424", + "ebb85543376e49688b7d83c5658125ab", + "0947bb8cceea4bb28a87fac1f8250232", + "060891df058b4969ab26f598b3f170dd", + "5124d05b487548efa8d632d2e56a378d", + "ccdad8c4322f40c4ac57458932dbe9b3", + "dd975925ee28497d8425b2df09239464", + "c04f95832eed46538193d7b797f9890e", + "3c2139e144914573a748b2666c10b37d", + "c56c938ec5d744e0b941703f7a0ed77b", + "54db197cbfd846d3b32cf4fcb012696a", + "bd438477c5f54c7db0b05a8d18eba83b", + "78b5b937718e42eca2a5837ddb4fbc98", + "882a86c503dd41fb897a03ac864750b5", + "ea6dda3be9ca41978dc3567502a41170", + "afda154ccc33433992dd92accd0ac591", + "d31515b9f55c4894b2fa569f22cbbf13", + "7772256ec8fa4892942cd40860413f9d", + "c4399ea347c241cf9a9c49ff507fc5d6", + "9b6606f7713144a9a1f80cb374bcd42d", + "7882b1253cef420a816ab2cd13dffebb", + "8e7e4acba3fc451baf6d93aa9c241e37", + "b367d095d4634e1eaac4505b3a1bffdc", + "7ca6bbcc482c4d5588432c794ef831b1", + "873bcb9bc9cf4be1b68966a33f69ffed", + "6720555b58834eb8bef867b7ae17ab44", + "6ed844b9111c4bfeb9f1786ed547b6c5", + "4746fe0ceedf4534a42f15d0be5ca294", + "39ef7fec58834e6da5e70f600b12ad9a", + "6ecd1219fad749bdae0d35037c2b7413", + "c3d40f44bde140a1b55af0a6b356fe7e", + "7771c3a2c15c425e9a6926a4a0723763", + "83c1964e90c3435bb5747b7dbc16ab51", + "d9be164056d14240a7e4815e4b1849fb", + "8f0686cb6eb64035900ed8143f46e34d", + "b9d1d2482bb14caabaa444ccf9c61810", + "7ee5183e24204234a4151a76a8e37e80", + "d3cac820aec145d6a74cf6620d67dc6c", + "b190c98a9ed141f29be9f02816c2280b", + "ad50688abbcd4ccf9a61066c2e0fc1c3", + "f83ff6dd01c24bad90423d81874c3154", + "5561c483b5024516a87aa1f7d7b095d2", + "dd0a7f174bbd48498431bceae2aa692f", + "3e0ea1c5547241c5a1c185b36c4e514c", + "f5bd39c7f4364e86b047135a482dc946", + "ea2826306cfa4558a5d78c4b388a92f5", + "d809df30245345d9a92bc1062629ed38", + "ca33dc1d75d0479aaeb7b863e5b935af", + "d4c35e6ebb5d4826ac06c50921ef2ae0", + "0326964e0d474befb108fed0236c253b", + "d3a7df9ccbbf4ad89d34535e44ba7711", + "6507c2e2b0c34d7096607249b8c5c636", + "58733f51e37e461ab413c5955d07456d", + "f5285642b5dd4432938e96a298225ffe", + "0ad6d10f1d214401bc20c20e455bbe5f", + "56dd925432aa4c9f866ba1c1353fa532", + "eecae0d48e6d4964812b554fdf153927", + "1d1a072f1df841e29bd7475a79b41cbb", + "bc9de510af614acbb63e6df52f40b180", + "ae569cea2e2940599c76c5b42eb2a7b1", + "ed694382db2643c69498e549dd6257ed", + "a5621fadc92e4550bc4a08ad25542a42", + "b6e50daae0d6485fa4be921c49e44951", + "49c1979a183c448ba7af8340d3b6eded", + "a9fac8fea20949f8af0351be359c27ca", + "fc552eb6d9144e0fb54ced38fd137967", + "0e31715ce25c48f8b2602013c683e88e", + "00a3247d646a46649c5b1adc45f6e61e", + "237a4edbd79745ee99839d0009ca7974", + "c6893e198a9448d39c76735ea7196599", + "9841879c2a81434bb105f35c0c0df4da", + "41ba6851eafc4654b1cd1fe70fd27b72", + "bd08b32fe5d24bb78ba7bc9afae74ea6", + "bcaf3335f18f4b6fa31ad56aa347d128", + "5ca92f977bfb43db851d4db95ae90aaa", + "e63571309d9c42edb5452b94d79b6b45", + "036113a86f91452393df58707c767159", + "6ca94caf7310455bb0c688ffcef45c9e", + "669d576e6f024beb9701e43f57c2e2bd", + "1960633ff12643f2a66ef6cff4d5ab08", + "226feeb0196a40798a9a35ed28b83310", + "2e7601a2dde946bbb5fc76a49eb3c8a5", + "f41daa4104104a9f830e6fa66a65a4e5", + "1085542911ba4dea95caa46de62887e4", + "12aa1feb707e49b1a4e1a46a9b6f4978", + "f50d1ec2115f4e049309276bfb13d911", + "7dc0b113b55f4bc8aad222d6acd25fe2", + "ff9e7cf2f0f14219babf1537fbc929ea", + "1b8ddeb958b64f9db5fcd5934f5dd7ac", + "60267770a60447998c42712e77fe7c8c", + "8fe7cac3a4ff46a0a4e9a94884d857ba", + "3064c9a6f7654d6e94a859f441abe372", + "ebfe58a7b017416caa315c19edbd04d1", + "832972f79a544b33861b298ffdea3484", + "4e0d59d31090455ea10b9587d8ef11c8", + "83a5a5be6f824984a83cd991e8075864", + "c3f5b71ae28a4adeb8dc70d54f4807cb", + "a14fb84a7f6c4128ad0c54087fc98399", + "f714752eaa21402d9a301549edbc7e86", + "3e708cfca5224302972c6c13896e3019", + "417d5670a562486284b44b58d3458ef2", + "82778277f0a44a83891454cf9135beb0", + "7e9f69e2bf77462a9105267211219efd", + "6aeb0f6b8794407ead0aa701a564c84e", + "0cb4d00cbc7e4efb9f7fd15dea54dac7", + "d22fd3cddb8448feb537b362ae9041a1", + "b9a01776915a4d939a876658f2ef8984", + "30cc8609dff24add8192ad97cdc7f042", + "63da9593d9994d1080b0bd16e2e4c987", + "79714744b45745488af5545cef0c720f", + "07b06439eb084c5e9cfaecc5b5996244", + "c7c520e2701a4a51bc972cfddee30499", + "2fa444afdf754af69560e0f9b485b0a6", + "974419ca5f8d411e826f2a4077801a8f", + "f3c30b8ba65d4d2c9d5f152dfef0f478", + "26f025ffa2234877a6ffced3b1a6da51", + "991eab57abbd4fc19105b8a173e48df0", + "5b4ecdf8e7614e639d8a0541032e9980", + "de28ca925ec14b4f86cb08cdaf953246", + "2520aa9405554ecea28472beecc63d8c", + "b85c8cdc101c4791b89239c579529e3a", + "3ba1351969ae40d19c7b4a4231771d56", + "a90a52b56c224656a0b7b0c8f9e195e9", + "7e1c1e9f666346f5bf20813e1d89c40c", + "5888a3665e6d46f98e3776190f3d88d1", + "4eda23a75fc14231a320394e1704459b", + "5cedf51ced2f4aa1b9700fefcb8683c9", + "afce541f51e04b779b700bdd395be9b9", + "c96cd84932f6465fbfb73b45c0fbd390", + "ac4b54ee1eb64084947416a0598c5531", + "7804eed2d5f140789848956e85f47964", + "6238f55ec53d46a3b239fffe41564abe", + "176d069c0e1d464ea00ebf26997860b7", + "39c291de497a4b9f80446153862920dc", + "c0580c1200ae4329beb9746e6014df57", + "d3b1db7a42784782b269924e5acf01b1", + "d6aad418994141818e52e7f7e15cb395", + "281e4f80cc94472fbb477c08990d6a12", + "1ef7fc87f5044c1bbedeaabbaa611e35", + "821cd12c69cb412aa814f4a435e15d3f", + "11f1e3ee5b754675ad6c23373b146d93", + "0d979fbe8f374ddba5883007e1257b61", + "2c467a6882ce41d694ca7427e3fee1c7", + "9deb32ec403340cbab9cb7ff9fb1e695", + "3679794cc9864a11ba7e8adfc2c169fb", + "8921caaee769428d80124c8838a3afcc", + "686597fa4a0a422c93eeed3a921151d3", + "8311331c9df54dd48a646234f483f2b9", + "014447a3fc5d48ec8e1b23ff4160e6e5", + "a8a2e244598c4198b2798c3add87387e", + "c3d4ad9cbdba425681e8ea1d9eacc517", + "a580de56d00242d8a0cc32ad0c058966", + "701087566c2b43d3bda29e5880cf8b8d", + "0e3a2e96e0ca4c758073f29980686847", + "3ba43419ab4646e795850a1c845ed2e4", + "76bdb4d6547d45afb26a4280f2d80fe1", + "e50590f164474358b7e3906e97712a99", + "01c746e3760b4d6abb558b7225a4baec", + "8650efda983f4a7faf8ef53db4940bb2", + "74e48d4a8d5b40b28c34f83747e6787d", + "d250e38369ba4a3aa5b252094a8a64d4", + "685925c238db45f2a2a8a7d21f28409b", + "8c6c12887ec74e26b54233315d7281c3", + "29f539376c974eeeb112c7192edbfbe6", + "bee2979d2ef54113bfce39025caeedb3", + "96b4a032bfae4f69bfbc827cc5e87ae1", + "5754663c6bd24e358d0b8f2fafc150e8", + "b79e971158ea42caaf3204ad582e4acb", + "a7bc995b6ef8467c8a6ef63da807bec6", + "c46de3bd9e39471fa8826ace3e593e34", + "b6035ff4e176400cb12400f639ed222c", + "31db83b5d183478a83b249fa55abea52", + "2599b0241a1c4c03b81ad8c4f894196c", + "f016e66bd00c462aba684b558f896bde", + "68b65617c5f64a74ba6078ed325f2c5d", + "086869a1529d4b1d951702359d013603", + "f10173ae54894b08abdd88f1337dd675", + "ad6eaa821fbb4d1586805401b998b3ab", + "cd3a1bcf9f064cacb0446ffd70c0f5ec", + "928ecbd873974212a48eeefc62c7d46a", + "e6f2f9ba954e428fb9697d6d77cd8f50", + "0d65f204a24945ac8b5dce0912410924", + "53c6a488d72840fd871b8d1b672815e1", + "527f0f3cdd754811ab288d0e3ba54ecd", + "66370b7b5d6b4d22a1e4f8c4c1e2e893", + "4ba69633b4184008806878b702e3e834", + "50190b37f58949a4a5ff88bab653d211", + "5f8e32bb621d4235894a6e81cc2a2bbc", + "9adc0b70c4ad43db90ea9d871c07cd62", + "8e26ca16de8e4db8afe8a0773d2a9782", + "dd0a0337da5a4fc6b6d8c5170f13d7a2", + "3c22071591ca4bd2890b250e91232451", + "65a92039c1d94f1bae1fdb8a6358d9a3", + "298fb7d77549426480ca49ff41c2e5b6", + "2578c814a4b44ee2b41c8dd397dd8714", + "0b37c34db84946058a6f471002e4db0d", + "c2c360cb61024279a9e8a867514f3903", + "afa207ff18074896bbf0f6ec9840c5f3", + "0e7fcd7d29c9412ea0397a5fe53c0b83", + "0946eb66f7af47bdbf7d040aa260ebea", + "45e244788ee2426bb492171e5cb72d46", + "264ac655bc6748f09f441e6d0b68e974", + "fa1d910e3ad94f5ba006ebf1bac1c8c1", + "aeecfc01b3eb41bfb7277e40805c914f", + "b03546ce252c4fdf9bb8e828e87951fc", + "fd5915c6296f4bb6bd5439a918345c7a", + "2bb7313bc37c4ae48cdbc2f3e4c8ffff", + "200faf085a1b48c9b65cdc00f40f5330", + "7ef501ab01ec4305ab55274000ce513d", + "ea63c7becf2f4da786ec1a17ca848b4e", + "f7abafc20e7a4d4e8a3e1d85589d1cff", + "7c3402a99e9349c9a2849af2eedd8edf", + "0627a9c525594144906b94adec04687d", + "a3e9d05981b04f10829d10c3a2087a4c", + "ca0b14b068f042768cc84f71d8c39428", + "be99131b676f4b0fa91b1f8b8cbae825", + "686c0955edcf48d4823b4497a4a16e5d", + "598c4b8ce296478f8a63a7d51bf54882", + "f7ede2830bbd4ba391268f5416cdea2e", + "81a006c41ff046b3803a6bfbb577dd2d", + "c8f89d4734944fe69f30b1fdc6d396a3", + "1fce43f844104e15a1fe1e4e6f2bacf6", + "33c8984b05be45249a9cec4b6f30e557", + "5b98619e1642470a83db408af792086a", + "e889ea7b6d374e11a17e533e05b4802e", + "876435064a6446488a819cb8e57fd165", + "92c7a231bb9648be9fc53628f41b64be", + "1190b01146ec41f7a0aceee35df65021", + "7b1350e28f0f47ba8cefc94189f5fe3d", + "a2aace6cb622465fb39f75bc80365b9b", + "b952a86b6b504e6890e4e82157a87b69", + "588e5cbb4faa40fda9c24258fe8f983c", + "c050d49700d74c93a6b28a09dfc36bbc", + "5fe3c5c3200645eb9609310043194ebd", + "76e45856e51d4e38b27c3d04bff6809f", + "7a9afe15dd9c4ba3b6791e724ff46610", + "c88d8ec092fb40828a3149790ef51c74", + "01a6be0efa3f4ce5a1ff848c9077142c", + "b3fb82365c43434086393575c36bfaef", + "9f581d40efd7478bad2c66140e577d16", + "36680e0aac814e9abdbb95b4f2489442", + "2086ecef6725402aa77717a82d5fbbc4", + "0e690f0a05f34a55a366078cc23e2568", + "db3b5313ec074f2982a8cf7f4bed9ed2", + "c10ec37dcf0a4f488167cfc3ca11a172", + "bce09c6f8fa84eb090d75b2f4b67a00f", + "43931fd664fe4d1795255a891e665f6f", + "56bcbbb34a6e4fe9bccf9c5ad2566d0f", + "df7bb97a5eb449e0ab6375a95da8a117", + "aaa6e63627b44fcb91a37f4738a05080", + "7b9c6ee0ef4d4d47bbbcea900926c99b", + "95dd033d9a7a44969129f78cccb7643f", + "6540329b1c7a4e7c90ad5bc20808b47d", + "1efee3f357d04e29aa3480b047f28d1d", + "aeb1adb24db24044b2737d71dff859fe", + "5e882f7275344a9b8cb4c5cc9eb135a6", + "51121ce95fcf4cebac50bfd1f57acae7", + "dbead73c0da64125ab069298107bfd24", + "32cbe71c52534f1aaf7494c96d0247ea", + "f277c93f6ae440478501bf147cd7d642", + "06ebf21b46be43b0adf1550edaf1d796", + "752f13b1a37b477d8ad32d2dd86f0ea1", + "533a176e5fb54491a61477ea6c4763e9", + "7e3ec2fc6557421f8a10bd04ed29c5fb", + "b6878cee877240ee88b39773f56c4b45", + "fa4c81193d4e4446a945d79e74d27794", + "47885cddea8f43ee91a70e65f1cf8c1e", + "203a367fe5a147a18e0bc5caa4b2e540", + "0089739b556f4083bb15f50adb715a49", + "a1b7db5b93d447c989bab61e5797f609", + "0b7fea7a9a2a4872b3ef551d6d85220c", + "837702e731cf444197c2de70871d5360", + "fdbd924d6e434a5c92e4df4513cce214", + "604c30e532e047c095dbb49ef04e3598", + "6bd4f6b0d34a4366a96feb1b58a45ea7", + "f63a2bfe570041baa07cb0ca4ea19f88", + "27379643e5c6411f8e1b04a53c82e392", + "844a7fc3dcb74040af34e404d362c12e", + "3071712324ba435789ba93d9a0986f76", + "3e2fd6381a6f4c629ea789fd2d3d42df", + "6f740f0f307d42eaa45bacabbf43a0b5", + "0415d00a8c4b4fff996550ae14b72482", + "1a15e71130f54349912fc4415673b1f5", + "7191c900a189420cb189affb5002fa71", + "122d582ae16a4f53b4a0ca0c76bc325d", + "efd40e56f61e455e8496e56dbdd5b24e", + "9254a68c5bfb483fbe9a99a7a1ce0e8e", + "12d0b7c4839b43e5bab001a928efb340", + "2c5aa8935bf643baa4fd877792614ecd", + "d645e112376644cbacceac5c00764f88", + "d783c5440f2749cda662c01bb21610e7", + "6f0f01c52a4b4c9ea2964de549f52856", + "fe3f3ba39af74239a24a7392477d5d10", + "209c90d800bb43678308f30a32293a11", + "e4af1be30b1f46edaa0aba62cda01c59", + "ac3c82b658ab4969b2f2666693ef865a", + "5fc2dbb4097d40d180f026b893e56149", + "11f83c8fc3b742c79d672b6ff72164a5", + "1741fd84e52e41b3a9da234c1cadf5be", + "1f4197197afa4dd480cc296e520c409c", + "e9b44aeccd92433fb829e6bc88e4f1b5" + ] + }, + "outputId": "658de5e8-293a-45d6-89cd-d302fbabddd2" + }, + "outputs": [ + { + "output_type": "stream", + "name": "stdout", + "text": [ + "running: clip-base32 + opt-125m (frozen LM, longer train)\n", + "Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.processors.label_processor:Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.processors.label_processor:Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.processors.label_processor:Label answer vocab: {'no': 0, 'yes': 1}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "MedFlamingo(\n", + " (_vision_encoder): CLIPVisionModel(\n", + " (vision_model): CLIPVisionTransformer(\n", + " (embeddings): CLIPVisionEmbeddings(\n", + " (patch_embedding): Conv2d(3, 768, kernel_size=(32, 32), stride=(32, 32), bias=False)\n", + " (position_embedding): Embedding(50, 768)\n", + " )\n", + " (pre_layrnorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (encoder): CLIPEncoder(\n", + " (layers): ModuleList(\n", + " (0-11): 12 x CLIPEncoderLayer(\n", + " (self_attn): CLIPAttention(\n", + " (k_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (v_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (q_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (out_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " (layer_norm1): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (mlp): CLIPMLP(\n", + " (activation_fn): QuickGELUActivation()\n", + " (fc1): Linear(in_features=768, out_features=3072, bias=True)\n", + " (fc2): Linear(in_features=3072, out_features=768, bias=True)\n", + " )\n", + " (layer_norm2): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " )\n", + " (post_layernorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " (_lang_model): OPTForCausalLM(\n", + " (model): OPTModel(\n", + " (decoder): OPTDecoder(\n", + " (embed_tokens): Embedding(50272, 768, padding_idx=1)\n", + " (embed_positions): OPTLearnedPositionalEmbedding(2050, 768)\n", + " (final_layer_norm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (layers): ModuleList(\n", + " (0-11): 12 x OPTDecoderLayer(\n", + " (self_attn): OPTAttention(\n", + " (k_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (v_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (q_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (out_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " (activation_fn): ReLU()\n", + " (self_attn_layer_norm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (fc1): Linear(in_features=768, out_features=3072, bias=True)\n", + " (fc2): Linear(in_features=3072, out_features=768, bias=True)\n", + " (final_layer_norm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " )\n", + " )\n", + " (lm_head): Linear(in_features=768, out_features=50272, bias=False)\n", + " )\n", + " (_xattn_layers): ModuleList(\n", + " (0-2): 3 x MedFlamingoLayer(\n", + " (perceiver_resampler): PerceiverResampler(\n", + " (cross_attn_layers): ModuleList(\n", + " (0-5): 6 x MultiheadAttention(\n", + " (out_proj): NonDynamicallyQuantizableLinear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " )\n", + " (ff_layers): ModuleList(\n", + " (0-5): 6 x Sequential(\n", + " (0): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (1): Linear(in_features=768, out_features=3072, bias=True)\n", + " (2): GELU(approximate='none')\n", + " (3): Dropout(p=0.1, inplace=False)\n", + " (4): Linear(in_features=3072, out_features=768, bias=True)\n", + " (5): Dropout(p=0.1, inplace=False)\n", + " )\n", + " )\n", + " (norms): ModuleList(\n", + " (0-5): 6 x LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " (vision_proj): Identity()\n", + " (norm_lang): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (gated_xattn): MultiheadAttention(\n", + " (out_proj): NonDynamicallyQuantizableLinear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " (norm_ff): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (ff): Sequential(\n", + " (0): Linear(in_features=768, out_features=3072, bias=True)\n", + " (1): GELU(approximate='none')\n", + " (2): Dropout(p=0.1, inplace=False)\n", + " (3): Linear(in_features=3072, out_features=768, bias=True)\n", + " )\n", + " )\n", + " )\n", + " (_fc): Linear(in_features=768, out_features=2, bias=True)\n", + ")\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:MedFlamingo(\n", + " (_vision_encoder): CLIPVisionModel(\n", + " (vision_model): CLIPVisionTransformer(\n", + " (embeddings): CLIPVisionEmbeddings(\n", + " (patch_embedding): Conv2d(3, 768, kernel_size=(32, 32), stride=(32, 32), bias=False)\n", + " (position_embedding): Embedding(50, 768)\n", + " )\n", + " (pre_layrnorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (encoder): CLIPEncoder(\n", + " (layers): ModuleList(\n", + " (0-11): 12 x CLIPEncoderLayer(\n", + " (self_attn): CLIPAttention(\n", + " (k_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (v_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (q_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (out_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " (layer_norm1): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (mlp): CLIPMLP(\n", + " (activation_fn): QuickGELUActivation()\n", + " (fc1): Linear(in_features=768, out_features=3072, bias=True)\n", + " (fc2): Linear(in_features=3072, out_features=768, bias=True)\n", + " )\n", + " (layer_norm2): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " )\n", + " (post_layernorm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " (_lang_model): OPTForCausalLM(\n", + " (model): OPTModel(\n", + " (decoder): OPTDecoder(\n", + " (embed_tokens): Embedding(50272, 768, padding_idx=1)\n", + " (embed_positions): OPTLearnedPositionalEmbedding(2050, 768)\n", + " (final_layer_norm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (layers): ModuleList(\n", + " (0-11): 12 x OPTDecoderLayer(\n", + " (self_attn): OPTAttention(\n", + " (k_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (v_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (q_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " (out_proj): Linear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " (activation_fn): ReLU()\n", + " (self_attn_layer_norm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (fc1): Linear(in_features=768, out_features=3072, bias=True)\n", + " (fc2): Linear(in_features=3072, out_features=768, bias=True)\n", + " (final_layer_norm): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " )\n", + " )\n", + " (lm_head): Linear(in_features=768, out_features=50272, bias=False)\n", + " )\n", + " (_xattn_layers): ModuleList(\n", + " (0-2): 3 x MedFlamingoLayer(\n", + " (perceiver_resampler): PerceiverResampler(\n", + " (cross_attn_layers): ModuleList(\n", + " (0-5): 6 x MultiheadAttention(\n", + " (out_proj): NonDynamicallyQuantizableLinear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " )\n", + " (ff_layers): ModuleList(\n", + " (0-5): 6 x Sequential(\n", + " (0): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (1): Linear(in_features=768, out_features=3072, bias=True)\n", + " (2): GELU(approximate='none')\n", + " (3): Dropout(p=0.1, inplace=False)\n", + " (4): Linear(in_features=3072, out_features=768, bias=True)\n", + " (5): Dropout(p=0.1, inplace=False)\n", + " )\n", + " )\n", + " (norms): ModuleList(\n", + " (0-5): 6 x LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " )\n", + " )\n", + " (vision_proj): Identity()\n", + " (norm_lang): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (gated_xattn): MultiheadAttention(\n", + " (out_proj): NonDynamicallyQuantizableLinear(in_features=768, out_features=768, bias=True)\n", + " )\n", + " (norm_ff): LayerNorm((768,), eps=1e-05, elementwise_affine=True)\n", + " (ff): Sequential(\n", + " (0): Linear(in_features=768, out_features=3072, bias=True)\n", + " (1): GELU(approximate='none')\n", + " (2): Dropout(p=0.1, inplace=False)\n", + " (3): Linear(in_features=3072, out_features=768, bias=True)\n", + " )\n", + " )\n", + " )\n", + " (_fc): Linear(in_features=768, out_features=2, bias=True)\n", + ")\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Metrics: ['accuracy', 'f1_macro']\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Metrics: ['accuracy', 'f1_macro']\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Device: cuda\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Device: cuda\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Training:\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Training:\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Batch size: 8\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Batch size: 8\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer params: {'lr': 0.0001}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer params: {'lr': 0.0001}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Epochs: 8\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Epochs: 8\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Epoch 0 / 8: 0%| | 0/111 [00:00\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer params: {'lr': 3e-05}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer params: {'lr': 3e-05}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Epoch 0 / 6: 0%| | 0/111 [00:00\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer params: {'lr': 0.0001}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer params: {'lr': 0.0001}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Epochs: 8\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Epochs: 8\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Epoch 0 / 8: 0%| | 0/111 [00:00\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer params: {'lr': 3e-05}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer params: {'lr': 3e-05}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Epoch 0 / 6: 0%| | 0/222 [00:00\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Optimizer params: {'lr': 5e-05}\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Optimizer params: {'lr': 5e-05}\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Weight decay: 0.0001\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Max grad norm: None\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Val dataloader: \n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor: f1_macro\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Monitor criterion: max\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Epochs: 6\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:Patience: 3\n" + ] + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "\n" + ] + }, + { + "output_type": "stream", + "name": "stderr", + "text": [ + "INFO:pyhealth.trainer:\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Epoch 0 / 6: 0%| | 0/111 [00:00\n", + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
accuracyf1_macrolossnamevision_model_namelang_model_namefreeze_lmfreeze_visionepochslrbatch_sizepatience
00.6411480.6371280.824713clip-base32 + distilgpt2 (unfrozen LM)openai/clip-vit-base-patch32distilgpt2FalseTrue60.0000583
10.6363640.6339420.902683clip-base32 + opt-125m (unfrozen LM, low LR)openai/clip-vit-base-patch32facebook/opt-125mFalseTrue60.0000383
20.6028710.6019600.970539clip-base32 + opt-125m (frozen LM, longer train)openai/clip-vit-base-patch32facebook/opt-125mTrueTrue80.0001083
30.5933010.5929661.074032clip-base16 + opt-125m (frozen LM)openai/clip-vit-base-patch16facebook/opt-125mTrueTrue80.0001083
40.5885170.5857761.141470clip-base16 + opt-125m (unfrozen LM, low LR)openai/clip-vit-base-patch16facebook/opt-125mFalseTrue60.0000343
\n", + "
\n", + "
\n", + "\n", + "
\n", + " \n", + "\n", + " \n", + "\n", + " \n", + "
\n", + "\n", + "\n", + "
\n", + " \n" + ], + "application/vnd.google.colaboratory.intrinsic+json": { + "type": "dataframe", + "summary": "{\n \"name\": \" print(\\\"Skipping PyHealth classifier sweep\",\n \"rows\": 5,\n \"fields\": [\n {\n \"column\": \"accuracy\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.02463069411719379,\n \"min\": 0.5885167464114832,\n \"max\": 0.6411483253588517,\n \"num_unique_values\": 5,\n \"samples\": [\n 0.6363636363636364,\n 0.5885167464114832,\n 0.6028708133971292\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"f1_macro\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.023717530915923487,\n \"min\": 0.585776179941003,\n \"max\": 0.6371275783040489,\n \"num_unique_values\": 5,\n \"samples\": [\n 0.6339417404129793,\n 0.585776179941003,\n 0.6019595695371835\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"loss\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.12753429637269645,\n \"min\": 0.8247126909317793,\n \"max\": 1.1414699515345383,\n \"num_unique_values\": 5,\n \"samples\": [\n 0.9026827889460104,\n 1.1414699515345383,\n 0.9705385046976583\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"name\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 5,\n \"samples\": [\n \"clip-base32 + opt-125m (unfrozen LM, low LR)\",\n \"clip-base16 + opt-125m (unfrozen LM, low LR)\",\n \"clip-base32 + opt-125m (frozen LM, longer train)\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"vision_model_name\",\n \"properties\": {\n \"dtype\": \"category\",\n \"num_unique_values\": 2,\n \"samples\": [\n \"openai/clip-vit-base-patch16\",\n \"openai/clip-vit-base-patch32\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"lang_model_name\",\n \"properties\": {\n \"dtype\": \"category\",\n \"num_unique_values\": 2,\n \"samples\": [\n \"facebook/opt-125m\",\n \"distilgpt2\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"freeze_lm\",\n \"properties\": {\n \"dtype\": \"boolean\",\n \"num_unique_values\": 2,\n \"samples\": [\n true,\n false\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"freeze_vision\",\n \"properties\": {\n \"dtype\": \"boolean\",\n \"num_unique_values\": 1,\n \"samples\": [\n true\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"epochs\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 1,\n \"min\": 6,\n \"max\": 8,\n \"num_unique_values\": 2,\n \"samples\": [\n 8\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"lr\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 3.5637059362410926e-05,\n \"min\": 3e-05,\n \"max\": 0.0001,\n \"num_unique_values\": 3,\n \"samples\": [\n 5e-05\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"batch_size\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 1,\n \"min\": 4,\n \"max\": 8,\n \"num_unique_values\": 2,\n \"samples\": [\n 4\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"patience\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0,\n \"min\": 3,\n \"max\": 3,\n \"num_unique_values\": 1,\n \"samples\": [\n 3\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n }\n ]\n}" + } + }, + "metadata": {} + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "best PyHealth sweep run:\n", + "{\n", + " \"accuracy\": 0.6411483253588517,\n", + " \"f1_macro\": 0.6371275783040489,\n", + " \"loss\": 0.8247126909317793,\n", + " \"name\": \"clip-base32 + distilgpt2 (unfrozen LM)\",\n", + " \"vision_model_name\": \"openai/clip-vit-base-patch32\",\n", + " \"lang_model_name\": \"distilgpt2\",\n", + " \"freeze_lm\": false,\n", + " \"freeze_vision\": true,\n", + " \"epochs\": 6,\n", + " \"lr\": 5e-05,\n", + " \"batch_size\": 8,\n", + " \"patience\": 3\n", + "}\n", + "saved to /content/PyHealth/output/colab_eval/pyhealth_classifier_sweep.csv\n" + ] + } + ], + "source": [ + "\n", + "if RUN_PYHEALTH_SWEEP:\n", + " import gc\n", + " import random\n", + " import numpy as np\n", + "\n", + " from pyhealth.datasets import get_dataloader\n", + " from pyhealth.models.medflamingo import MedFlamingo\n", + " from pyhealth.trainer import Trainer\n", + "\n", + " def set_seed(seed: int = 42):\n", + " random.seed(seed)\n", + " np.random.seed(seed)\n", + " torch.manual_seed(seed)\n", + " if torch.cuda.is_available():\n", + " torch.cuda.manual_seed_all(seed)\n", + "\n", + " def run_pyhealth_experiment(config):\n", + " set_seed(SEED)\n", + " train_dataset = make_dataset(train_rows, image_size=config.get(\"image_size\", 224))\n", + " val_dataset = make_dataset(val_rows, image_size=config.get(\"image_size\", 224))\n", + " test_dataset = make_dataset(test_rows, image_size=config.get(\"image_size\", 224))\n", + "\n", + " train_loader = get_dataloader(train_dataset, batch_size=config.get(\"batch_size\", 4), shuffle=True)\n", + " val_loader = get_dataloader(val_dataset, batch_size=config.get(\"batch_size\", 4), shuffle=False)\n", + " test_loader = get_dataloader(test_dataset, batch_size=config.get(\"batch_size\", 4), shuffle=False)\n", + "\n", + " model = MedFlamingo(\n", + " dataset=train_dataset,\n", + " vision_model_name=config[\"vision_model_name\"],\n", + " lang_model_name=config[\"lang_model_name\"],\n", + " freeze_vision=config.get(\"freeze_vision\", True),\n", + " freeze_lm=config.get(\"freeze_lm\", True),\n", + " )\n", + "\n", + " trainer = Trainer(\n", + " model=model,\n", + " metrics=[\"accuracy\", \"f1_macro\"],\n", + " device=DEVICE,\n", + " enable_logging=False,\n", + " )\n", + " trainer.train(\n", + " train_dataloader=train_loader,\n", + " val_dataloader=val_loader,\n", + " epochs=config.get(\"epochs\", 4),\n", + " optimizer_params={\"lr\": config.get(\"lr\", 1e-4)},\n", + " weight_decay=config.get(\"weight_decay\", 1e-4),\n", + " monitor=\"f1_macro\",\n", + " monitor_criterion=\"max\",\n", + " patience=config.get(\"patience\", 2),\n", + " )\n", + " metrics = trainer.evaluate(test_loader)\n", + " _, _, loss_mean = trainer.inference(test_loader)\n", + " metrics.update(config)\n", + " metrics[\"loss\"] = float(loss_mean)\n", + " return metrics\n", + "\n", + " sweep = [\n", + " {\n", + " \"name\": \"clip-base32 + opt-125m (frozen LM, longer train)\",\n", + " \"vision_model_name\": \"openai/clip-vit-base-patch32\",\n", + " \"lang_model_name\": \"facebook/opt-125m\",\n", + " \"freeze_lm\": True,\n", + " \"freeze_vision\": True,\n", + " \"epochs\": 8,\n", + " \"lr\": 1e-4,\n", + " \"batch_size\": 8,\n", + " \"patience\": 3,\n", + " },\n", + " {\n", + " \"name\": \"clip-base32 + opt-125m (unfrozen LM, low LR)\",\n", + " \"vision_model_name\": \"openai/clip-vit-base-patch32\",\n", + " \"lang_model_name\": \"facebook/opt-125m\",\n", + " \"freeze_lm\": False,\n", + " \"freeze_vision\": True,\n", + " \"epochs\": 6,\n", + " \"lr\": 3e-5,\n", + " \"batch_size\": 8,\n", + " \"patience\": 3,\n", + " },\n", + " {\n", + " \"name\": \"clip-base16 + opt-125m (frozen LM)\",\n", + " \"vision_model_name\": \"openai/clip-vit-base-patch16\",\n", + " \"lang_model_name\": \"facebook/opt-125m\",\n", + " \"freeze_lm\": True,\n", + " \"freeze_vision\": True,\n", + " \"epochs\": 8,\n", + " \"lr\": 1e-4,\n", + " \"batch_size\": 8,\n", + " \"patience\": 3,\n", + " },\n", + " {\n", + " \"name\": \"clip-base16 + opt-125m (unfrozen LM, low LR)\",\n", + " \"vision_model_name\": \"openai/clip-vit-base-patch16\",\n", + " \"lang_model_name\": \"facebook/opt-125m\",\n", + " \"freeze_lm\": False,\n", + " \"freeze_vision\": True,\n", + " \"epochs\": 6,\n", + " \"lr\": 3e-5,\n", + " \"batch_size\": 4,\n", + " \"patience\": 3,\n", + " },\n", + " {\n", + " \"name\": \"clip-base32 + distilgpt2 (unfrozen LM)\",\n", + " \"vision_model_name\": \"openai/clip-vit-base-patch32\",\n", + " \"lang_model_name\": \"distilgpt2\",\n", + " \"freeze_lm\": False,\n", + " \"freeze_vision\": True,\n", + " \"epochs\": 6,\n", + " \"lr\": 5e-5,\n", + " \"batch_size\": 8,\n", + " \"patience\": 3,\n", + " },\n", + " ]\n", + "\n", + " sweep_rows = []\n", + " for config in sweep:\n", + " print(f\"running: {config['name']}\")\n", + " try:\n", + " row = run_pyhealth_experiment(config)\n", + " except Exception as exc:\n", + " row = {\"name\": config[\"name\"], \"error\": repr(exc)}\n", + " sweep_rows.append(row)\n", + " gc.collect()\n", + " if torch.cuda.is_available():\n", + " torch.cuda.empty_cache()\n", + "\n", + " sweep_df = pd.DataFrame(sweep_rows)\n", + " sweep_csv = RESULTS_DIR / \"pyhealth_classifier_sweep.csv\"\n", + " sweep_df.to_csv(sweep_csv, index=False)\n", + " display(sweep_df.sort_values([\"accuracy\", \"f1_macro\"], ascending=False, na_position=\"last\").reset_index(drop=True))\n", + "\n", + " valid_sweep_df = sweep_df.dropna(subset=[\"accuracy\", \"f1_macro\"], how=\"any\")\n", + " if not valid_sweep_df.empty:\n", + " best_pyhealth_sweep = valid_sweep_df.sort_values([\"accuracy\", \"f1_macro\"], ascending=False).iloc[0].to_dict()\n", + " print(\"best PyHealth sweep run:\")\n", + " print(json.dumps(best_pyhealth_sweep, indent=2, default=str))\n", + " else:\n", + " best_pyhealth_sweep = None\n", + "\n", + " print(\"saved to\", sweep_csv)\n", + "else:\n", + " print(\"Skipping PyHealth classifier sweep.\")\n" + ], + "id": "pyhealth-sweep" + }, + { + "cell_type": "markdown", + "metadata": { + "id": "official-notes" + }, + "source": [ + "## Official Med-Flamingo stretch track\n", + "\n", + "Use this section only on a **high-memory Colab GPU** such as A100 or L4.\n", + "\n", + "What this section does:\n", + "\n", + "- loads the official `med-flamingo/med-flamingo` checkpoint,\n", + "- builds few-shot prompts on VQA-RAD,\n", + "- generates open-ended answers,\n", + "- scores them with exact match, BERTScore-F1, and yes/no accuracy when applicable.\n", + "\n", + "What this section does **not** do:\n", + "\n", + "- it does not reproduce the paper's clinician-rating protocol,\n", + "- it does not guarantee paper-matching numbers,\n", + "- it does not magically make the PyHealth classifier path paper-faithful.\n", + "\n", + "It is still valuable because it gives you a real generative evaluation path using the official checkpoint instead of the classifier-only fallback." + ], + "id": "official-notes" + }, + { + "cell_type": "code", + "execution_count": 7, + "metadata": { + "id": "official-helpers" + }, + "outputs": [], + "source": [ + "\n", + "if RUN_OFFICIAL_MEDFLAMINGO:\n", + " import gc\n", + " import re\n", + " import string\n", + " import random\n", + " import warnings\n", + " from collections import Counter\n", + "\n", + " import numpy as np\n", + " from PIL import Image\n", + " from einops import repeat\n", + " from bert_score import score as bert_score\n", + " from huggingface_hub import hf_hub_download\n", + " from sklearn.metrics import accuracy_score, confusion_matrix, f1_score\n", + " from transformers.utils import logging as transformers_logging\n", + "\n", + " transformers_logging.set_verbosity_error()\n", + " warnings.filterwarnings(\"ignore\", message=r\"The following generation flags are not valid\")\n", + "\n", + " subprocess.run([\n", + " sys.executable,\n", + " \"-m\",\n", + " \"pip\",\n", + " \"install\",\n", + " \"-q\",\n", + " \"open-flamingo==0.0.2\",\n", + " \"einops-exts\",\n", + " \"open_clip_torch\",\n", + " ], check=True)\n", + "\n", + " from open_flamingo import create_model_and_transforms\n", + "\n", + " HF_TOKEN = os.environ.get(\"HF_TOKEN\")\n", + " if IN_COLAB:\n", + " try:\n", + " from google.colab import userdata\n", + " HF_TOKEN = HF_TOKEN or userdata.get(\"HF_TOKEN\")\n", + " except Exception:\n", + " pass\n", + "\n", + " PROMPT_PREFIXES = [\n", + " \"You are answering medical visual questions from radiology images. Respond with exactly one lowercase word: yes or no. Do not explain your answer.\",\n", + " \"You are a medical VQA assistant. Choose only one answer from {yes, no}. Output only the answer word in lowercase.\",\n", + " \"Read the image and question carefully. Reply using only yes or no. No punctuation. No explanation.\",\n", + " ]\n", + "\n", + " def normalize_answer(text: str) -> str:\n", + " text = str(text).strip().lower()\n", + " text = text.translate(str.maketrans(\"\", \"\", string.punctuation))\n", + " text = re.sub(r\"\\s+\", \" \", text)\n", + " return text.strip()\n", + "\n", + " def normalize_yesno_prediction(text: str) -> str:\n", + " text = normalize_answer(text)\n", + " if not text:\n", + " return \"\"\n", + " tokens = text.split()\n", + " yes_positions = [idx for idx, token in enumerate(tokens) if token == 'yes']\n", + " no_positions = [idx for idx, token in enumerate(tokens) if token == 'no']\n", + " if yes_positions and no_positions:\n", + " return 'yes' if yes_positions[0] < no_positions[0] else 'no'\n", + " if yes_positions:\n", + " return 'yes'\n", + " if no_positions:\n", + " return 'no'\n", + " if text in {'y', 'yeah', 'yep'}:\n", + " return 'yes'\n", + " if text in {'n', 'nah', 'nope'}:\n", + " return 'no'\n", + " return \"\"\n", + "\n", + " def clean_generation(text: str) -> str:\n", + " text = text.replace(\"\", \" \").strip()\n", + " if \"Answer:\" in text:\n", + " text = text.split(\"Answer:\")[-1]\n", + " text = text.split(\"<|endofchunk|>\")[0]\n", + " text = text.split(\"Question:\")[0]\n", + " text = text.split(\"\\n\")[0]\n", + " return text.strip()\n", + "\n", + " def build_few_shot_prompt(examples, query_question, variant: int = 0):\n", + " prompt = PROMPT_PREFIXES[variant % len(PROMPT_PREFIXES)] + \" \"\n", + " for ex in examples:\n", + " prompt += f\"Question: {ex['question']}\\nAnswer: {normalize_answer(ex['answer'])}<|endofchunk|>\"\n", + " prompt += f\"Question: {query_question}\\nAnswer:\"\n", + " return prompt\n", + "\n", + " def balanced_sample(candidates, k: int, rng: random.Random):\n", + " yes_pool = [row for row in candidates if row['answer'] == 'yes']\n", + " no_pool = [row for row in candidates if row['answer'] == 'no']\n", + " rng.shuffle(yes_pool)\n", + " rng.shuffle(no_pool)\n", + " target_yes = k // 2\n", + " target_no = k - target_yes\n", + " selected = yes_pool[:min(target_yes, len(yes_pool))] + no_pool[:min(target_no, len(no_pool))]\n", + " selected_ids = {row['sample_id'] for row in selected}\n", + " remaining = [row for row in candidates if row['sample_id'] not in selected_ids]\n", + " rng.shuffle(remaining)\n", + " selected.extend(remaining[:max(0, k - len(selected))])\n", + " return selected[:k]\n", + "\n", + " def sample_few_shot_examples(pool, query_row, k=6, seed=42, vote_idx: int = 0):\n", + " rng = random.Random(seed + (query_row['sample_id'] + 1) * 1009 + vote_idx * 9173)\n", + " candidates = [\n", + " row for row in pool\n", + " if row['image_name'] != query_row['image_name']\n", + " and row['question'].strip().lower() != query_row['question'].strip().lower()\n", + " and row['answer'] in {'yes', 'no'}\n", + " ]\n", + " levels = [\n", + " [row for row in candidates if row['question_type'] == query_row['question_type'] and row['image_organ'] == query_row['image_organ']],\n", + " [row for row in candidates if row['question_type'] == query_row['question_type']],\n", + " [row for row in candidates if row['image_organ'] == query_row['image_organ']],\n", + " candidates,\n", + " ]\n", + " for level in levels:\n", + " if len(level) >= k and {'yes', 'no'}.issubset({row['answer'] for row in level}):\n", + " return balanced_sample(level, k, rng)\n", + " return balanced_sample(candidates, k, rng)\n", + "\n", + " class FlamingoProcessor:\n", + " def __init__(self, tokenizer, vision_processor):\n", + " self.tokenizer = tokenizer\n", + " self.vision_processor = vision_processor\n", + "\n", + " def encode_text(self, prompt):\n", + " self.tokenizer.padding_side = \"left\"\n", + " return self.tokenizer([prompt], return_tensors=\"pt\")\n", + "\n", + " def preprocess_images(self, images):\n", + " vision_x = [self.vision_processor(im).unsqueeze(0) for im in images]\n", + " return torch.cat(vision_x, dim=0)\n", + "\n", + " def load_official_medflamingo(device=\"cuda\"):\n", + " model, image_processor, tokenizer = create_model_and_transforms(\n", + " clip_vision_encoder_path=\"ViT-L-14\",\n", + " clip_vision_encoder_pretrained=\"openai\",\n", + " lang_encoder_path=\"huggyllama/llama-7b\",\n", + " tokenizer_path=\"huggyllama/llama-7b\",\n", + " cross_attn_every_n_layers=4,\n", + " )\n", + " checkpoint_path = hf_hub_download(\n", + " repo_id=\"med-flamingo/med-flamingo\",\n", + " filename=\"model.pt\",\n", + " token=HF_TOKEN,\n", + " )\n", + " state_dict = torch.load(checkpoint_path, map_location=\"cpu\")\n", + " model.load_state_dict(state_dict, strict=False)\n", + " model = model.to(device)\n", + " model.eval()\n", + "\n", + " for generation_config in [\n", + " getattr(model, 'generation_config', None),\n", + " getattr(getattr(model, 'lang_encoder', None), 'generation_config', None),\n", + " ]:\n", + " if generation_config is None:\n", + " continue\n", + " for attr, value in [('top_k', None), ('top_p', None), ('temperature', 1.0), ('do_sample', False)]:\n", + " if hasattr(generation_config, attr):\n", + " try:\n", + " setattr(generation_config, attr, value)\n", + " except Exception:\n", + " pass\n", + "\n", + " processor = FlamingoProcessor(tokenizer, image_processor)\n", + " return model, processor\n", + "\n", + " def generate_vote_prediction(model, processor, row, shots, vote_idx: int):\n", + " prompt = build_few_shot_prompt(shots, row['question'], variant=vote_idx)\n", + " images = []\n", + " for example in shots + [row]:\n", + " with Image.open(example['image']) as image:\n", + " images.append(image.convert('RGB'))\n", + "\n", + " tokenized = processor.encode_text(prompt)\n", + " prompt_length = tokenized['input_ids'].shape[1]\n", + " pixels = processor.preprocess_images(images)\n", + " pixels = repeat(pixels, 'N c h w -> b N T c h w', b=1, T=1).to(DEVICE)\n", + "\n", + " with torch.no_grad():\n", + " generated = model.generate(\n", + " vision_x=pixels,\n", + " lang_x=tokenized['input_ids'].to(DEVICE),\n", + " attention_mask=tokenized['attention_mask'].to(DEVICE),\n", + " max_new_tokens=OFFICIAL_MAX_NEW_TOKENS,\n", + " )\n", + "\n", + " completion_tokens = generated[0][prompt_length:]\n", + " decoded = processor.tokenizer.decode(completion_tokens, skip_special_tokens=True)\n", + " pred = clean_generation(decoded)\n", + " pred_norm = normalize_answer(pred)\n", + " pred_label = normalize_yesno_prediction(pred) if SUBSET == 'yesno' else pred_norm\n", + " return {\n", + " 'raw_text': pred,\n", + " 'pred_norm': pred_norm,\n", + " 'pred_label': pred_label,\n", + " 'prompt': prompt,\n", + " }\n", + "\n", + " def choose_vote(vote_records):\n", + " valid_votes = [vote for vote in vote_records if vote['pred_label'] in {'yes', 'no'}]\n", + " if not valid_votes:\n", + " return vote_records[0], 0\n", + "\n", + " counts = Counter(vote['pred_label'] for vote in valid_votes)\n", + " max_count = max(counts.values())\n", + " winners = {label for label, count in counts.items() if count == max_count}\n", + " for vote in valid_votes:\n", + " if vote['pred_label'] in winners:\n", + " return vote, len(valid_votes)\n", + " return valid_votes[0], len(valid_votes)\n", + "\n", + " def evaluate_official_medflamingo(limit=100, k=6, seed=42):\n", + " split_local = split_fn(rows, seed)\n", + " train_local = subset_rows(rows, split_local['train'])\n", + " test_local = subset_rows(rows, split_local['test'])\n", + " if limit is not None:\n", + " test_local = test_local[:limit]\n", + "\n", + " model, processor = load_official_medflamingo(device=DEVICE)\n", + " predictions = []\n", + " y_true_yesno = []\n", + " y_pred_yesno = []\n", + " invalid_predictions = 0\n", + "\n", + " for row in test_local:\n", + " vote_records = []\n", + " for vote_idx in range(max(1, OFFICIAL_VOTE_PASSES)):\n", + " shots = sample_few_shot_examples(train_local, row, k=k, seed=seed, vote_idx=vote_idx)\n", + " vote = generate_vote_prediction(model, processor, row, shots, vote_idx=vote_idx)\n", + " vote['shot_answers'] = [example['answer'] for example in shots]\n", + " vote_records.append(vote)\n", + "\n", + " chosen_vote, valid_vote_count = choose_vote(vote_records)\n", + " true_norm = normalize_answer(row['answer'])\n", + " scored_pred = chosen_vote['pred_label'] if SUBSET == 'yesno' else chosen_vote['pred_norm']\n", + " strict_exact_match = int(chosen_vote['pred_norm'] == true_norm)\n", + " exact_match = int(scored_pred == true_norm)\n", + "\n", + " if SUBSET == 'yesno':\n", + " y_true_yesno.append(true_norm)\n", + " y_pred_yesno.append(scored_pred if scored_pred else 'invalid')\n", + " if scored_pred not in {'yes', 'no'}:\n", + " invalid_predictions += 1\n", + "\n", + " predictions.append({\n", + " 'image_name': row['image_name'],\n", + " 'question': row['question'],\n", + " 'answer_true': row['answer'],\n", + " 'answer_pred': chosen_vote['raw_text'],\n", + " 'answer_true_norm': true_norm,\n", + " 'answer_pred_norm': chosen_vote['pred_norm'],\n", + " 'answer_pred_scored': scored_pred,\n", + " 'strict_exact_match': strict_exact_match,\n", + " 'exact_match': exact_match,\n", + " 'valid_vote_count': valid_vote_count,\n", + " 'vote_labels': [vote['pred_label'] for vote in vote_records],\n", + " 'vote_outputs': [vote['raw_text'] for vote in vote_records],\n", + " })\n", + "\n", + " refs = [row['answer_true'] for row in predictions]\n", + " hyps = [row['answer_pred'] or 'invalid' for row in predictions]\n", + " _, _, bert_f1 = bert_score(hyps, refs, lang='en', verbose=False)\n", + "\n", + " result = {\n", + " 'n_test': len(predictions),\n", + " 'strict_exact_match': float(np.mean([row['strict_exact_match'] for row in predictions])),\n", + " 'exact_match': float(np.mean([row['exact_match'] for row in predictions])),\n", + " 'bertscore_f1': float(bert_f1.mean().item()),\n", + " 'predictions': predictions,\n", + " }\n", + "\n", + " if SUBSET == 'yesno':\n", + " labels = sorted(set(y_true_yesno) | set(y_pred_yesno))\n", + " result['yesno_accuracy'] = float(accuracy_score(y_true_yesno, y_pred_yesno))\n", + " result['yesno_f1_macro'] = float(f1_score(y_true_yesno, y_pred_yesno, average='macro', zero_division=0))\n", + " result['invalid_predictions'] = int(invalid_predictions)\n", + " result['invalid_rate'] = float(invalid_predictions / max(1, len(predictions)))\n", + " result['confusion_labels'] = labels\n", + " result['confusion_matrix'] = confusion_matrix(y_true_yesno, y_pred_yesno, labels=labels).tolist()\n", + "\n", + " del model\n", + " gc.collect()\n", + " if torch.cuda.is_available():\n", + " torch.cuda.empty_cache()\n", + " return result\n", + "else:\n", + " print('Set RUN_OFFICIAL_MEDFLAMINGO = True to enable the official checkpoint section.')\n" + ], + "id": "official-helpers" + }, + { + "cell_type": "code", + "execution_count": 14, + "metadata": { + "id": "official-run", + "colab": { + "base_uri": "https://localhost:8080/", + "height": 1000, + "referenced_widgets": [ + "a456930f8a104093a1e85f76f638ad37", + "d6a16d5d62414a819dae4a09b43186d8", + "2264ef816ca943769ffbe3786c9582ba", + "9d099eb9207f497eb96bf9ce85d102fb", + "b85b6b81f4b247cfabf527db219a0b01", + "5be5c96d632c4d249a62c0a436efe4ec", + "b699354d85b2493cb14d18d621abe873", + "b97badea86c6424a94f8b1ff3b386ae9", + "bb2d2934b5e94f94b563657cb8cf4b09", + "d522e146953944ebb3bbde5a6350cd5f", + "1b396ddd99d740f0962deff57c11edc6" + ] + }, + "outputId": "bb701ac5-1d36-429e-cc02-3d166825e62f" + }, + "outputs": [ + { + "output_type": "stream", + "name": "stderr", + "text": [ + "/usr/local/lib/python3.12/dist-packages/open_clip/factory.py:450: UserWarning: QuickGELU mismatch between final model config (quick_gelu=False) and pretrained tag 'openai' (quick_gelu=True).\n", + " warnings.warn(\n" + ] + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + "Loading checkpoint shards: 0%| | 0/2 [00:00\n", + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
image_namequestionanswer_trueanswer_predanswer_pred_scoredvalid_vote_countvote_labelsstrict_exact_matchexact_match
0synpic25821.jpgIs there evidence of any fractures of the ribs?nonono3[no, no, no]11
1synpic34515.jpgIs there evidence of small bowel obstruction o...yesnono3[no, no, no]00
2synpic55245.jpgIs the mass compressing the mid brain on this ...yesnono3[no, no, no]00
3synpic50962.jpgAre brain structures shifted across the midline?nonono3[no, no, no]11
4synpic50962.jpgis there a midline shift of the cerebral paren...nonono3[no, no, no]11
5synpic50962.jpgIs this a transverse section?yesnono3[no, yes, no]00
6synpic34515.jpgany observed degenerative changes in the verte...nonono3[no, no, no]11
7synpic39088.jpgIs there evidence of air fluid levels in the p...yesnono3[no, no, no]00
8synpic26925.jpgIs the vertebral artery/basilar artery located...yesnono3[no, no, no]00
9synpic34947.jpgis the plane of section transverse?yesyesyes3[yes, yes, yes]11
\n", + "
\n", + "
\n", + "\n", + "
\n", + " \n", + "\n", + " \n", + "\n", + " \n", + "
\n", + "\n", + "\n", + "
\n", + " \n" + ], + "application/vnd.google.colaboratory.intrinsic+json": { + "type": "dataframe", + "summary": "{\n \"name\": \" print('Official checkpoint path is disabled\",\n \"rows\": 10,\n \"fields\": [\n {\n \"column\": \"image_name\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 7,\n \"samples\": [\n \"synpic25821.jpg\",\n \"synpic34515.jpg\",\n \"synpic26925.jpg\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"question\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 10,\n \"samples\": [\n \"Is the vertebral artery/basilar artery located in this image?\",\n \"Is there evidence of small bowel obstruction on this image?\",\n \"Is this a transverse section?\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"answer_true\",\n \"properties\": {\n \"dtype\": \"category\",\n \"num_unique_values\": 2,\n \"samples\": [\n \"yes\",\n \"no\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"answer_pred\",\n \"properties\": {\n \"dtype\": \"category\",\n \"num_unique_values\": 2,\n \"samples\": [\n \"yes\",\n \"no\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"answer_pred_scored\",\n \"properties\": {\n \"dtype\": \"category\",\n \"num_unique_values\": 2,\n \"samples\": [\n \"yes\",\n \"no\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"valid_vote_count\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0,\n \"min\": 3,\n \"max\": 3,\n \"num_unique_values\": 1,\n \"samples\": [\n 3\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"vote_labels\",\n \"properties\": {\n \"dtype\": \"object\",\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"strict_exact_match\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0,\n \"min\": 0,\n \"max\": 1,\n \"num_unique_values\": 2,\n \"samples\": [\n 0\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"exact_match\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0,\n \"min\": 0,\n \"max\": 1,\n \"num_unique_values\": 2,\n \"samples\": [\n 0\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n }\n ]\n}" + } + }, + "metadata": {} + }, + { + "output_type": "display_data", + "data": { + "text/plain": [ + " experiment accuracy f1_macro \\\n", + "0 Best PyHealth sweep: clip-base32 + distilgpt2 ... 0.641148 0.637128 \n", + "1 Question-only baseline 0.602871 0.602871 \n", + "2 Official Med-Flamingo (3-vote yes/no) 0.598086 0.513846 \n", + "3 PyHealth MedFlamingo classifier 0.459330 0.314754 \n", + "4 Majority baseline 0.459330 0.314754 \n", + "\n", + " notes \n", + "0 best single sweep configuration \n", + "1 tuned on validation split \n", + "2 strict_em=0.598, invalid_rate=0.000 \n", + "3 single classifier run \n", + "4 train-set majority answer " + ], + "text/html": [ + "\n", + "
\n", + "
\n", + "\n", + "\n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + " \n", + "
experimentaccuracyf1_macronotes
0Best PyHealth sweep: clip-base32 + distilgpt2 ...0.6411480.637128best single sweep configuration
1Question-only baseline0.6028710.602871tuned on validation split
2Official Med-Flamingo (3-vote yes/no)0.5980860.513846strict_em=0.598, invalid_rate=0.000
3PyHealth MedFlamingo classifier0.4593300.314754single classifier run
4Majority baseline0.4593300.314754train-set majority answer
\n", + "
\n", + "
\n", + "\n", + "
\n", + " \n", + "\n", + " \n", + "\n", + " \n", + "
\n", + "\n", + "\n", + "
\n", + "
\n" + ], + "application/vnd.google.colaboratory.intrinsic+json": { + "type": "dataframe", + "summary": "{\n \"name\": \" print('Official checkpoint path is disabled\",\n \"rows\": 5,\n \"fields\": [\n {\n \"column\": \"experiment\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 5,\n \"samples\": [\n \"Question-only baseline\",\n \"Majority baseline\",\n \"Official Med-Flamingo (3-vote yes/no)\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"accuracy\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.08636330501334731,\n \"min\": 0.45933014354066987,\n \"max\": 0.6411483253588517,\n \"num_unique_values\": 4,\n \"samples\": [\n 0.6028708133971292,\n 0.45933014354066987,\n 0.6411483253588517\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"f1_macro\",\n \"properties\": {\n \"dtype\": \"number\",\n \"std\": 0.15450634227893512,\n \"min\": 0.31475409836065577,\n \"max\": 0.6371275783040489,\n \"num_unique_values\": 4,\n \"samples\": [\n 0.6028708133971292,\n 0.31475409836065577,\n 0.6371275783040489\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n },\n {\n \"column\": \"notes\",\n \"properties\": {\n \"dtype\": \"string\",\n \"num_unique_values\": 5,\n \"samples\": [\n \"tuned on validation split\",\n \"train-set majority answer\",\n \"strict_em=0.598, invalid_rate=0.000\"\n ],\n \"semantic_type\": \"\",\n \"description\": \"\"\n }\n }\n ]\n}" + } + }, + "metadata": {} + }, + { + "output_type": "stream", + "name": "stdout", + "text": [ + "saved to /content/PyHealth/output/colab_eval/official_medflamingo_yesno_250.json\n" + ] + } + ], + "source": [ + "if RUN_OFFICIAL_MEDFLAMINGO:\n", + " official_results = evaluate_official_medflamingo(\n", + " limit=OFFICIAL_EVAL_LIMIT,\n", + " k=FEW_SHOT_K,\n", + " seed=SEED,\n", + " )\n", + "\n", + " RESULTS_DIR.mkdir(parents=True, exist_ok=True)\n", + " official_json = RESULTS_DIR / f\"official_medflamingo_{SUBSET}_{OFFICIAL_EVAL_LIMIT}.json\"\n", + " official_json.write_text(json.dumps(official_results, indent=2))\n", + "\n", + " summary_payload = {k: v for k, v in official_results.items() if k != 'predictions'}\n", + " print(json.dumps(summary_payload, indent=2))\n", + "\n", + " preview_df = pd.DataFrame(official_results['predictions'])\n", + " preview_cols = [\n", + " 'image_name',\n", + " 'question',\n", + " 'answer_true',\n", + " 'answer_pred',\n", + " 'answer_pred_scored',\n", + " 'valid_vote_count',\n", + " 'vote_labels',\n", + " 'strict_exact_match',\n", + " 'exact_match',\n", + " ]\n", + " preview_cols = [col for col in preview_cols if col in preview_df.columns]\n", + " display(preview_df[preview_cols].head(10))\n", + "\n", + " leaderboard_rows = []\n", + " if 'results' in globals():\n", + " leaderboard_rows.extend([\n", + " {\n", + " 'experiment': 'Question-only baseline',\n", + " 'accuracy': results['baselines']['question_only_tfidf_logreg']['accuracy'],\n", + " 'f1_macro': results['baselines']['question_only_tfidf_logreg']['f1_macro'],\n", + " 'notes': 'tuned on validation split',\n", + " },\n", + " {\n", + " 'experiment': 'PyHealth MedFlamingo classifier',\n", + " 'accuracy': results['medflamingo_classifier']['metrics']['accuracy'],\n", + " 'f1_macro': results['medflamingo_classifier']['metrics']['f1_macro'],\n", + " 'notes': 'single classifier run',\n", + " },\n", + " {\n", + " 'experiment': 'Majority baseline',\n", + " 'accuracy': results['baselines']['majority']['accuracy'],\n", + " 'f1_macro': results['baselines']['majority']['f1_macro'],\n", + " 'notes': 'train-set majority answer',\n", + " },\n", + " ])\n", + "\n", + " if 'best_pyhealth_sweep' in globals() and best_pyhealth_sweep is not None:\n", + " leaderboard_rows.append({\n", + " 'experiment': f\"Best PyHealth sweep: {best_pyhealth_sweep['name']}\",\n", + " 'accuracy': best_pyhealth_sweep.get('accuracy'),\n", + " 'f1_macro': best_pyhealth_sweep.get('f1_macro'),\n", + " 'notes': 'best single sweep configuration',\n", + " })\n", + "\n", + " leaderboard_rows.append({\n", + " 'experiment': f'Official Med-Flamingo ({OFFICIAL_VOTE_PASSES}-vote yes/no)',\n", + " 'accuracy': official_results.get('yesno_accuracy', official_results.get('exact_match')),\n", + " 'f1_macro': official_results.get('yesno_f1_macro', float('nan')),\n", + " 'notes': f\"strict_em={official_results.get('strict_exact_match', float('nan')):.3f}, invalid_rate={official_results.get('invalid_rate', 0.0):.3f}\",\n", + " })\n", + "\n", + " leaderboard_df = pd.DataFrame(leaderboard_rows)\n", + " display(leaderboard_df.sort_values(['accuracy', 'f1_macro'], ascending=False, na_position='last').reset_index(drop=True))\n", + " print('saved to', official_json)\n", + "else:\n", + " print('Official checkpoint path is disabled. Flip RUN_OFFICIAL_MEDFLAMINGO = True when you have a high-memory Colab GPU.')\n" + ], + "id": "official-run" + } + ], + "metadata": { + "colab": { + "provenance": [], + "machine_shape": "hm", + "gpuType": "A100" + }, + "kernelspec": { + "display_name": "Python 3", + "name": "python3" + }, + "language_info": { + "name": "python" + }, + "accelerator": "GPU", + "widgets": { + "application/vnd.jupyter.widget-state+json": { + "acd7b5ff0a9f4d63bf388a6b2cebf7a2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_6ce4ca3ff29845708d4e160e2d85fdc1", + "IPY_MODEL_e7cc05e116b14a8295aa2d4a6a3e1f71", + "IPY_MODEL_51846529e85e4328861363b04f407960" + ], + "layout": "IPY_MODEL_84a9e53caa524c13bfdb20a1c3186df9" + } + }, + "6ce4ca3ff29845708d4e160e2d85fdc1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_604a6b8efbb74826bd7d45be941838bb", + "placeholder": "​", + "style": "IPY_MODEL_f42d1d7b51884b7bbb4e5104b8798d7a", + "value": "config.json: " + } + }, + "e7cc05e116b14a8295aa2d4a6a3e1f71": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_6bdc1e828cfc482aab413da3a8d7dfb2", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ab1eb004b8714a25b1edfb6f76f3faa4", + "value": 1 + } + }, + "51846529e85e4328861363b04f407960": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_e958a0ce91d7432cb0dc0e89d9a70b89", + "placeholder": "​", + "style": "IPY_MODEL_d176747e04594af089dc09a824a9a43e", + "value": " 4.19k/? [00:00<00:00, 423kB/s]" + } + }, + "84a9e53caa524c13bfdb20a1c3186df9": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "604a6b8efbb74826bd7d45be941838bb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f42d1d7b51884b7bbb4e5104b8798d7a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6bdc1e828cfc482aab413da3a8d7dfb2": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "ab1eb004b8714a25b1edfb6f76f3faa4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "e958a0ce91d7432cb0dc0e89d9a70b89": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "d176747e04594af089dc09a824a9a43e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "33a890b9ac7a44b69e5abdf5551db128": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_ed45b1fda48345ce8c9608f763a48aa5", + "IPY_MODEL_2cc26f3bf8e0447eaacf48ab265f1389", + "IPY_MODEL_e17f0eac5ea24bc586bb58fca8abd2ad" + ], + "layout": "IPY_MODEL_14ef024857914b20bbd6df5a31210300" + } + }, + "ed45b1fda48345ce8c9608f763a48aa5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_72099a28efa445baa050627a973cb7b8", + "placeholder": "​", + "style": "IPY_MODEL_535461538b114038ab19a1c17809c1be", + "value": "pytorch_model.bin: 100%" + } + }, + "2cc26f3bf8e0447eaacf48ab265f1389": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7e3f94ef50a040b68ef7bff8792db3e9", + "max": 605247071, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_35a841d9e83643ed95e63b52b4e15c0b", + "value": 605247071 + } + }, + "e17f0eac5ea24bc586bb58fca8abd2ad": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_15d2c16bc4b847aba7c776beaffdb905", + "placeholder": "​", + "style": "IPY_MODEL_70c468a5e51f44ecad363d3f85699ea5", + "value": " 605M/605M [00:03<00:00, 381MB/s]" + } + }, + "14ef024857914b20bbd6df5a31210300": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "72099a28efa445baa050627a973cb7b8": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "535461538b114038ab19a1c17809c1be": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "7e3f94ef50a040b68ef7bff8792db3e9": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "35a841d9e83643ed95e63b52b4e15c0b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "15d2c16bc4b847aba7c776beaffdb905": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "70c468a5e51f44ecad363d3f85699ea5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3d48ab5e157041ae964922f5b8eb72df": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_e255481141fc46689caadd7fcd205e29", + "IPY_MODEL_5a0cabdcdee348c295ea35b28a8fa5d4", + "IPY_MODEL_adce6d44c61f4371841466da8baf6712" + ], + "layout": "IPY_MODEL_264141dd7200401b893a2d829db47b14" + } + }, + "e255481141fc46689caadd7fcd205e29": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_be9a8d0b703543b3bdebe4309e7ce82b", + "placeholder": "​", + "style": "IPY_MODEL_4557092565324dfa902e8ff4dbc26c68", + "value": "config.json: 100%" + } + }, + "5a0cabdcdee348c295ea35b28a8fa5d4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_d40c4075d42945aabbb1a55758fc0fcb", + "max": 651, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ff025ea9727441e589dd2957ffeb280e", + "value": 651 + } + }, + "adce6d44c61f4371841466da8baf6712": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_0d2f7497712646d9a34f74c1b21eec6d", + "placeholder": "​", + "style": "IPY_MODEL_2bbffd286d814603a3eb1f31d8bca9f3", + "value": " 651/651 [00:00<00:00, 80.6kB/s]" + } + }, + "264141dd7200401b893a2d829db47b14": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "be9a8d0b703543b3bdebe4309e7ce82b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "4557092565324dfa902e8ff4dbc26c68": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "d40c4075d42945aabbb1a55758fc0fcb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ff025ea9727441e589dd2957ffeb280e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "0d2f7497712646d9a34f74c1b21eec6d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2bbffd286d814603a3eb1f31d8bca9f3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6df7f59495984977839277d0cc8d189a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_9cadcb737429486fa6a89b8885af0985", + "IPY_MODEL_4a61a2ae454e4fdb8f64a95f86645891", + "IPY_MODEL_07d0a0d199cb4e0e91fb5a7c54cea000" + ], + "layout": "IPY_MODEL_007aa53b17aa4ca09b1cefa9230d2d15" + } + }, + "9cadcb737429486fa6a89b8885af0985": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f28787e3e77c4f4e8b2094a367d423d8", + "placeholder": "​", + "style": "IPY_MODEL_c1a31e283c9c4ada8bd2854ca9a91357", + "value": "model.safetensors: 100%" + } + }, + "4a61a2ae454e4fdb8f64a95f86645891": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7812a101bd714c6fa6aadba6ceb5aff8", + "max": 605157884, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_cc537b85722c4477a8df63c3e6e7caaf", + "value": 605157884 + } + }, + "07d0a0d199cb4e0e91fb5a7c54cea000": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_cb1a6b1e367c4f74abea9fd73f2f6102", + "placeholder": "​", + "style": "IPY_MODEL_b56c07b88dd74ee9a6a93e91bf62ff65", + "value": " 605M/605M [00:01<00:00, 512MB/s]" + } + }, + "007aa53b17aa4ca09b1cefa9230d2d15": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f28787e3e77c4f4e8b2094a367d423d8": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c1a31e283c9c4ada8bd2854ca9a91357": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "7812a101bd714c6fa6aadba6ceb5aff8": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "cc537b85722c4477a8df63c3e6e7caaf": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "cb1a6b1e367c4f74abea9fd73f2f6102": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b56c07b88dd74ee9a6a93e91bf62ff65": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6c2c78aa71a64f7697b6773553d2be58": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_ad71a53f38a741f8940205b98999f809", + "IPY_MODEL_111dd33db66c4a249fc5a4d04b07629b", + "IPY_MODEL_795539cbc9a24914832bf895a2aac7c9" + ], + "layout": "IPY_MODEL_4e74fd0dd3fd47258aa7cc1e38e2544d" + } + }, + "ad71a53f38a741f8940205b98999f809": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_ac11b99982db41cf92cc62eca6a263f0", + "placeholder": "​", + "style": "IPY_MODEL_57290ee4890b4331ae2660b097fd4991", + "value": "pytorch_model.bin: 100%" + } + }, + "111dd33db66c4a249fc5a4d04b07629b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_e22bf4d7e1a94284a6a8941286ef9bbe", + "max": 250540281, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_a0bbd056beb840b9805dfdc35a2192cc", + "value": 250540281 + } + }, + "795539cbc9a24914832bf895a2aac7c9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5180a004be02425ebc69e844e6b98f59", + "placeholder": "​", + "style": "IPY_MODEL_aa2779d0654b466b8ce27f1b4db7d57d", + "value": " 251M/251M [00:01<00:00, 449MB/s]" + } + }, + "4e74fd0dd3fd47258aa7cc1e38e2544d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ac11b99982db41cf92cc62eca6a263f0": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "57290ee4890b4331ae2660b097fd4991": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "e22bf4d7e1a94284a6a8941286ef9bbe": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a0bbd056beb840b9805dfdc35a2192cc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "5180a004be02425ebc69e844e6b98f59": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "aa2779d0654b466b8ce27f1b4db7d57d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "df31724f148447ff81138992c0e814a8": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_448f69945f6846728ec99f3151c11d5f", + "IPY_MODEL_bac56295e5404e2e8f4130dd43b8dc87", + "IPY_MODEL_f88e40b41867487fb4f5ae3d2567d18b" + ], + "layout": "IPY_MODEL_a9ffa517041f45d7b9645b5778aff6b6" + } + }, + "448f69945f6846728ec99f3151c11d5f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_737a19e36d7849b3a6d54273058bb24f", + "placeholder": "​", + "style": "IPY_MODEL_a9695a8d5d794c17a530444cb9c3f8c7", + "value": "generation_config.json: 100%" + } + }, + "bac56295e5404e2e8f4130dd43b8dc87": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_9a327b232eab43cb89aa6d052a81a484", + "max": 137, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_9f3dd0a3b96c4aff9cdce6897f79e99d", + "value": 137 + } + }, + "f88e40b41867487fb4f5ae3d2567d18b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f1abec28112c43adabc687cc29369161", + "placeholder": "​", + "style": "IPY_MODEL_21ab6280076f46b6bfc9cf6acfde6cf6", + "value": " 137/137 [00:00<00:00, 17.4kB/s]" + } + }, + "a9ffa517041f45d7b9645b5778aff6b6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "737a19e36d7849b3a6d54273058bb24f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a9695a8d5d794c17a530444cb9c3f8c7": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "9a327b232eab43cb89aa6d052a81a484": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9f3dd0a3b96c4aff9cdce6897f79e99d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "f1abec28112c43adabc687cc29369161": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "21ab6280076f46b6bfc9cf6acfde6cf6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "de81098896474289a6d1d02fe347bef5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_b36e1c7acb9d419baa38c1b181fde85e", + "IPY_MODEL_bc63f48c3d844f8c92112fd37918604e", + "IPY_MODEL_0e2fcd45270e4a858f4762e3bd550f1b" + ], + "layout": "IPY_MODEL_0fb6ae683fd1433cb6546f95b2fd2a04" + } + }, + "b36e1c7acb9d419baa38c1b181fde85e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1a4f12b78d9b42cba0c1cf174dbc595b", + "placeholder": "​", + "style": "IPY_MODEL_8ab706519d06433ebf8bab25f9390089", + "value": "tokenizer_config.json: 100%" + } + }, + "bc63f48c3d844f8c92112fd37918604e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_cdc86b05cf9b4919b0cc4947bc315dd3", + "max": 685, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_efc6c70f89054c5ea2f8b9a6d1136e06", + "value": 685 + } + }, + "0e2fcd45270e4a858f4762e3bd550f1b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a605993ade3d4b5a8ad84d8f4a57f85b", + "placeholder": "​", + "style": "IPY_MODEL_9019fa49eb2a4f78a3921dbb7b6e8eea", + "value": " 685/685 [00:00<00:00, 93.8kB/s]" + } + }, + "0fb6ae683fd1433cb6546f95b2fd2a04": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "1a4f12b78d9b42cba0c1cf174dbc595b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "8ab706519d06433ebf8bab25f9390089": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "cdc86b05cf9b4919b0cc4947bc315dd3": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "efc6c70f89054c5ea2f8b9a6d1136e06": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "a605993ade3d4b5a8ad84d8f4a57f85b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9019fa49eb2a4f78a3921dbb7b6e8eea": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "40ad37b58fbd4d4aab2b9a2e315007c5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_f173a0ce82c24009aa805aa7fd3fc921", + "IPY_MODEL_25c658dba47749a3a4b801b54dbf3548", + "IPY_MODEL_b92bc1018872457a9958937aa89e4fcb" + ], + "layout": "IPY_MODEL_3a0f366445b84d988e18c50d864e495b" + } + }, + "f173a0ce82c24009aa805aa7fd3fc921": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a8bd363d80eb423891ee5e5e0f7dd864", + "placeholder": "​", + "style": "IPY_MODEL_e12cf301b1d24037b3e09ba5339c844f", + "value": "vocab.json: " + } + }, + "25c658dba47749a3a4b801b54dbf3548": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_510ea785201540fdafc7d432e028f4ea", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_9142ed175d47474b98bfd7bb9130d182", + "value": 1 + } + }, + "b92bc1018872457a9958937aa89e4fcb": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_330a88de89704bd39052c9f2bbf60145", + "placeholder": "​", + "style": "IPY_MODEL_a1aaedeba6424a25b350cb841afc1afe", + "value": " 899k/? [00:00<00:00, 49.5MB/s]" + } + }, + "3a0f366445b84d988e18c50d864e495b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a8bd363d80eb423891ee5e5e0f7dd864": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "e12cf301b1d24037b3e09ba5339c844f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "510ea785201540fdafc7d432e028f4ea": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "9142ed175d47474b98bfd7bb9130d182": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "330a88de89704bd39052c9f2bbf60145": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a1aaedeba6424a25b350cb841afc1afe": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "5c952ff026394314b7e00dc82a0f9ace": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_05b9bf390e844bc09cc85fac66bee422", + "IPY_MODEL_185f78f0257541d2b415b347426e9142", + "IPY_MODEL_7c49107925b84bda99ef04c0efedb186" + ], + "layout": "IPY_MODEL_5329cb3d6bf04370ae2a0c3bf560895f" + } + }, + "05b9bf390e844bc09cc85fac66bee422": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_cf8ab0c005bf41c581c0a9613070964d", + "placeholder": "​", + "style": "IPY_MODEL_2460c4d003784bcabc36ab5e7a5f0669", + "value": "model.safetensors: 100%" + } + }, + "185f78f0257541d2b415b347426e9142": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_9ac18602e4df4c078ebdce876e42e56d", + "max": 250501000, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_67a31b05d10d4b6c98e45f5697eddd19", + "value": 250501000 + } + }, + "7c49107925b84bda99ef04c0efedb186": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_8101ae386c89401d9b75a66a3de58e56", + "placeholder": "​", + "style": "IPY_MODEL_1e1c77c6cea74aeb8d7f45cbf29f3683", + "value": " 251M/251M [00:01<00:00, 1.25GB/s]" + } + }, + "5329cb3d6bf04370ae2a0c3bf560895f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "cf8ab0c005bf41c581c0a9613070964d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2460c4d003784bcabc36ab5e7a5f0669": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "9ac18602e4df4c078ebdce876e42e56d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "67a31b05d10d4b6c98e45f5697eddd19": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "8101ae386c89401d9b75a66a3de58e56": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "1e1c77c6cea74aeb8d7f45cbf29f3683": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "e102b122537443f891706bd41de97b4b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_8e96694b501c4f7a8464b47acf253b8e", + "IPY_MODEL_cd81cd4d272e415f81561d68be6b162f", + "IPY_MODEL_ff42f063011946fe9d2b6e4522fecfc4" + ], + "layout": "IPY_MODEL_2690b31acc4b4d42b7b4862850cc3e88" + } + }, + "8e96694b501c4f7a8464b47acf253b8e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_06bd1e8d1d5642749398e18c54587815", + "placeholder": "​", + "style": "IPY_MODEL_b719b5dea4624ae883b127b94696fa85", + "value": "merges.txt: " + } + }, + "cd81cd4d272e415f81561d68be6b162f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_82855b27296c4c03abde7160e0865a27", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_0b12db7d9ac6442999e3d890f2b7d5a8", + "value": 1 + } + }, + "ff42f063011946fe9d2b6e4522fecfc4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_cd0a9a11b1c4448580648b3f6ea4dab3", + "placeholder": "​", + "style": "IPY_MODEL_73bb2fee355b49929698f7d5c996c827", + "value": " 456k/? [00:00<00:00, 30.7MB/s]" + } + }, + "2690b31acc4b4d42b7b4862850cc3e88": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "06bd1e8d1d5642749398e18c54587815": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b719b5dea4624ae883b127b94696fa85": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "82855b27296c4c03abde7160e0865a27": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "0b12db7d9ac6442999e3d890f2b7d5a8": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "cd0a9a11b1c4448580648b3f6ea4dab3": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "73bb2fee355b49929698f7d5c996c827": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "b0ba18baafe64e97b89862ed539a60c6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_d3d4fc244fbe468bad7f6a95bc4675b1", + "IPY_MODEL_4e4ea5e920be49a6a632b23546dd1fa6", + "IPY_MODEL_69ced8f4c9974550bd52f4a24652cf7e" + ], + "layout": "IPY_MODEL_eedc6168a02842b9a83e0cdd460681dd" + } + }, + "d3d4fc244fbe468bad7f6a95bc4675b1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_075e02f3c4b246179b546623b536db87", + "placeholder": "​", + "style": "IPY_MODEL_848f2f530934403eb7a28eeaec02ab26", + "value": "special_tokens_map.json: 100%" + } + }, + "4e4ea5e920be49a6a632b23546dd1fa6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_e09a891456dc4055bcf284b1e8851534", + "max": 441, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_c4bf481975ee42048291acadc69d6e92", + "value": 441 + } + }, + "69ced8f4c9974550bd52f4a24652cf7e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7ad717da24b84648ae69628175bfb97a", + "placeholder": "​", + "style": "IPY_MODEL_0b72a826af984585ba0f940c723cd2a2", + "value": " 441/441 [00:00<00:00, 57.6kB/s]" + } + }, + "eedc6168a02842b9a83e0cdd460681dd": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "075e02f3c4b246179b546623b536db87": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "848f2f530934403eb7a28eeaec02ab26": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "e09a891456dc4055bcf284b1e8851534": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c4bf481975ee42048291acadc69d6e92": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "7ad717da24b84648ae69628175bfb97a": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0b72a826af984585ba0f940c723cd2a2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "4e5d106e8df04c04afb93e2e05f1e5a1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_62e662eb82a544eba23f8578efce062c", + "IPY_MODEL_08395f2628dc47fdb78b587af045a3f2", + "IPY_MODEL_b7c2c2fd01974ed7aa8a2eca8ef5eca2" + ], + "layout": "IPY_MODEL_c51a2a4822a0407d85b0a020d19daafd" + } + }, + "62e662eb82a544eba23f8578efce062c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_2271ce8b6041454c8678dae859015a88", + "placeholder": "​", + "style": "IPY_MODEL_9eb6b05ccf9b41739b50f349e36ba236", + "value": "Epoch 0 / 6: 100%" + } + }, + "08395f2628dc47fdb78b587af045a3f2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_21e18797a53b4cc38a1b1d84fa2e2a7e", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ef374580a3894b72beaac1da7303ad74", + "value": 111 + } + }, + "b7c2c2fd01974ed7aa8a2eca8ef5eca2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_57e07801abda45538ab02ab77fd629e2", + "placeholder": "​", + "style": "IPY_MODEL_0b906d06ba884f39bfc620c184262efa", + "value": " 111/111 [00:10<00:00, 10.43it/s]" + } + }, + "c51a2a4822a0407d85b0a020d19daafd": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2271ce8b6041454c8678dae859015a88": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9eb6b05ccf9b41739b50f349e36ba236": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "21e18797a53b4cc38a1b1d84fa2e2a7e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ef374580a3894b72beaac1da7303ad74": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "57e07801abda45538ab02ab77fd629e2": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0b906d06ba884f39bfc620c184262efa": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "0eeb71251ae74f7798e40fc402ec3427": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_b1850f5ba46747a6900488e407a8df81", + "IPY_MODEL_bdafccd294e8413fb95244e0709b5842", + "IPY_MODEL_852a157855904586b528183eec635bd9" + ], + "layout": "IPY_MODEL_bbefe3ed5e3b4097a7a266fcc606268f" + } + }, + "b1850f5ba46747a6900488e407a8df81": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_b37c43fec7a44e0db34d84d0a496c688", + "placeholder": "​", + "style": "IPY_MODEL_3c44188c263e45859f046dd07432d62b", + "value": "Epoch 1 / 6: 100%" + } + }, + "bdafccd294e8413fb95244e0709b5842": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_588ead88ef904af6abebd242bd800216", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_9e324017db60470880d8dc46f9bdda3b", + "value": 111 + } + }, + "852a157855904586b528183eec635bd9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_afbf2f67e75149e39d8a5a8ddefd48d0", + "placeholder": "​", + "style": "IPY_MODEL_b95bf85d7d2f4961855001d74c766d28", + "value": " 111/111 [00:10<00:00, 10.67it/s]" + } + }, + "bbefe3ed5e3b4097a7a266fcc606268f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b37c43fec7a44e0db34d84d0a496c688": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "3c44188c263e45859f046dd07432d62b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "588ead88ef904af6abebd242bd800216": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9e324017db60470880d8dc46f9bdda3b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "afbf2f67e75149e39d8a5a8ddefd48d0": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b95bf85d7d2f4961855001d74c766d28": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "8c13701c53964a22877473434a8e8844": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_771a173d67c64675a4bde9b588437eeb", + "IPY_MODEL_b21efd6076d54af1a9befdd6577f1a4b", + "IPY_MODEL_c487e8a64c12441393671b85b0ff060a" + ], + "layout": "IPY_MODEL_a5f1f050588c4fedae9e8ecebc60e7f4" + } + }, + "771a173d67c64675a4bde9b588437eeb": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_33bfbc7cdd1b4416a1d9fde8a1f9a1d0", + "placeholder": "​", + "style": "IPY_MODEL_2fcedae3d8364a29b72b26f0b03fc4fe", + "value": "Epoch 2 / 6: 100%" + } + }, + "b21efd6076d54af1a9befdd6577f1a4b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a0b5e642d64c436b9eea9e6a484a4681", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_f51216bd565748edb00b4c57b03df263", + "value": 111 + } + }, + "c487e8a64c12441393671b85b0ff060a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_ecde02d5df744d778d02289194a2863b", + "placeholder": "​", + "style": "IPY_MODEL_9afc1eb6281942d28ccfd1b63fec2c30", + "value": " 111/111 [00:10<00:00, 10.68it/s]" + } + }, + "a5f1f050588c4fedae9e8ecebc60e7f4": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "33bfbc7cdd1b4416a1d9fde8a1f9a1d0": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2fcedae3d8364a29b72b26f0b03fc4fe": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "a0b5e642d64c436b9eea9e6a484a4681": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f51216bd565748edb00b4c57b03df263": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "ecde02d5df744d778d02289194a2863b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9afc1eb6281942d28ccfd1b63fec2c30": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "80841cd9a1254ed0921488602f6bc8ac": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_c688eb8ecea244d8a245e6e3be7683af", + "IPY_MODEL_566507e388a848a38b1785242c3b9b2b", + "IPY_MODEL_b01057a1da144b3681dbfafec8da839d" + ], + "layout": "IPY_MODEL_8efe4de11d9c459cb81c21c45a79ca0a" + } + }, + "c688eb8ecea244d8a245e6e3be7683af": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_63b658ed08fc4d54940f4b7291b521d6", + "placeholder": "​", + "style": "IPY_MODEL_c47f0e121e8b4b6fad6f858b289cffc1", + "value": "Epoch 0 / 8: 100%" + } + }, + "566507e388a848a38b1785242c3b9b2b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1d5097f6a7604a4da336025434cd1c62", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_11959cbae10d4c89818789ac4c50a4bf", + "value": 111 + } + }, + "b01057a1da144b3681dbfafec8da839d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_d26b7c8e8e474342af8b1089a724963e", + "placeholder": "​", + "style": "IPY_MODEL_546b7e7a9dfb40799926f4119073c79a", + "value": " 111/111 [00:10<00:00, 10.53it/s]" + } + }, + "8efe4de11d9c459cb81c21c45a79ca0a": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "63b658ed08fc4d54940f4b7291b521d6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c47f0e121e8b4b6fad6f858b289cffc1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1d5097f6a7604a4da336025434cd1c62": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "11959cbae10d4c89818789ac4c50a4bf": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "d26b7c8e8e474342af8b1089a724963e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "546b7e7a9dfb40799926f4119073c79a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "ef2a8f30246043ae980116b11c539b1d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_91fbed90a14f4b9687a156076b390a9e", + "IPY_MODEL_ada4773fe1bd4b5cbe53bbc2c5c27132", + "IPY_MODEL_03f9004190e444e3a21823a37d92a6ff" + ], + "layout": "IPY_MODEL_7b0aa6b967a442348757dbf9dd6f83ad" + } + }, + "91fbed90a14f4b9687a156076b390a9e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7a65709f790e47ff81a4c8b63922d640", + "placeholder": "​", + "style": "IPY_MODEL_6a66a27f072649a6845ab3abd90840e1", + "value": "Epoch 1 / 8: 100%" + } + }, + "ada4773fe1bd4b5cbe53bbc2c5c27132": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_e02393dd2e1047b2a0ab9e83704fdacf", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_da6714b4927a4b7294a1e442352a7e3a", + "value": 111 + } + }, + "03f9004190e444e3a21823a37d92a6ff": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_83797fbe29e0458199c77adaa0eabf0b", + "placeholder": "​", + "style": "IPY_MODEL_69304c74bd8148a8ae99dab82089196c", + "value": " 111/111 [00:10<00:00, 10.47it/s]" + } + }, + "7b0aa6b967a442348757dbf9dd6f83ad": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7a65709f790e47ff81a4c8b63922d640": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6a66a27f072649a6845ab3abd90840e1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "e02393dd2e1047b2a0ab9e83704fdacf": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "da6714b4927a4b7294a1e442352a7e3a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "83797fbe29e0458199c77adaa0eabf0b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "69304c74bd8148a8ae99dab82089196c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3cfc7bbf94ab40a892f58f08043bcabe": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_72ce6cdb5c8f4e72a9c03902294a316e", + "IPY_MODEL_57194c49e2ef496abf2379890d123e48", + "IPY_MODEL_e7e94a76ba09422782db85cd5926d51e" + ], + "layout": "IPY_MODEL_8d17b811e4fb4ce9b5b0f955b405212f" + } + }, + "72ce6cdb5c8f4e72a9c03902294a316e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_021efbd1c85c443abebc06c2d821d456", + "placeholder": "​", + "style": "IPY_MODEL_7400bd7f2d954f7eaff2c5db059b91fa", + "value": "Epoch 2 / 8: 100%" + } + }, + "57194c49e2ef496abf2379890d123e48": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7098375b0e984f0d870efc564d47cca6", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_d204c069e2ca45b38e73d20cb47633a9", + "value": 111 + } + }, + "e7e94a76ba09422782db85cd5926d51e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_ea0ce3580cca4e94b494bcaf78bfffc4", + "placeholder": "​", + "style": "IPY_MODEL_c438ac99da5b4812ae6f533172d9cb2b", + "value": " 111/111 [00:10<00:00, 10.40it/s]" + } + }, + "8d17b811e4fb4ce9b5b0f955b405212f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "021efbd1c85c443abebc06c2d821d456": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7400bd7f2d954f7eaff2c5db059b91fa": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "7098375b0e984f0d870efc564d47cca6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "d204c069e2ca45b38e73d20cb47633a9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "ea0ce3580cca4e94b494bcaf78bfffc4": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c438ac99da5b4812ae6f533172d9cb2b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "0a9df723378742c9b3ee6692e7c4011d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_7596b808b94348eb821eb59952a8b626", + "IPY_MODEL_f0309f365e8845bb92e199e036ffdd6c", + "IPY_MODEL_1ef630ff54034d7fbfc3a00f7804390b" + ], + "layout": "IPY_MODEL_50203f5f18c649b3a7a16527da71beeb" + } + }, + "7596b808b94348eb821eb59952a8b626": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_33483e6d078b4038bef307c8124c7e06", + "placeholder": "​", + "style": "IPY_MODEL_f379fcc06939403fa7584ad92b2fc225", + "value": "Epoch 3 / 8: 100%" + } + }, + "f0309f365e8845bb92e199e036ffdd6c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1b0602499e9b44b4be89e0cdcb14d4b5", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_a8ae55b1ee0b4220afce303cd998594f", + "value": 111 + } + }, + "1ef630ff54034d7fbfc3a00f7804390b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_aab89be66e5d46039d1821b3b12404d5", + "placeholder": "​", + "style": "IPY_MODEL_2b067b12b02547ccbc379492b409c0f0", + "value": " 111/111 [00:10<00:00, 10.36it/s]" + } + }, + "50203f5f18c649b3a7a16527da71beeb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "33483e6d078b4038bef307c8124c7e06": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f379fcc06939403fa7584ad92b2fc225": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1b0602499e9b44b4be89e0cdcb14d4b5": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a8ae55b1ee0b4220afce303cd998594f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "aab89be66e5d46039d1821b3b12404d5": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2b067b12b02547ccbc379492b409c0f0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "75afd19177b2493084219a30b1ea80e9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_a2d1baee530344d49ec19081e91f113a", + "IPY_MODEL_ae8cb23915d14a1180d825e502084bd0", + "IPY_MODEL_011a2e3962ad423cb2db2ecdab5a88ce" + ], + "layout": "IPY_MODEL_6005f7f4716344d3af218ce48c5f0a75" + } + }, + "a2d1baee530344d49ec19081e91f113a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_d0a86be8167a4ef0a7b7d0992489a1f3", + "placeholder": "​", + "style": "IPY_MODEL_ca5262b4ab044828afb48613b837352a", + "value": "Epoch 4 / 8: 100%" + } + }, + "ae8cb23915d14a1180d825e502084bd0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7e6d57fc41074dd19e0c2cabf17734ff", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_0b8c1d90464740ee874c8bd230734920", + "value": 111 + } + }, + "011a2e3962ad423cb2db2ecdab5a88ce": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a0465d556fc44a8cbe2386ab9e730ac8", + "placeholder": "​", + "style": "IPY_MODEL_40ff142202c34dbab894478bd2a0c763", + "value": " 111/111 [00:10<00:00, 10.56it/s]" + } + }, + "6005f7f4716344d3af218ce48c5f0a75": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "d0a86be8167a4ef0a7b7d0992489a1f3": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ca5262b4ab044828afb48613b837352a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "7e6d57fc41074dd19e0c2cabf17734ff": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0b8c1d90464740ee874c8bd230734920": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "a0465d556fc44a8cbe2386ab9e730ac8": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "40ff142202c34dbab894478bd2a0c763": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "45aa5c19f3174cf4a87b79fa8461b8e9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_935fd5c59e7840caa5bf9f14040dd606", + "IPY_MODEL_f4eb7e61c5bd4359aeea893256cfff96", + "IPY_MODEL_9b253f5fa36f4164a3f6b3a50279b0d4" + ], + "layout": "IPY_MODEL_24a7bd5f50ed489a985c64560ba92fdd" + } + }, + "935fd5c59e7840caa5bf9f14040dd606": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_433065beb5494c72acda32eb5f3e3e4c", + "placeholder": "​", + "style": "IPY_MODEL_8f33b37f98d44b86a308deb2ee21f9e8", + "value": "Epoch 5 / 8: 100%" + } + }, + "f4eb7e61c5bd4359aeea893256cfff96": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_6140a2cfb0db4e219a2fa19b567e51bf", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_2fe3b150673a428da0d6c05df7ae02fc", + "value": 111 + } + }, + "9b253f5fa36f4164a3f6b3a50279b0d4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_74fc91c342204076a772d6e9d5a3b060", + "placeholder": "​", + "style": "IPY_MODEL_bca338b823bb4459ba3eb00498452cc5", + "value": " 111/111 [00:10<00:00, 10.51it/s]" + } + }, + "24a7bd5f50ed489a985c64560ba92fdd": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "433065beb5494c72acda32eb5f3e3e4c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "8f33b37f98d44b86a308deb2ee21f9e8": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6140a2cfb0db4e219a2fa19b567e51bf": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2fe3b150673a428da0d6c05df7ae02fc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "74fc91c342204076a772d6e9d5a3b060": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "bca338b823bb4459ba3eb00498452cc5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1a19f0f8d5424461b0b1f02f7e9cad5b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_298f8eb6997b482290cc15add49ca2b1", + "IPY_MODEL_8c32ecde0b6f4535abc2b9cffe795d46", + "IPY_MODEL_1e4b66d2b5b247c7b507cea16926e199" + ], + "layout": "IPY_MODEL_0587a9a7e9974ce38fb89a4afa2bc0f9" + } + }, + "298f8eb6997b482290cc15add49ca2b1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_6c75063db68e4c68ae9e70f05b9fb1fe", + "placeholder": "​", + "style": "IPY_MODEL_6acae9f7715a4596aa5aac6680691864", + "value": "Epoch 6 / 8: 100%" + } + }, + "8c32ecde0b6f4535abc2b9cffe795d46": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1e0d83608f0648369ec93b617ffa5045", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_755446d0cbec41a99d3c5bb0fea917ee", + "value": 111 + } + }, + "1e4b66d2b5b247c7b507cea16926e199": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_9b677e467d6445f3a0c1ea363edf2b60", + "placeholder": "​", + "style": "IPY_MODEL_9c6276465c8f433e8985e27b8842fc72", + "value": " 111/111 [00:10<00:00, 10.29it/s]" + } + }, + "0587a9a7e9974ce38fb89a4afa2bc0f9": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6c75063db68e4c68ae9e70f05b9fb1fe": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6acae9f7715a4596aa5aac6680691864": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1e0d83608f0648369ec93b617ffa5045": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "755446d0cbec41a99d3c5bb0fea917ee": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "9b677e467d6445f3a0c1ea363edf2b60": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9c6276465c8f433e8985e27b8842fc72": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "152e4561c7e74ad0877a5160552309b4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_72e2d3013ea14bb59d684051b1b3af10", + "IPY_MODEL_1cd2ee93a65f4d4a81abc1c75d40c893", + "IPY_MODEL_473c81b40a4a4750b791862c5334cbff" + ], + "layout": "IPY_MODEL_8ad962104312403eaf803d2bf4d996a1" + } + }, + "72e2d3013ea14bb59d684051b1b3af10": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_bab012306a4b47ea9fcf46a0d7a7a6fa", + "placeholder": "​", + "style": "IPY_MODEL_166228d1351d4d2f8ee91c3c744345d1", + "value": "Epoch 7 / 8: 100%" + } + }, + "1cd2ee93a65f4d4a81abc1c75d40c893": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_9ae45a2f9f304fe1b0a2df60517a047b", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_c31e9412f7ac4f27999385a65596825a", + "value": 111 + } + }, + "473c81b40a4a4750b791862c5334cbff": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_05910225a05d4b3f8a00c7252fde722c", + "placeholder": "​", + "style": "IPY_MODEL_f8b7ca8a9f2f45d49c54db98eed9bcc4", + "value": " 111/111 [00:10<00:00, 10.51it/s]" + } + }, + "8ad962104312403eaf803d2bf4d996a1": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "bab012306a4b47ea9fcf46a0d7a7a6fa": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "166228d1351d4d2f8ee91c3c744345d1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "9ae45a2f9f304fe1b0a2df60517a047b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c31e9412f7ac4f27999385a65596825a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "05910225a05d4b3f8a00c7252fde722c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f8b7ca8a9f2f45d49c54db98eed9bcc4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1208473505b2422ab4d340f75fdf2624": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_eaf7dc090a2747b5877e57f4e425edf9", + "IPY_MODEL_0f624dde85fe446092a9491797a1badc", + "IPY_MODEL_ae52b5b14f254763b75e4820ed2b2a46" + ], + "layout": "IPY_MODEL_2481fee3c6f8401e86d2292b4468e631" + } + }, + "eaf7dc090a2747b5877e57f4e425edf9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1ec05e0fd379402299be8643e14af459", + "placeholder": "​", + "style": "IPY_MODEL_3837db24ce4f44c9acdde8e045c3b27c", + "value": "Epoch 0 / 6: 100%" + } + }, + "0f624dde85fe446092a9491797a1badc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_56eacd3cfe8440a4a76805c72a8b2d6b", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_f1a528076c1846d8b62b61411a128510", + "value": 111 + } + }, + "ae52b5b14f254763b75e4820ed2b2a46": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_454b1597345b4423bd7cfdb48135798f", + "placeholder": "​", + "style": "IPY_MODEL_aa13b7cdd8054693aef1544a6d5ebae9", + "value": " 111/111 [00:12<00:00,  8.94it/s]" + } + }, + "2481fee3c6f8401e86d2292b4468e631": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "1ec05e0fd379402299be8643e14af459": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "3837db24ce4f44c9acdde8e045c3b27c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "56eacd3cfe8440a4a76805c72a8b2d6b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f1a528076c1846d8b62b61411a128510": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "454b1597345b4423bd7cfdb48135798f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "aa13b7cdd8054693aef1544a6d5ebae9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "f2172f4996634581895b73ee0fbde3b0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_b95ec2a4fc4c4dfa8464ea5fbc8b44ba", + "IPY_MODEL_535195a6b5f64e3890094ca90f8f2f8a", + "IPY_MODEL_0e73f2a03d8a4eb3aaf0b82c3afd478c" + ], + "layout": "IPY_MODEL_45e10b07350d4aa8804720fa91de07dc" + } + }, + "b95ec2a4fc4c4dfa8464ea5fbc8b44ba": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_54540285fbe640d6b235079a1b7ec76b", + "placeholder": "​", + "style": "IPY_MODEL_ec3631996290457395f020b96f39059e", + "value": "Epoch 1 / 6: 100%" + } + }, + "535195a6b5f64e3890094ca90f8f2f8a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_6a8cc3adc3b747cbba369de2f30e9755", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ac22b083ff09461ba8d019a263b1ac7f", + "value": 111 + } + }, + "0e73f2a03d8a4eb3aaf0b82c3afd478c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_2dafaea27a0e4366b023c6ba95640c70", + "placeholder": "​", + "style": "IPY_MODEL_e8c5d749601d4087b26d5f3dc693a7fc", + "value": " 111/111 [00:12<00:00,  9.06it/s]" + } + }, + "45e10b07350d4aa8804720fa91de07dc": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "54540285fbe640d6b235079a1b7ec76b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ec3631996290457395f020b96f39059e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6a8cc3adc3b747cbba369de2f30e9755": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ac22b083ff09461ba8d019a263b1ac7f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "2dafaea27a0e4366b023c6ba95640c70": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "e8c5d749601d4087b26d5f3dc693a7fc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "145d00cbda4e4fa19f0cb196a03e95a2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_cd89abe8df764865aa5c11eab0a89edb", + "IPY_MODEL_6647f50265294df1a9534b7c4c062329", + "IPY_MODEL_40d8277ba9c94613bb52570b2b97a75b" + ], + "layout": "IPY_MODEL_8d4c3e7c6b554b0fba64463a031b5365" + } + }, + "cd89abe8df764865aa5c11eab0a89edb": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_cd14a2c1a2ae45f48dca89ec4f84e45b", + "placeholder": "​", + "style": "IPY_MODEL_ae2d1da470b2401e8145d1e458c3937f", + "value": "Epoch 2 / 6: 100%" + } + }, + "6647f50265294df1a9534b7c4c062329": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5b63eaf4f2b44638a73c6751e89ee78c", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_8c8e6bef4c0f4f069c0a67bbc9923270", + "value": 111 + } + }, + "40d8277ba9c94613bb52570b2b97a75b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_2dad35f0ee3c4650a6993b272ea97604", + "placeholder": "​", + "style": "IPY_MODEL_c3628243bd6f421d8c3a4759c4af037e", + "value": " 111/111 [00:12<00:00,  9.02it/s]" + } + }, + "8d4c3e7c6b554b0fba64463a031b5365": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "cd14a2c1a2ae45f48dca89ec4f84e45b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ae2d1da470b2401e8145d1e458c3937f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "5b63eaf4f2b44638a73c6751e89ee78c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "8c8e6bef4c0f4f069c0a67bbc9923270": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "2dad35f0ee3c4650a6993b272ea97604": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c3628243bd6f421d8c3a4759c4af037e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "8aa897cf6ab94954b05f025a70ac74fa": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_850516d144dd46d1aa2ad8dfa44ed356", + "IPY_MODEL_21513d8d090c45a492e38d1b0a8d1b0b", + "IPY_MODEL_bc8c53cacd4f4cf6a9ae68ec46d5599e" + ], + "layout": "IPY_MODEL_d6c219545379465b81a638a0bd0cf5ab" + } + }, + "850516d144dd46d1aa2ad8dfa44ed356": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_c956ee98682d4ffd989d13d11e4b76b1", + "placeholder": "​", + "style": "IPY_MODEL_59ac6c87be9840b1b139ecd8f1ebe4f0", + "value": "Epoch 3 / 6: 100%" + } + }, + "21513d8d090c45a492e38d1b0a8d1b0b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f6edc0c3386549419ebdd1aa9629720e", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_d6fab987b0ee49e7bd81a0662d542f95", + "value": 111 + } + }, + "bc8c53cacd4f4cf6a9ae68ec46d5599e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1b36ff4df62e415986e872a517090489", + "placeholder": "​", + "style": "IPY_MODEL_213556c80e074d62b9f0846a3d03c6c4", + "value": " 111/111 [00:12<00:00,  9.02it/s]" + } + }, + "d6c219545379465b81a638a0bd0cf5ab": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c956ee98682d4ffd989d13d11e4b76b1": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "59ac6c87be9840b1b139ecd8f1ebe4f0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "f6edc0c3386549419ebdd1aa9629720e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "d6fab987b0ee49e7bd81a0662d542f95": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "1b36ff4df62e415986e872a517090489": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "213556c80e074d62b9f0846a3d03c6c4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "cc0975931770476d95b005c2563add64": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_940a7d31a7ec4658b643c1c42ef2ce7e", + "IPY_MODEL_bb0290be80194bffad16464ee66a53ef", + "IPY_MODEL_0d449cccd9ae476a8afeb48f8064bdbf" + ], + "layout": "IPY_MODEL_160daa533558466bbadb9deadf06f5fa" + } + }, + "940a7d31a7ec4658b643c1c42ef2ce7e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_81ace1e936c844739907de1a9d505ad6", + "placeholder": "​", + "style": "IPY_MODEL_7fb2ba8001fb4b9088970b067689451b", + "value": "Epoch 4 / 6: 100%" + } + }, + "bb0290be80194bffad16464ee66a53ef": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5f352652b4304795ac2498ca48c38acd", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_7fc7a26fedb34b4e9c0e179f1488a518", + "value": 111 + } + }, + "0d449cccd9ae476a8afeb48f8064bdbf": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a80f84c3148e4520885d34fad70a12f8", + "placeholder": "​", + "style": "IPY_MODEL_1ffbd742231947a2941d0fbdd226dba1", + "value": " 111/111 [00:12<00:00,  9.07it/s]" + } + }, + "160daa533558466bbadb9deadf06f5fa": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "81ace1e936c844739907de1a9d505ad6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7fb2ba8001fb4b9088970b067689451b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "5f352652b4304795ac2498ca48c38acd": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7fc7a26fedb34b4e9c0e179f1488a518": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "a80f84c3148e4520885d34fad70a12f8": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "1ffbd742231947a2941d0fbdd226dba1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "17ff4d9b65144ddb8fa8dd31c2d07123": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_9c1a1447733342b8864f9613bb9e0ece", + "IPY_MODEL_75650414085e46058c2d6780c165bc15", + "IPY_MODEL_e89739f7f6a746cfa7d886f12a0b85e0" + ], + "layout": "IPY_MODEL_02404160a8cf400b8847bed459f99adb" + } + }, + "9c1a1447733342b8864f9613bb9e0ece": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_20ebb4063ffe4605a6b32b713e40872b", + "placeholder": "​", + "style": "IPY_MODEL_f667456051a1407f8fec350fb8308544", + "value": "config.json: " + } + }, + "75650414085e46058c2d6780c165bc15": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_408284147a604a61aecb1e7987674424", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ebb85543376e49688b7d83c5658125ab", + "value": 1 + } + }, + "e89739f7f6a746cfa7d886f12a0b85e0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_0947bb8cceea4bb28a87fac1f8250232", + "placeholder": "​", + "style": "IPY_MODEL_060891df058b4969ab26f598b3f170dd", + "value": " 4.10k/? [00:00<00:00, 446kB/s]" + } + }, + "02404160a8cf400b8847bed459f99adb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "20ebb4063ffe4605a6b32b713e40872b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f667456051a1407f8fec350fb8308544": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "408284147a604a61aecb1e7987674424": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "ebb85543376e49688b7d83c5658125ab": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "0947bb8cceea4bb28a87fac1f8250232": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "060891df058b4969ab26f598b3f170dd": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "5124d05b487548efa8d632d2e56a378d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_ccdad8c4322f40c4ac57458932dbe9b3", + "IPY_MODEL_dd975925ee28497d8425b2df09239464", + "IPY_MODEL_c04f95832eed46538193d7b797f9890e" + ], + "layout": "IPY_MODEL_3c2139e144914573a748b2666c10b37d" + } + }, + "ccdad8c4322f40c4ac57458932dbe9b3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_c56c938ec5d744e0b941703f7a0ed77b", + "placeholder": "​", + "style": "IPY_MODEL_54db197cbfd846d3b32cf4fcb012696a", + "value": "pytorch_model.bin: 100%" + } + }, + "dd975925ee28497d8425b2df09239464": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_bd438477c5f54c7db0b05a8d18eba83b", + "max": 598641023, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_78b5b937718e42eca2a5837ddb4fbc98", + "value": 598641023 + } + }, + "c04f95832eed46538193d7b797f9890e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_882a86c503dd41fb897a03ac864750b5", + "placeholder": "​", + "style": "IPY_MODEL_ea6dda3be9ca41978dc3567502a41170", + "value": " 599M/599M [00:03<00:00, 509MB/s]" + } + }, + "3c2139e144914573a748b2666c10b37d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c56c938ec5d744e0b941703f7a0ed77b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "54db197cbfd846d3b32cf4fcb012696a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "bd438477c5f54c7db0b05a8d18eba83b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "78b5b937718e42eca2a5837ddb4fbc98": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "882a86c503dd41fb897a03ac864750b5": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ea6dda3be9ca41978dc3567502a41170": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "afda154ccc33433992dd92accd0ac591": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_d31515b9f55c4894b2fa569f22cbbf13", + "IPY_MODEL_7772256ec8fa4892942cd40860413f9d", + "IPY_MODEL_c4399ea347c241cf9a9c49ff507fc5d6" + ], + "layout": "IPY_MODEL_9b6606f7713144a9a1f80cb374bcd42d" + } + }, + "d31515b9f55c4894b2fa569f22cbbf13": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7882b1253cef420a816ab2cd13dffebb", + "placeholder": "​", + "style": "IPY_MODEL_8e7e4acba3fc451baf6d93aa9c241e37", + "value": "model.safetensors: 100%" + } + }, + "7772256ec8fa4892942cd40860413f9d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_b367d095d4634e1eaac4505b3a1bffdc", + "max": 598532764, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_7ca6bbcc482c4d5588432c794ef831b1", + "value": 598532764 + } + }, + "c4399ea347c241cf9a9c49ff507fc5d6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_873bcb9bc9cf4be1b68966a33f69ffed", + "placeholder": "​", + "style": "IPY_MODEL_6720555b58834eb8bef867b7ae17ab44", + "value": " 599M/599M [00:02<00:00, 596MB/s]" + } + }, + "9b6606f7713144a9a1f80cb374bcd42d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7882b1253cef420a816ab2cd13dffebb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "8e7e4acba3fc451baf6d93aa9c241e37": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "b367d095d4634e1eaac4505b3a1bffdc": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7ca6bbcc482c4d5588432c794ef831b1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "873bcb9bc9cf4be1b68966a33f69ffed": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6720555b58834eb8bef867b7ae17ab44": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6ed844b9111c4bfeb9f1786ed547b6c5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_4746fe0ceedf4534a42f15d0be5ca294", + "IPY_MODEL_39ef7fec58834e6da5e70f600b12ad9a", + "IPY_MODEL_6ecd1219fad749bdae0d35037c2b7413" + ], + "layout": "IPY_MODEL_c3d40f44bde140a1b55af0a6b356fe7e" + } + }, + "4746fe0ceedf4534a42f15d0be5ca294": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7771c3a2c15c425e9a6926a4a0723763", + "placeholder": "​", + "style": "IPY_MODEL_83c1964e90c3435bb5747b7dbc16ab51", + "value": "Epoch 0 / 8: 100%" + } + }, + "39ef7fec58834e6da5e70f600b12ad9a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_d9be164056d14240a7e4815e4b1849fb", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_8f0686cb6eb64035900ed8143f46e34d", + "value": 111 + } + }, + "6ecd1219fad749bdae0d35037c2b7413": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_b9d1d2482bb14caabaa444ccf9c61810", + "placeholder": "​", + "style": "IPY_MODEL_7ee5183e24204234a4151a76a8e37e80", + "value": " 111/111 [00:12<00:00,  8.94it/s]" + } + }, + "c3d40f44bde140a1b55af0a6b356fe7e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7771c3a2c15c425e9a6926a4a0723763": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "83c1964e90c3435bb5747b7dbc16ab51": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "d9be164056d14240a7e4815e4b1849fb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "8f0686cb6eb64035900ed8143f46e34d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "b9d1d2482bb14caabaa444ccf9c61810": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7ee5183e24204234a4151a76a8e37e80": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "d3cac820aec145d6a74cf6620d67dc6c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_b190c98a9ed141f29be9f02816c2280b", + "IPY_MODEL_ad50688abbcd4ccf9a61066c2e0fc1c3", + "IPY_MODEL_f83ff6dd01c24bad90423d81874c3154" + ], + "layout": "IPY_MODEL_5561c483b5024516a87aa1f7d7b095d2" + } + }, + "b190c98a9ed141f29be9f02816c2280b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_dd0a7f174bbd48498431bceae2aa692f", + "placeholder": "​", + "style": "IPY_MODEL_3e0ea1c5547241c5a1c185b36c4e514c", + "value": "Epoch 1 / 8: 100%" + } + }, + "ad50688abbcd4ccf9a61066c2e0fc1c3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f5bd39c7f4364e86b047135a482dc946", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ea2826306cfa4558a5d78c4b388a92f5", + "value": 111 + } + }, + "f83ff6dd01c24bad90423d81874c3154": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_d809df30245345d9a92bc1062629ed38", + "placeholder": "​", + "style": "IPY_MODEL_ca33dc1d75d0479aaeb7b863e5b935af", + "value": " 111/111 [00:12<00:00,  9.18it/s]" + } + }, + "5561c483b5024516a87aa1f7d7b095d2": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "dd0a7f174bbd48498431bceae2aa692f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "3e0ea1c5547241c5a1c185b36c4e514c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "f5bd39c7f4364e86b047135a482dc946": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ea2826306cfa4558a5d78c4b388a92f5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "d809df30245345d9a92bc1062629ed38": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ca33dc1d75d0479aaeb7b863e5b935af": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "d4c35e6ebb5d4826ac06c50921ef2ae0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_0326964e0d474befb108fed0236c253b", + "IPY_MODEL_d3a7df9ccbbf4ad89d34535e44ba7711", + "IPY_MODEL_6507c2e2b0c34d7096607249b8c5c636" + ], + "layout": "IPY_MODEL_58733f51e37e461ab413c5955d07456d" + } + }, + "0326964e0d474befb108fed0236c253b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f5285642b5dd4432938e96a298225ffe", + "placeholder": "​", + "style": "IPY_MODEL_0ad6d10f1d214401bc20c20e455bbe5f", + "value": "Epoch 2 / 8: 100%" + } + }, + "d3a7df9ccbbf4ad89d34535e44ba7711": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_56dd925432aa4c9f866ba1c1353fa532", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_eecae0d48e6d4964812b554fdf153927", + "value": 111 + } + }, + "6507c2e2b0c34d7096607249b8c5c636": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1d1a072f1df841e29bd7475a79b41cbb", + "placeholder": "​", + "style": "IPY_MODEL_bc9de510af614acbb63e6df52f40b180", + "value": " 111/111 [00:12<00:00,  9.14it/s]" + } + }, + "58733f51e37e461ab413c5955d07456d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f5285642b5dd4432938e96a298225ffe": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0ad6d10f1d214401bc20c20e455bbe5f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "56dd925432aa4c9f866ba1c1353fa532": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "eecae0d48e6d4964812b554fdf153927": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "1d1a072f1df841e29bd7475a79b41cbb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "bc9de510af614acbb63e6df52f40b180": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "ae569cea2e2940599c76c5b42eb2a7b1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_ed694382db2643c69498e549dd6257ed", + "IPY_MODEL_a5621fadc92e4550bc4a08ad25542a42", + "IPY_MODEL_b6e50daae0d6485fa4be921c49e44951" + ], + "layout": "IPY_MODEL_49c1979a183c448ba7af8340d3b6eded" + } + }, + "ed694382db2643c69498e549dd6257ed": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a9fac8fea20949f8af0351be359c27ca", + "placeholder": "​", + "style": "IPY_MODEL_fc552eb6d9144e0fb54ced38fd137967", + "value": "Epoch 3 / 8: 100%" + } + }, + "a5621fadc92e4550bc4a08ad25542a42": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_0e31715ce25c48f8b2602013c683e88e", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_00a3247d646a46649c5b1adc45f6e61e", + "value": 111 + } + }, + "b6e50daae0d6485fa4be921c49e44951": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_237a4edbd79745ee99839d0009ca7974", + "placeholder": "​", + "style": "IPY_MODEL_c6893e198a9448d39c76735ea7196599", + "value": " 111/111 [00:12<00:00,  9.14it/s]" + } + }, + "49c1979a183c448ba7af8340d3b6eded": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a9fac8fea20949f8af0351be359c27ca": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "fc552eb6d9144e0fb54ced38fd137967": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "0e31715ce25c48f8b2602013c683e88e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "00a3247d646a46649c5b1adc45f6e61e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "237a4edbd79745ee99839d0009ca7974": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c6893e198a9448d39c76735ea7196599": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "9841879c2a81434bb105f35c0c0df4da": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_41ba6851eafc4654b1cd1fe70fd27b72", + "IPY_MODEL_bd08b32fe5d24bb78ba7bc9afae74ea6", + "IPY_MODEL_bcaf3335f18f4b6fa31ad56aa347d128" + ], + "layout": "IPY_MODEL_5ca92f977bfb43db851d4db95ae90aaa" + } + }, + "41ba6851eafc4654b1cd1fe70fd27b72": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_e63571309d9c42edb5452b94d79b6b45", + "placeholder": "​", + "style": "IPY_MODEL_036113a86f91452393df58707c767159", + "value": "Epoch 4 / 8: 100%" + } + }, + "bd08b32fe5d24bb78ba7bc9afae74ea6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_6ca94caf7310455bb0c688ffcef45c9e", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_669d576e6f024beb9701e43f57c2e2bd", + "value": 111 + } + }, + "bcaf3335f18f4b6fa31ad56aa347d128": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1960633ff12643f2a66ef6cff4d5ab08", + "placeholder": "​", + "style": "IPY_MODEL_226feeb0196a40798a9a35ed28b83310", + "value": " 111/111 [00:12<00:00,  9.03it/s]" + } + }, + "5ca92f977bfb43db851d4db95ae90aaa": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "e63571309d9c42edb5452b94d79b6b45": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "036113a86f91452393df58707c767159": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6ca94caf7310455bb0c688ffcef45c9e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "669d576e6f024beb9701e43f57c2e2bd": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "1960633ff12643f2a66ef6cff4d5ab08": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "226feeb0196a40798a9a35ed28b83310": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "2e7601a2dde946bbb5fc76a49eb3c8a5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_f41daa4104104a9f830e6fa66a65a4e5", + "IPY_MODEL_1085542911ba4dea95caa46de62887e4", + "IPY_MODEL_12aa1feb707e49b1a4e1a46a9b6f4978" + ], + "layout": "IPY_MODEL_f50d1ec2115f4e049309276bfb13d911" + } + }, + "f41daa4104104a9f830e6fa66a65a4e5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7dc0b113b55f4bc8aad222d6acd25fe2", + "placeholder": "​", + "style": "IPY_MODEL_ff9e7cf2f0f14219babf1537fbc929ea", + "value": "Epoch 5 / 8: 100%" + } + }, + "1085542911ba4dea95caa46de62887e4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1b8ddeb958b64f9db5fcd5934f5dd7ac", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_60267770a60447998c42712e77fe7c8c", + "value": 111 + } + }, + "12aa1feb707e49b1a4e1a46a9b6f4978": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_8fe7cac3a4ff46a0a4e9a94884d857ba", + "placeholder": "​", + "style": "IPY_MODEL_3064c9a6f7654d6e94a859f441abe372", + "value": " 111/111 [00:12<00:00,  9.21it/s]" + } + }, + "f50d1ec2115f4e049309276bfb13d911": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7dc0b113b55f4bc8aad222d6acd25fe2": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ff9e7cf2f0f14219babf1537fbc929ea": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1b8ddeb958b64f9db5fcd5934f5dd7ac": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "60267770a60447998c42712e77fe7c8c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "8fe7cac3a4ff46a0a4e9a94884d857ba": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "3064c9a6f7654d6e94a859f441abe372": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "ebfe58a7b017416caa315c19edbd04d1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_832972f79a544b33861b298ffdea3484", + "IPY_MODEL_4e0d59d31090455ea10b9587d8ef11c8", + "IPY_MODEL_83a5a5be6f824984a83cd991e8075864" + ], + "layout": "IPY_MODEL_c3f5b71ae28a4adeb8dc70d54f4807cb" + } + }, + "832972f79a544b33861b298ffdea3484": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a14fb84a7f6c4128ad0c54087fc98399", + "placeholder": "​", + "style": "IPY_MODEL_f714752eaa21402d9a301549edbc7e86", + "value": "Epoch 6 / 8: 100%" + } + }, + "4e0d59d31090455ea10b9587d8ef11c8": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_3e708cfca5224302972c6c13896e3019", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_417d5670a562486284b44b58d3458ef2", + "value": 111 + } + }, + "83a5a5be6f824984a83cd991e8075864": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_82778277f0a44a83891454cf9135beb0", + "placeholder": "​", + "style": "IPY_MODEL_7e9f69e2bf77462a9105267211219efd", + "value": " 111/111 [00:12<00:00,  9.05it/s]" + } + }, + "c3f5b71ae28a4adeb8dc70d54f4807cb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a14fb84a7f6c4128ad0c54087fc98399": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f714752eaa21402d9a301549edbc7e86": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3e708cfca5224302972c6c13896e3019": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "417d5670a562486284b44b58d3458ef2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "82778277f0a44a83891454cf9135beb0": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7e9f69e2bf77462a9105267211219efd": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "6aeb0f6b8794407ead0aa701a564c84e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_0cb4d00cbc7e4efb9f7fd15dea54dac7", + "IPY_MODEL_d22fd3cddb8448feb537b362ae9041a1", + "IPY_MODEL_b9a01776915a4d939a876658f2ef8984" + ], + "layout": "IPY_MODEL_30cc8609dff24add8192ad97cdc7f042" + } + }, + "0cb4d00cbc7e4efb9f7fd15dea54dac7": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_63da9593d9994d1080b0bd16e2e4c987", + "placeholder": "​", + "style": "IPY_MODEL_79714744b45745488af5545cef0c720f", + "value": "Epoch 7 / 8: 100%" + } + }, + "d22fd3cddb8448feb537b362ae9041a1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_07b06439eb084c5e9cfaecc5b5996244", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_c7c520e2701a4a51bc972cfddee30499", + "value": 111 + } + }, + "b9a01776915a4d939a876658f2ef8984": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_2fa444afdf754af69560e0f9b485b0a6", + "placeholder": "​", + "style": "IPY_MODEL_974419ca5f8d411e826f2a4077801a8f", + "value": " 111/111 [00:12<00:00,  9.08it/s]" + } + }, + "30cc8609dff24add8192ad97cdc7f042": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "63da9593d9994d1080b0bd16e2e4c987": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "79714744b45745488af5545cef0c720f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "07b06439eb084c5e9cfaecc5b5996244": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c7c520e2701a4a51bc972cfddee30499": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "2fa444afdf754af69560e0f9b485b0a6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "974419ca5f8d411e826f2a4077801a8f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "f3c30b8ba65d4d2c9d5f152dfef0f478": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_26f025ffa2234877a6ffced3b1a6da51", + "IPY_MODEL_991eab57abbd4fc19105b8a173e48df0", + "IPY_MODEL_5b4ecdf8e7614e639d8a0541032e9980" + ], + "layout": "IPY_MODEL_de28ca925ec14b4f86cb08cdaf953246" + } + }, + "26f025ffa2234877a6ffced3b1a6da51": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_2520aa9405554ecea28472beecc63d8c", + "placeholder": "​", + "style": "IPY_MODEL_b85c8cdc101c4791b89239c579529e3a", + "value": "Epoch 0 / 6: 100%" + } + }, + "991eab57abbd4fc19105b8a173e48df0": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_3ba1351969ae40d19c7b4a4231771d56", + "max": 222, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_a90a52b56c224656a0b7b0c8f9e195e9", + "value": 222 + } + }, + "5b4ecdf8e7614e639d8a0541032e9980": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7e1c1e9f666346f5bf20813e1d89c40c", + "placeholder": "​", + "style": "IPY_MODEL_5888a3665e6d46f98e3776190f3d88d1", + "value": " 222/222 [00:25<00:00,  8.85it/s]" + } + }, + "de28ca925ec14b4f86cb08cdaf953246": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2520aa9405554ecea28472beecc63d8c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b85c8cdc101c4791b89239c579529e3a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3ba1351969ae40d19c7b4a4231771d56": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a90a52b56c224656a0b7b0c8f9e195e9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "7e1c1e9f666346f5bf20813e1d89c40c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "5888a3665e6d46f98e3776190f3d88d1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "4eda23a75fc14231a320394e1704459b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_5cedf51ced2f4aa1b9700fefcb8683c9", + "IPY_MODEL_afce541f51e04b779b700bdd395be9b9", + "IPY_MODEL_c96cd84932f6465fbfb73b45c0fbd390" + ], + "layout": "IPY_MODEL_ac4b54ee1eb64084947416a0598c5531" + } + }, + "5cedf51ced2f4aa1b9700fefcb8683c9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7804eed2d5f140789848956e85f47964", + "placeholder": "​", + "style": "IPY_MODEL_6238f55ec53d46a3b239fffe41564abe", + "value": "Epoch 1 / 6: 100%" + } + }, + "afce541f51e04b779b700bdd395be9b9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_176d069c0e1d464ea00ebf26997860b7", + "max": 222, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_39c291de497a4b9f80446153862920dc", + "value": 222 + } + }, + "c96cd84932f6465fbfb73b45c0fbd390": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_c0580c1200ae4329beb9746e6014df57", + "placeholder": "​", + "style": "IPY_MODEL_d3b1db7a42784782b269924e5acf01b1", + "value": " 222/222 [00:25<00:00,  8.82it/s]" + } + }, + "ac4b54ee1eb64084947416a0598c5531": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7804eed2d5f140789848956e85f47964": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6238f55ec53d46a3b239fffe41564abe": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "176d069c0e1d464ea00ebf26997860b7": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "39c291de497a4b9f80446153862920dc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "c0580c1200ae4329beb9746e6014df57": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "d3b1db7a42784782b269924e5acf01b1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "d6aad418994141818e52e7f7e15cb395": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_281e4f80cc94472fbb477c08990d6a12", + "IPY_MODEL_1ef7fc87f5044c1bbedeaabbaa611e35", + "IPY_MODEL_821cd12c69cb412aa814f4a435e15d3f" + ], + "layout": "IPY_MODEL_11f1e3ee5b754675ad6c23373b146d93" + } + }, + "281e4f80cc94472fbb477c08990d6a12": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_0d979fbe8f374ddba5883007e1257b61", + "placeholder": "​", + "style": "IPY_MODEL_2c467a6882ce41d694ca7427e3fee1c7", + "value": "Epoch 2 / 6: 100%" + } + }, + "1ef7fc87f5044c1bbedeaabbaa611e35": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_9deb32ec403340cbab9cb7ff9fb1e695", + "max": 222, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_3679794cc9864a11ba7e8adfc2c169fb", + "value": 222 + } + }, + "821cd12c69cb412aa814f4a435e15d3f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_8921caaee769428d80124c8838a3afcc", + "placeholder": "​", + "style": "IPY_MODEL_686597fa4a0a422c93eeed3a921151d3", + "value": " 222/222 [00:25<00:00,  8.82it/s]" + } + }, + "11f1e3ee5b754675ad6c23373b146d93": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0d979fbe8f374ddba5883007e1257b61": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2c467a6882ce41d694ca7427e3fee1c7": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "9deb32ec403340cbab9cb7ff9fb1e695": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "3679794cc9864a11ba7e8adfc2c169fb": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "8921caaee769428d80124c8838a3afcc": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "686597fa4a0a422c93eeed3a921151d3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "8311331c9df54dd48a646234f483f2b9": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_014447a3fc5d48ec8e1b23ff4160e6e5", + "IPY_MODEL_a8a2e244598c4198b2798c3add87387e", + "IPY_MODEL_c3d4ad9cbdba425681e8ea1d9eacc517" + ], + "layout": "IPY_MODEL_a580de56d00242d8a0cc32ad0c058966" + } + }, + "014447a3fc5d48ec8e1b23ff4160e6e5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_701087566c2b43d3bda29e5880cf8b8d", + "placeholder": "​", + "style": "IPY_MODEL_0e3a2e96e0ca4c758073f29980686847", + "value": "Epoch 3 / 6: 100%" + } + }, + "a8a2e244598c4198b2798c3add87387e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_3ba43419ab4646e795850a1c845ed2e4", + "max": 222, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_76bdb4d6547d45afb26a4280f2d80fe1", + "value": 222 + } + }, + "c3d4ad9cbdba425681e8ea1d9eacc517": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_e50590f164474358b7e3906e97712a99", + "placeholder": "​", + "style": "IPY_MODEL_01c746e3760b4d6abb558b7225a4baec", + "value": " 222/222 [00:25<00:00,  8.88it/s]" + } + }, + "a580de56d00242d8a0cc32ad0c058966": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "701087566c2b43d3bda29e5880cf8b8d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0e3a2e96e0ca4c758073f29980686847": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3ba43419ab4646e795850a1c845ed2e4": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "76bdb4d6547d45afb26a4280f2d80fe1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "e50590f164474358b7e3906e97712a99": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "01c746e3760b4d6abb558b7225a4baec": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "8650efda983f4a7faf8ef53db4940bb2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_74e48d4a8d5b40b28c34f83747e6787d", + "IPY_MODEL_d250e38369ba4a3aa5b252094a8a64d4", + "IPY_MODEL_685925c238db45f2a2a8a7d21f28409b" + ], + "layout": "IPY_MODEL_8c6c12887ec74e26b54233315d7281c3" + } + }, + "74e48d4a8d5b40b28c34f83747e6787d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_29f539376c974eeeb112c7192edbfbe6", + "placeholder": "​", + "style": "IPY_MODEL_bee2979d2ef54113bfce39025caeedb3", + "value": "Epoch 4 / 6: 100%" + } + }, + "d250e38369ba4a3aa5b252094a8a64d4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_96b4a032bfae4f69bfbc827cc5e87ae1", + "max": 222, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_5754663c6bd24e358d0b8f2fafc150e8", + "value": 222 + } + }, + "685925c238db45f2a2a8a7d21f28409b": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_b79e971158ea42caaf3204ad582e4acb", + "placeholder": "​", + "style": "IPY_MODEL_a7bc995b6ef8467c8a6ef63da807bec6", + "value": " 222/222 [00:25<00:00,  8.88it/s]" + } + }, + "8c6c12887ec74e26b54233315d7281c3": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "29f539376c974eeeb112c7192edbfbe6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "bee2979d2ef54113bfce39025caeedb3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "96b4a032bfae4f69bfbc827cc5e87ae1": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "5754663c6bd24e358d0b8f2fafc150e8": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "b79e971158ea42caaf3204ad582e4acb": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a7bc995b6ef8467c8a6ef63da807bec6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "c46de3bd9e39471fa8826ace3e593e34": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_b6035ff4e176400cb12400f639ed222c", + "IPY_MODEL_31db83b5d183478a83b249fa55abea52", + "IPY_MODEL_2599b0241a1c4c03b81ad8c4f894196c" + ], + "layout": "IPY_MODEL_f016e66bd00c462aba684b558f896bde" + } + }, + "b6035ff4e176400cb12400f639ed222c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_68b65617c5f64a74ba6078ed325f2c5d", + "placeholder": "​", + "style": "IPY_MODEL_086869a1529d4b1d951702359d013603", + "value": "config.json: 100%" + } + }, + "31db83b5d183478a83b249fa55abea52": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f10173ae54894b08abdd88f1337dd675", + "max": 762, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_ad6eaa821fbb4d1586805401b998b3ab", + "value": 762 + } + }, + "2599b0241a1c4c03b81ad8c4f894196c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_cd3a1bcf9f064cacb0446ffd70c0f5ec", + "placeholder": "​", + "style": "IPY_MODEL_928ecbd873974212a48eeefc62c7d46a", + "value": " 762/762 [00:00<00:00, 97.7kB/s]" + } + }, + "f016e66bd00c462aba684b558f896bde": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "68b65617c5f64a74ba6078ed325f2c5d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "086869a1529d4b1d951702359d013603": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "f10173ae54894b08abdd88f1337dd675": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ad6eaa821fbb4d1586805401b998b3ab": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "cd3a1bcf9f064cacb0446ffd70c0f5ec": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "928ecbd873974212a48eeefc62c7d46a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "e6f2f9ba954e428fb9697d6d77cd8f50": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_0d65f204a24945ac8b5dce0912410924", + "IPY_MODEL_53c6a488d72840fd871b8d1b672815e1", + "IPY_MODEL_527f0f3cdd754811ab288d0e3ba54ecd" + ], + "layout": "IPY_MODEL_66370b7b5d6b4d22a1e4f8c4c1e2e893" + } + }, + "0d65f204a24945ac8b5dce0912410924": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_4ba69633b4184008806878b702e3e834", + "placeholder": "​", + "style": "IPY_MODEL_50190b37f58949a4a5ff88bab653d211", + "value": "model.safetensors: 100%" + } + }, + "53c6a488d72840fd871b8d1b672815e1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5f8e32bb621d4235894a6e81cc2a2bbc", + "max": 352824413, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_9adc0b70c4ad43db90ea9d871c07cd62", + "value": 352824413 + } + }, + "527f0f3cdd754811ab288d0e3ba54ecd": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_8e26ca16de8e4db8afe8a0773d2a9782", + "placeholder": "​", + "style": "IPY_MODEL_dd0a0337da5a4fc6b6d8c5170f13d7a2", + "value": " 353M/353M [00:02<00:00, 636MB/s]" + } + }, + "66370b7b5d6b4d22a1e4f8c4c1e2e893": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "4ba69633b4184008806878b702e3e834": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "50190b37f58949a4a5ff88bab653d211": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "5f8e32bb621d4235894a6e81cc2a2bbc": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "9adc0b70c4ad43db90ea9d871c07cd62": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "8e26ca16de8e4db8afe8a0773d2a9782": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "dd0a0337da5a4fc6b6d8c5170f13d7a2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3c22071591ca4bd2890b250e91232451": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_65a92039c1d94f1bae1fdb8a6358d9a3", + "IPY_MODEL_298fb7d77549426480ca49ff41c2e5b6", + "IPY_MODEL_2578c814a4b44ee2b41c8dd397dd8714" + ], + "layout": "IPY_MODEL_0b37c34db84946058a6f471002e4db0d" + } + }, + "65a92039c1d94f1bae1fdb8a6358d9a3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_c2c360cb61024279a9e8a867514f3903", + "placeholder": "​", + "style": "IPY_MODEL_afa207ff18074896bbf0f6ec9840c5f3", + "value": "generation_config.json: 100%" + } + }, + "298fb7d77549426480ca49ff41c2e5b6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_0e7fcd7d29c9412ea0397a5fe53c0b83", + "max": 124, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_0946eb66f7af47bdbf7d040aa260ebea", + "value": 124 + } + }, + "2578c814a4b44ee2b41c8dd397dd8714": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_45e244788ee2426bb492171e5cb72d46", + "placeholder": "​", + "style": "IPY_MODEL_264ac655bc6748f09f441e6d0b68e974", + "value": " 124/124 [00:00<00:00, 15.1kB/s]" + } + }, + "0b37c34db84946058a6f471002e4db0d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "c2c360cb61024279a9e8a867514f3903": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "afa207ff18074896bbf0f6ec9840c5f3": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "0e7fcd7d29c9412ea0397a5fe53c0b83": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0946eb66f7af47bdbf7d040aa260ebea": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "45e244788ee2426bb492171e5cb72d46": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "264ac655bc6748f09f441e6d0b68e974": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "fa1d910e3ad94f5ba006ebf1bac1c8c1": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_aeecfc01b3eb41bfb7277e40805c914f", + "IPY_MODEL_b03546ce252c4fdf9bb8e828e87951fc", + "IPY_MODEL_fd5915c6296f4bb6bd5439a918345c7a" + ], + "layout": "IPY_MODEL_2bb7313bc37c4ae48cdbc2f3e4c8ffff" + } + }, + "aeecfc01b3eb41bfb7277e40805c914f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_200faf085a1b48c9b65cdc00f40f5330", + "placeholder": "​", + "style": "IPY_MODEL_7ef501ab01ec4305ab55274000ce513d", + "value": "tokenizer_config.json: 100%" + } + }, + "b03546ce252c4fdf9bb8e828e87951fc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_ea63c7becf2f4da786ec1a17ca848b4e", + "max": 26, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_f7abafc20e7a4d4e8a3e1d85589d1cff", + "value": 26 + } + }, + "fd5915c6296f4bb6bd5439a918345c7a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_7c3402a99e9349c9a2849af2eedd8edf", + "placeholder": "​", + "style": "IPY_MODEL_0627a9c525594144906b94adec04687d", + "value": " 26.0/26.0 [00:00<00:00, 3.63kB/s]" + } + }, + "2bb7313bc37c4ae48cdbc2f3e4c8ffff": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "200faf085a1b48c9b65cdc00f40f5330": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7ef501ab01ec4305ab55274000ce513d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "ea63c7becf2f4da786ec1a17ca848b4e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f7abafc20e7a4d4e8a3e1d85589d1cff": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "7c3402a99e9349c9a2849af2eedd8edf": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "0627a9c525594144906b94adec04687d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "a3e9d05981b04f10829d10c3a2087a4c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_ca0b14b068f042768cc84f71d8c39428", + "IPY_MODEL_be99131b676f4b0fa91b1f8b8cbae825", + "IPY_MODEL_686c0955edcf48d4823b4497a4a16e5d" + ], + "layout": "IPY_MODEL_598c4b8ce296478f8a63a7d51bf54882" + } + }, + "ca0b14b068f042768cc84f71d8c39428": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f7ede2830bbd4ba391268f5416cdea2e", + "placeholder": "​", + "style": "IPY_MODEL_81a006c41ff046b3803a6bfbb577dd2d", + "value": "vocab.json: " + } + }, + "be99131b676f4b0fa91b1f8b8cbae825": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_c8f89d4734944fe69f30b1fdc6d396a3", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_1fce43f844104e15a1fe1e4e6f2bacf6", + "value": 1 + } + }, + "686c0955edcf48d4823b4497a4a16e5d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_33c8984b05be45249a9cec4b6f30e557", + "placeholder": "​", + "style": "IPY_MODEL_5b98619e1642470a83db408af792086a", + "value": " 1.04M/? [00:00<00:00, 2.01MB/s]" + } + }, + "598c4b8ce296478f8a63a7d51bf54882": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "f7ede2830bbd4ba391268f5416cdea2e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "81a006c41ff046b3803a6bfbb577dd2d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "c8f89d4734944fe69f30b1fdc6d396a3": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "1fce43f844104e15a1fe1e4e6f2bacf6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "33c8984b05be45249a9cec4b6f30e557": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "5b98619e1642470a83db408af792086a": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "e889ea7b6d374e11a17e533e05b4802e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_876435064a6446488a819cb8e57fd165", + "IPY_MODEL_92c7a231bb9648be9fc53628f41b64be", + "IPY_MODEL_1190b01146ec41f7a0aceee35df65021" + ], + "layout": "IPY_MODEL_7b1350e28f0f47ba8cefc94189f5fe3d" + } + }, + "876435064a6446488a819cb8e57fd165": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_a2aace6cb622465fb39f75bc80365b9b", + "placeholder": "​", + "style": "IPY_MODEL_b952a86b6b504e6890e4e82157a87b69", + "value": "merges.txt: " + } + }, + "92c7a231bb9648be9fc53628f41b64be": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_588e5cbb4faa40fda9c24258fe8f983c", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_c050d49700d74c93a6b28a09dfc36bbc", + "value": 1 + } + }, + "1190b01146ec41f7a0aceee35df65021": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5fe3c5c3200645eb9609310043194ebd", + "placeholder": "​", + "style": "IPY_MODEL_76e45856e51d4e38b27c3d04bff6809f", + "value": " 456k/? [00:00<00:00, 3.17MB/s]" + } + }, + "7b1350e28f0f47ba8cefc94189f5fe3d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "a2aace6cb622465fb39f75bc80365b9b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b952a86b6b504e6890e4e82157a87b69": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "588e5cbb4faa40fda9c24258fe8f983c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "c050d49700d74c93a6b28a09dfc36bbc": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "5fe3c5c3200645eb9609310043194ebd": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "76e45856e51d4e38b27c3d04bff6809f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "7a9afe15dd9c4ba3b6791e724ff46610": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_c88d8ec092fb40828a3149790ef51c74", + "IPY_MODEL_01a6be0efa3f4ce5a1ff848c9077142c", + "IPY_MODEL_b3fb82365c43434086393575c36bfaef" + ], + "layout": "IPY_MODEL_9f581d40efd7478bad2c66140e577d16" + } + }, + "c88d8ec092fb40828a3149790ef51c74": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_36680e0aac814e9abdbb95b4f2489442", + "placeholder": "​", + "style": "IPY_MODEL_2086ecef6725402aa77717a82d5fbbc4", + "value": "tokenizer.json: " + } + }, + "01a6be0efa3f4ce5a1ff848c9077142c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_0e690f0a05f34a55a366078cc23e2568", + "max": 1, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_db3b5313ec074f2982a8cf7f4bed9ed2", + "value": 1 + } + }, + "b3fb82365c43434086393575c36bfaef": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_c10ec37dcf0a4f488167cfc3ca11a172", + "placeholder": "​", + "style": "IPY_MODEL_bce09c6f8fa84eb090d75b2f4b67a00f", + "value": " 1.36M/? [00:00<00:00, 3.88MB/s]" + } + }, + "9f581d40efd7478bad2c66140e577d16": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "36680e0aac814e9abdbb95b4f2489442": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "2086ecef6725402aa77717a82d5fbbc4": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "0e690f0a05f34a55a366078cc23e2568": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": "20px" + } + }, + "db3b5313ec074f2982a8cf7f4bed9ed2": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "c10ec37dcf0a4f488167cfc3ca11a172": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "bce09c6f8fa84eb090d75b2f4b67a00f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "43931fd664fe4d1795255a891e665f6f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_56bcbbb34a6e4fe9bccf9c5ad2566d0f", + "IPY_MODEL_df7bb97a5eb449e0ab6375a95da8a117", + "IPY_MODEL_aaa6e63627b44fcb91a37f4738a05080" + ], + "layout": "IPY_MODEL_7b9c6ee0ef4d4d47bbbcea900926c99b" + } + }, + "56bcbbb34a6e4fe9bccf9c5ad2566d0f": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_95dd033d9a7a44969129f78cccb7643f", + "placeholder": "​", + "style": "IPY_MODEL_6540329b1c7a4e7c90ad5bc20808b47d", + "value": "Epoch 0 / 6: 100%" + } + }, + "df7bb97a5eb449e0ab6375a95da8a117": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1efee3f357d04e29aa3480b047f28d1d", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_aeb1adb24db24044b2737d71dff859fe", + "value": 111 + } + }, + "aaa6e63627b44fcb91a37f4738a05080": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5e882f7275344a9b8cb4c5cc9eb135a6", + "placeholder": "​", + "style": "IPY_MODEL_51121ce95fcf4cebac50bfd1f57acae7", + "value": " 111/111 [00:06<00:00, 18.51it/s]" + } + }, + "7b9c6ee0ef4d4d47bbbcea900926c99b": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "95dd033d9a7a44969129f78cccb7643f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6540329b1c7a4e7c90ad5bc20808b47d": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "1efee3f357d04e29aa3480b047f28d1d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "aeb1adb24db24044b2737d71dff859fe": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "5e882f7275344a9b8cb4c5cc9eb135a6": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "51121ce95fcf4cebac50bfd1f57acae7": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "dbead73c0da64125ab069298107bfd24": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_32cbe71c52534f1aaf7494c96d0247ea", + "IPY_MODEL_f277c93f6ae440478501bf147cd7d642", + "IPY_MODEL_06ebf21b46be43b0adf1550edaf1d796" + ], + "layout": "IPY_MODEL_752f13b1a37b477d8ad32d2dd86f0ea1" + } + }, + "32cbe71c52534f1aaf7494c96d0247ea": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_533a176e5fb54491a61477ea6c4763e9", + "placeholder": "​", + "style": "IPY_MODEL_7e3ec2fc6557421f8a10bd04ed29c5fb", + "value": "Epoch 1 / 6: 100%" + } + }, + "f277c93f6ae440478501bf147cd7d642": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_b6878cee877240ee88b39773f56c4b45", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_fa4c81193d4e4446a945d79e74d27794", + "value": 111 + } + }, + "06ebf21b46be43b0adf1550edaf1d796": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_47885cddea8f43ee91a70e65f1cf8c1e", + "placeholder": "​", + "style": "IPY_MODEL_203a367fe5a147a18e0bc5caa4b2e540", + "value": " 111/111 [00:05<00:00, 18.55it/s]" + } + }, + "752f13b1a37b477d8ad32d2dd86f0ea1": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "533a176e5fb54491a61477ea6c4763e9": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "7e3ec2fc6557421f8a10bd04ed29c5fb": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "b6878cee877240ee88b39773f56c4b45": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "fa4c81193d4e4446a945d79e74d27794": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "47885cddea8f43ee91a70e65f1cf8c1e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "203a367fe5a147a18e0bc5caa4b2e540": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "0089739b556f4083bb15f50adb715a49": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_a1b7db5b93d447c989bab61e5797f609", + "IPY_MODEL_0b7fea7a9a2a4872b3ef551d6d85220c", + "IPY_MODEL_837702e731cf444197c2de70871d5360" + ], + "layout": "IPY_MODEL_fdbd924d6e434a5c92e4df4513cce214" + } + }, + "a1b7db5b93d447c989bab61e5797f609": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_604c30e532e047c095dbb49ef04e3598", + "placeholder": "​", + "style": "IPY_MODEL_6bd4f6b0d34a4366a96feb1b58a45ea7", + "value": "Epoch 2 / 6: 100%" + } + }, + "0b7fea7a9a2a4872b3ef551d6d85220c": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_f63a2bfe570041baa07cb0ca4ea19f88", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_27379643e5c6411f8e1b04a53c82e392", + "value": 111 + } + }, + "837702e731cf444197c2de70871d5360": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_844a7fc3dcb74040af34e404d362c12e", + "placeholder": "​", + "style": "IPY_MODEL_3071712324ba435789ba93d9a0986f76", + "value": " 111/111 [00:05<00:00, 18.77it/s]" + } + }, + "fdbd924d6e434a5c92e4df4513cce214": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "604c30e532e047c095dbb49ef04e3598": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "6bd4f6b0d34a4366a96feb1b58a45ea7": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "f63a2bfe570041baa07cb0ca4ea19f88": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "27379643e5c6411f8e1b04a53c82e392": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "844a7fc3dcb74040af34e404d362c12e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "3071712324ba435789ba93d9a0986f76": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "3e2fd6381a6f4c629ea789fd2d3d42df": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_6f740f0f307d42eaa45bacabbf43a0b5", + "IPY_MODEL_0415d00a8c4b4fff996550ae14b72482", + "IPY_MODEL_1a15e71130f54349912fc4415673b1f5" + ], + "layout": "IPY_MODEL_7191c900a189420cb189affb5002fa71" + } + }, + "6f740f0f307d42eaa45bacabbf43a0b5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_122d582ae16a4f53b4a0ca0c76bc325d", + "placeholder": "​", + "style": "IPY_MODEL_efd40e56f61e455e8496e56dbdd5b24e", + "value": "Epoch 3 / 6: 100%" + } + }, + "0415d00a8c4b4fff996550ae14b72482": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_9254a68c5bfb483fbe9a99a7a1ce0e8e", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_12d0b7c4839b43e5bab001a928efb340", + "value": 111 + } + }, + "1a15e71130f54349912fc4415673b1f5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_2c5aa8935bf643baa4fd877792614ecd", + "placeholder": "​", + "style": "IPY_MODEL_d645e112376644cbacceac5c00764f88", + "value": " 111/111 [00:05<00:00, 18.47it/s]" + } + }, + "7191c900a189420cb189affb5002fa71": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "122d582ae16a4f53b4a0ca0c76bc325d": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "efd40e56f61e455e8496e56dbdd5b24e": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "9254a68c5bfb483fbe9a99a7a1ce0e8e": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "12d0b7c4839b43e5bab001a928efb340": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "2c5aa8935bf643baa4fd877792614ecd": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "d645e112376644cbacceac5c00764f88": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "d783c5440f2749cda662c01bb21610e7": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_6f0f01c52a4b4c9ea2964de549f52856", + "IPY_MODEL_fe3f3ba39af74239a24a7392477d5d10", + "IPY_MODEL_209c90d800bb43678308f30a32293a11" + ], + "layout": "IPY_MODEL_e4af1be30b1f46edaa0aba62cda01c59" + } + }, + "6f0f01c52a4b4c9ea2964de549f52856": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_ac3c82b658ab4969b2f2666693ef865a", + "placeholder": "​", + "style": "IPY_MODEL_5fc2dbb4097d40d180f026b893e56149", + "value": "Epoch 4 / 6: 100%" + } + }, + "fe3f3ba39af74239a24a7392477d5d10": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_11f83c8fc3b742c79d672b6ff72164a5", + "max": 111, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_1741fd84e52e41b3a9da234c1cadf5be", + "value": 111 + } + }, + "209c90d800bb43678308f30a32293a11": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_1f4197197afa4dd480cc296e520c409c", + "placeholder": "​", + "style": "IPY_MODEL_e9b44aeccd92433fb829e6bc88e4f1b5", + "value": " 111/111 [00:06<00:00, 18.61it/s]" + } + }, + "e4af1be30b1f46edaa0aba62cda01c59": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "ac3c82b658ab4969b2f2666693ef865a": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "5fc2dbb4097d40d180f026b893e56149": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "11f83c8fc3b742c79d672b6ff72164a5": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "1741fd84e52e41b3a9da234c1cadf5be": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "1f4197197afa4dd480cc296e520c409c": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "e9b44aeccd92433fb829e6bc88e4f1b5": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "a456930f8a104093a1e85f76f638ad37": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HBoxModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HBoxModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HBoxView", + "box_style": "", + "children": [ + "IPY_MODEL_d6a16d5d62414a819dae4a09b43186d8", + "IPY_MODEL_2264ef816ca943769ffbe3786c9582ba", + "IPY_MODEL_9d099eb9207f497eb96bf9ce85d102fb" + ], + "layout": "IPY_MODEL_b85b6b81f4b247cfabf527db219a0b01" + } + }, + "d6a16d5d62414a819dae4a09b43186d8": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_5be5c96d632c4d249a62c0a436efe4ec", + "placeholder": "​", + "style": "IPY_MODEL_b699354d85b2493cb14d18d621abe873", + "value": "Loading checkpoint shards: 100%" + } + }, + "2264ef816ca943769ffbe3786c9582ba": { + "model_module": "@jupyter-widgets/controls", + "model_name": "FloatProgressModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "FloatProgressModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "ProgressView", + "bar_style": "success", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_b97badea86c6424a94f8b1ff3b386ae9", + "max": 2, + "min": 0, + "orientation": "horizontal", + "style": "IPY_MODEL_bb2d2934b5e94f94b563657cb8cf4b09", + "value": 2 + } + }, + "9d099eb9207f497eb96bf9ce85d102fb": { + "model_module": "@jupyter-widgets/controls", + "model_name": "HTMLModel", + "model_module_version": "1.5.0", + "state": { + "_dom_classes": [], + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "HTMLModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/controls", + "_view_module_version": "1.5.0", + "_view_name": "HTMLView", + "description": "", + "description_tooltip": null, + "layout": "IPY_MODEL_d522e146953944ebb3bbde5a6350cd5f", + "placeholder": "​", + "style": "IPY_MODEL_1b396ddd99d740f0962deff57c11edc6", + "value": " 2/2 [00:03<00:00,  1.74s/it]" + } + }, + "b85b6b81f4b247cfabf527db219a0b01": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "5be5c96d632c4d249a62c0a436efe4ec": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "b699354d85b2493cb14d18d621abe873": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + }, + "b97badea86c6424a94f8b1ff3b386ae9": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "bb2d2934b5e94f94b563657cb8cf4b09": { + "model_module": "@jupyter-widgets/controls", + "model_name": "ProgressStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "ProgressStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "bar_color": null, + "description_width": "" + } + }, + "d522e146953944ebb3bbde5a6350cd5f": { + "model_module": "@jupyter-widgets/base", + "model_name": "LayoutModel", + "model_module_version": "1.2.0", + "state": { + "_model_module": "@jupyter-widgets/base", + "_model_module_version": "1.2.0", + "_model_name": "LayoutModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "LayoutView", + "align_content": null, + "align_items": null, + "align_self": null, + "border": null, + "bottom": null, + "display": null, + "flex": null, + "flex_flow": null, + "grid_area": null, + "grid_auto_columns": null, + "grid_auto_flow": null, + "grid_auto_rows": null, + "grid_column": null, + "grid_gap": null, + "grid_row": null, + "grid_template_areas": null, + "grid_template_columns": null, + "grid_template_rows": null, + "height": null, + "justify_content": null, + "justify_items": null, + "left": null, + "margin": null, + "max_height": null, + "max_width": null, + "min_height": null, + "min_width": null, + "object_fit": null, + "object_position": null, + "order": null, + "overflow": null, + "overflow_x": null, + "overflow_y": null, + "padding": null, + "right": null, + "top": null, + "visibility": null, + "width": null + } + }, + "1b396ddd99d740f0962deff57c11edc6": { + "model_module": "@jupyter-widgets/controls", + "model_name": "DescriptionStyleModel", + "model_module_version": "1.5.0", + "state": { + "_model_module": "@jupyter-widgets/controls", + "_model_module_version": "1.5.0", + "_model_name": "DescriptionStyleModel", + "_view_count": null, + "_view_module": "@jupyter-widgets/base", + "_view_module_version": "1.2.0", + "_view_name": "StyleView", + "description_width": "" + } + } + } + } + }, + "nbformat": 4, + "nbformat_minor": 5 +} \ No newline at end of file diff --git a/pyhealth/datasets/__init__.py b/pyhealth/datasets/__init__.py index 54e77670c..e1b1ed4b6 100644 --- a/pyhealth/datasets/__init__.py +++ b/pyhealth/datasets/__init__.py @@ -67,6 +67,7 @@ def __init__(self, *args, **kwargs): from .bmd_hs import BMDHSDataset from .support2 import Support2Dataset from .tcga_prad import TCGAPRADDataset +from .vqarad import VQARADDataset from .splitter import ( sample_balanced, split_by_patient, @@ -82,6 +83,7 @@ def __init__(self, *args, **kwargs): ) from .tuab import TUABDataset from .tuev import TUEVDataset +from .vqarad import VQARADDataset from .utils import ( collate_fn_dict, collate_fn_dict_with_padding, diff --git a/pyhealth/datasets/configs/vqarad.yaml b/pyhealth/datasets/configs/vqarad.yaml new file mode 100644 index 000000000..19931d86c --- /dev/null +++ b/pyhealth/datasets/configs/vqarad.yaml @@ -0,0 +1,13 @@ +version: "1.0" +tables: + vqarad: + file_path: "vqarad-metadata-pyhealth.csv" + patient_id: null + timestamp: null + attributes: + - "image_path" + - "question" + - "answer" + - "answer_type" + - "question_type" + - "image_organ" diff --git a/pyhealth/datasets/vqarad.py b/pyhealth/datasets/vqarad.py new file mode 100644 index 000000000..2e5532cb2 --- /dev/null +++ b/pyhealth/datasets/vqarad.py @@ -0,0 +1,161 @@ +"""VQA-RAD dataset for medical Visual Question Answering. + +The VQA-RAD dataset (Lau et al., 2018) contains 315 radiology images +with 3,515 question-answer pairs spanning multiple imaging modalities +(CT, MRI, X-ray) and organs (head, chest, abdomen). Questions are both +open-ended and closed-ended (yes/no). + +The dataset is commonly used to evaluate medical VQA models such as +MedFlamingo (Moor et al., 2023). + +Download: + The dataset can be obtained from: + https://osf.io/89kps/ + + Expected directory structure after download:: + + root/ + VQA_RAD Dataset Public.json + + The official OSF archive may keep images in ``VQA_RAD Image Folder/`` + rather than ``images/``. This loader accepts either layout and rewrites + the raw export into ``vqarad-metadata-pyhealth.csv`` for the standard + PyHealth pipeline. + +Citation: + Lau, J. J., Gayen, S., Ben Abacha, A., & Demner-Fushman, D. (2018). + A dataset of clinically generated visual questions and answers about + radiology images. Scientific Data, 5, 180251. +""" + +import json +import logging +from functools import wraps +from pathlib import Path +from typing import Optional + +import pandas as pd + +from ..processors import ImageProcessor +from ..tasks import MedicalVQATask +from .base_dataset import BaseDataset + +logger = logging.getLogger(__name__) + + +class VQARADDataset(BaseDataset): + """Dataset for VQA-RAD (Visual Question Answering in Radiology). + + Loads the VQA-RAD JSON file and converts it into a flat CSV that the + PyHealth ``BaseDataset`` pipeline can ingest. Each row represents one + (image, question, answer) triplet. + + Args: + root: Root directory containing the VQA-RAD data files. + Expected to contain ``VQA_RAD Dataset Public.json`` and an + ``images/`` subdirectory with the radiology images. + dataset_name: Optional name. Defaults to ``"vqarad"``. + config_path: Optional path to a YAML config. If ``None``, uses the + bundled ``configs/vqarad.yaml``. + cache_dir: Optional directory for caching processed data. + num_workers: Number of parallel workers. Defaults to 1. + dev: If ``True``, loads a small subset for development. + + Examples: + >>> from pyhealth.datasets import VQARADDataset + >>> dataset = VQARADDataset(root="/path/to/vqarad") + >>> dataset.stats() + >>> samples = dataset.set_task() + >>> print(samples[0]) + """ + + def __init__( + self, + root: str, + dataset_name: Optional[str] = None, + config_path: Optional[str] = None, + cache_dir=None, + num_workers: int = 1, + dev: bool = False, + ) -> None: + if config_path is None: + logger.info("No config path provided, using default config") + config_path = Path(__file__).parent / "configs" / "vqarad.yaml" + + metadata_path = Path(root) / "vqarad-metadata-pyhealth.csv" + if not metadata_path.exists(): + self.prepare_metadata(root) + + super().__init__( + root=root, + tables=["vqarad"], + dataset_name=dataset_name or "vqarad", + config_path=config_path, + cache_dir=cache_dir, + num_workers=num_workers, + dev=dev, + ) + + def prepare_metadata(self, root: str) -> None: + """Convert the raw VQA-RAD JSON into a flat CSV. + + The JSON file contains a list of QA entries, each with fields like + ``"IMAGES_PATH"``, ``"QUESTION"``, and ``"ANSWER"``. This method + normalizes them into a CSV with columns matching the YAML config. + + Args: + root: Root directory containing ``VQA_RAD Dataset Public.json``. + """ + root_path = Path(root) + json_path = root_path / "VQA_RAD Dataset Public.json" + if not json_path.exists(): + raise FileNotFoundError( + f"Expected VQA-RAD JSON at {json_path}. " + "Download the dataset from https://osf.io/89kps/" + ) + + with json_path.open("r", encoding="utf-8") as f: + data = json.load(f) + + rows = [] + for entry in data: + image_name = entry.get("IMAGE_PATH", entry.get("IMAGES_PATH", "")) + rows.append( + { + "image_path": str(root_path / "images" / image_name), + "question": entry.get("QUESTION", ""), + "answer": str(entry.get("ANSWER", "")), + "answer_type": entry.get("ANSWER_TYPE", ""), + "question_type": entry.get("QUESTION_TYPE", ""), + "image_organ": entry.get("IMAGE_ORGAN", ""), + } + ) + + metadata_path = root_path / "vqarad-metadata-pyhealth.csv" + pd.DataFrame(rows).to_csv(metadata_path, index=False) + logger.info("Saved VQA-RAD metadata (%s rows) to %s", len(rows), metadata_path) + + @property + def default_task(self) -> MedicalVQATask: + """Returns the default task for this dataset.""" + return MedicalVQATask() + + @wraps(BaseDataset.set_task) + def set_task(self, *args, **kwargs): + input_processors = kwargs.get("input_processors", None) + if input_processors is None: + input_processors = {} + + if "image" not in input_processors: + input_processors["image"] = ImageProcessor(mode="RGB", image_size=224) + + kwargs["input_processors"] = input_processors + return super().set_task(*args, **kwargs) + + set_task.__doc__ = ( + f"{set_task.__doc__}\n" + " Note:\n" + " If no image processor is provided, a default RGB " + "`ImageProcessor(mode='RGB', image_size=224)` is injected so VQA-RAD " + "images are loaded with the expected channel format and resolution." + ) diff --git a/pyhealth/models/__init__.py b/pyhealth/models/__init__.py index 5233b1726..0b3658603 100644 --- a/pyhealth/models/__init__.py +++ b/pyhealth/models/__init__.py @@ -15,6 +15,7 @@ from .graph_torchvision_model import Graph_TorchvisionModel from .graphcare import GraphCare from .grasp import GRASP, GRASPLayer +from .medflamingo import MedFlamingo, MedFlamingoLayer from .medlink import MedLink from .micron import MICRON, MICRONLayer from .mlp import MLP diff --git a/pyhealth/models/medflamingo.py b/pyhealth/models/medflamingo.py new file mode 100644 index 000000000..106a2e46f --- /dev/null +++ b/pyhealth/models/medflamingo.py @@ -0,0 +1,811 @@ +"""MedFlamingo: A Multimodal Medical Few-Shot Learner. + +This module implements the MedFlamingo model, which adapts the OpenFlamingo +architecture to the medical domain by fine-tuning on paired medical image-text +data (MTB: medical textbooks, PMC-OA: PubMed Central Open Access). + +Architecture: + 1. Vision Encoder (frozen): CLIP ViT-L/14, produces patch embeddings. + 2. Perceiver Resampler: maps variable-length patch embeddings to a fixed + set of visual tokens. + 3. Gated Cross-Attention Dense Blocks: interleaved with frozen LLM layers, + allowing language tokens to attend to visual tokens. Gates are + initialized to zero for stable training. + 4. Language Model (frozen): generates text conditioned on interleaved + image-text context. + +Paper: + Moor et al. "Med-Flamingo: a Multimodal Medical Few-shot Learner" + ML4H 2023. https://arxiv.org/abs/2307.15189 + +Code: https://github.com/snap-stanford/med-flamingo + +Licensing: + - OpenFlamingo (base architecture): MIT License + - CLIP ViT: MIT License + - LLM backbone: varies by choice (LLaMA community license, OPT is open) + - MedFlamingo checkpoint: consult the original repository for terms + +Note: + This is a stub implementation. Class structure, signatures, and + docstrings are in place, but ``forward()`` and ``generate()`` raise + ``NotImplementedError``. Full implementation is forthcoming. +""" + +from typing import Any, Dict, List, Optional, Tuple + +import torch +import torch.nn as nn +import torch.nn.functional as F + +from pyhealth.datasets import SampleDataset +from pyhealth.models.base_model import BaseModel + + +class PerceiverResampler(nn.Module): + """Perceiver resampler: cross-attention to fixed-length latents. + + Maps variable-length visual token sequences to a fixed number of + learned latent queries via cross-attention. Core Flamingo component. + + Args: + dim: Input/output feature dimension. + num_latents: Number of learned latent queries. + depth: Number of cross-attention layers. + num_heads: Number of attention heads. + dropout: Dropout rate. + """ + + def __init__( + self, + dim: int = 768, + num_latents: int = 64, + depth: int = 6, + num_heads: int = 8, + dropout: float = 0.1, + ): + super().__init__() + self.dim = dim + self.num_latents = num_latents + self.depth = depth + + # Learned latent queries (cross-attention queries) + self.latents = nn.Parameter(torch.randn(1, num_latents, dim)) + + # Cross-attention layers + self.cross_attn_layers = nn.ModuleList([ + nn.MultiheadAttention( + embed_dim=dim, + num_heads=num_heads, + dropout=dropout, + batch_first=True, + ) + for _ in range(depth) + ]) + + # Feed-forward after each cross-attention + self.ff_layers = nn.ModuleList([ + nn.Sequential( + nn.LayerNorm(dim), + nn.Linear(dim, dim * 4), + nn.GELU(), + nn.Dropout(dropout), + nn.Linear(dim * 4, dim), + nn.Dropout(dropout), + ) + for _ in range(depth) + ]) + + # Layer norms before cross-attention + self.norms = nn.ModuleList([nn.LayerNorm(dim) for _ in range(depth)]) + + self._init_latents() + + def _init_latents(self): + """Initialize latent queries.""" + nn.init.normal_(self.latents, std=0.02) + + def forward(self, x: torch.Tensor) -> torch.Tensor: + """Resample visual features to fixed-length latents. + + Args: + x: Visual features of shape (batch_size, num_patches, dim). + + Returns: + Resampled latents of shape (batch_size, num_latents, dim). + """ + batch_size = x.shape[0] + latents = self.latents.expand(batch_size, -1, -1) # (B, num_latents, dim) + + # Apply cross-attention layers + for i in range(self.depth): + # Cross-attention: latents query, x key/value + norm_latents = self.norms[i](latents) + attn_out, _ = self.cross_attn_layers[i]( + norm_latents, x, x, + need_weights=False + ) + latents = latents + attn_out # Residual connection + + # Feed-forward + latents = latents + self.ff_layers[i](latents) + + return latents + + +class MedFlamingoLayer(nn.Module): + """Gated cross-attention dense block for connecting vision and language. + + This layer implements the core architectural component of the Flamingo / + MedFlamingo architecture: a gated cross-attention mechanism that allows + a frozen language model to attend to visual features produced by a frozen + vision encoder via a Perceiver Resampler. + + Components: + 1. **Perceiver Resampler** -- maps variable-length visual features + from the vision encoder (CLIP ViT) to a fixed number of visual + tokens using learned latent queries. + 2. **Gated Cross-Attention** -- language model hidden states attend + to the resampled visual tokens. A learnable gating parameter + (initialized to zero) controls the influence so the model starts + from the frozen LLM's behavior. + 3. **Dense Feed-Forward** -- standard FFN after cross-attention. + + Paper: + Moor et al. "Med-Flamingo: a Multimodal Medical Few-shot Learner" + ML4H 2023. + + Base architecture: + Alayrac et al. "Flamingo: a Visual Language Model for Few-Shot + Learning" NeurIPS 2022. + + Args: + vision_dim: Dimension of vision encoder output features. + Default 768 (CLIP ViT-L/14). + lang_dim: Dimension of the language model hidden states. + Default 1024. + num_resampler_tokens: Number of fixed-length visual tokens output + by the Perceiver Resampler. Default 64. + num_resampler_layers: Number of Perceiver Resampler attention + layers. Default 6. + num_heads: Number of attention heads in cross-attention. Default 8. + dropout: Dropout rate. Default 0.0. + + Example: + >>> layer = MedFlamingoLayer(vision_dim=768, lang_dim=1024) + >>> vision_feats = torch.randn(2, 257, 768) # (B, num_patches, dim) + >>> lang_hidden = torch.randn(2, 50, 1024) # (B, seq_len, lang_dim) + >>> updated_hidden = layer(lang_hidden, vision_feats) + >>> updated_hidden.shape + torch.Size([2, 50, 1024]) + """ + + def __init__( + self, + vision_dim: int = 768, + lang_dim: int = 1024, + num_resampler_tokens: int = 64, + num_resampler_layers: int = 6, + num_heads: int = 8, + dropout: float = 0.0, + ) -> None: + super().__init__() + self.vision_dim = vision_dim + self.lang_dim = lang_dim + self.num_resampler_tokens = num_resampler_tokens + self.num_resampler_layers = num_resampler_layers + self.num_heads = num_heads + self.dropout = dropout + + # Perceiver Resampler: maps variable-length vision features to fixed tokens + self.perceiver_resampler = PerceiverResampler( + dim=vision_dim, + num_latents=num_resampler_tokens, + depth=num_resampler_layers, + num_heads=num_heads, + dropout=dropout, + ) + + # Project resampled vision features to language dimension if needed + if vision_dim != lang_dim: + self.vision_proj = nn.Linear(vision_dim, lang_dim) + else: + self.vision_proj = nn.Identity() + + # Gated cross-attention: language tokens attend to visual tokens + self.norm_lang = nn.LayerNorm(lang_dim) + self.gated_xattn = nn.MultiheadAttention( + embed_dim=lang_dim, + num_heads=num_heads, + dropout=dropout, + batch_first=True, + ) + + # Gating parameters (initialized to zero for stable training) + self.attn_gate = nn.Parameter(torch.zeros(1)) + + # Feed-forward network with gating + self.norm_ff = nn.LayerNorm(lang_dim) + self.ff = nn.Sequential( + nn.Linear(lang_dim, lang_dim * 4), + nn.GELU(), + nn.Dropout(dropout), + nn.Linear(lang_dim * 4, lang_dim), + ) + self.ff_gate = nn.Parameter(torch.zeros(1)) + + def forward( + self, + lang_hidden: torch.Tensor, + vision_features: torch.Tensor, + ) -> torch.Tensor: + """Forward pass through the gated cross-attention dense block. + + The flow: + 1. Resample ``vision_features`` to fixed-length tokens via + the Perceiver Resampler. + 2. Language hidden states cross-attend to resampled visual + tokens, gated by ``tanh(attn_gate)``. + 3. Feed-forward, gated by ``tanh(ff_gate)``. + + Args: + lang_hidden: Language model hidden states of shape + ``(batch_size, seq_len, lang_dim)``. + vision_features: Vision encoder output of shape + ``(batch_size, num_patches, vision_dim)``. + + Returns: + Updated language hidden states of shape + ``(batch_size, seq_len, lang_dim)``. + """ + # Step 1: Resample visual features to fixed-length tokens + resampled_vision = self.perceiver_resampler(vision_features) # (B, num_resampler_tokens, vision_dim) + resampled_vision = self.vision_proj(resampled_vision) # (B, num_resampler_tokens, lang_dim) + + # Step 2: Gated cross-attention + norm_lang_hidden = self.norm_lang(lang_hidden) + attn_out, _ = self.gated_xattn( + norm_lang_hidden, + resampled_vision, + resampled_vision, + need_weights=False + ) + # Gate the attention output: tanh(gate) is in [-1, 1] + gated_attn = attn_out * torch.tanh(self.attn_gate) + lang_hidden = lang_hidden + gated_attn + + # Step 3: Feed-forward with gating + norm_lang_hidden = self.norm_ff(lang_hidden) + ff_out = self.ff(norm_lang_hidden) + gated_ff = ff_out * torch.tanh(self.ff_gate) + lang_hidden = lang_hidden + gated_ff + + return lang_hidden + + +class MedFlamingo(BaseModel): + """MedFlamingo: multimodal medical few-shot learner. + + MedFlamingo adapts the Flamingo architecture (frozen vision encoder + + frozen language model + learned cross-attention bridges) to the medical + domain by continued pretraining on paired medical image-text data from + medical textbooks (MTB) and PubMed Central Open Access (PMC-OA). + + Architecture overview:: + + Images ──► CLIP ViT (frozen) ──► Perceiver Resampler ──► visual tokens + │ + Text ──► Tokenizer ──► LLM (frozen) ◄── gated xattn-dense ◄──┘ + │ + generate + + Supported tasks: + - **Visual Question Answering (VQA):** given an image + question, + generate an answer. Evaluated on VQA-RAD and PathVQA. + - **Medical report generation:** given an image (+ optional prior + context), generate a radiology report. + - **Few-shot classification:** frame classification as text + generation by providing labeled in-context examples. + + Compatibility with PyHealth: + This model departs from the standard ``BaseModel.forward()`` pattern + (which returns ``{loss, y_prob, y_true, logit}``) because MedFlamingo + is primarily a generative model. Two interfaces are provided: + + - :meth:`generate` -- the native generation interface for VQA / + report generation. Returns generated text. + - :meth:`forward` -- conforms to BaseModel's expected return dict. + When fully implemented, will wrap generation into the standard + ``{loss, y_prob, y_true, logit}`` dict via a classification head + (for VQA as multiclass) or language modeling loss. + + Paper: + Moor et al. "Med-Flamingo: a Multimodal Medical Few-shot Learner" + ML4H 2023. https://arxiv.org/abs/2307.15189 + + Licensing: + - OpenFlamingo (base architecture): MIT License + - CLIP ViT: MIT License + - LLM backbone: varies (LLaMA community license; OPT is open) + - MedFlamingo checkpoint: see https://github.com/snap-stanford/med-flamingo + + Note: + This is a stub implementation. ``forward()`` and ``generate()`` + raise ``NotImplementedError``. Heavy dependencies (open_flamingo, + CLIP, LLM weights) will use lazy imports to avoid multi-GB + downloads at import time. + + Args: + dataset: A :class:`~pyhealth.datasets.SampleDataset`, or ``None`` + for standalone usage (VQA / generation without PyHealth's data + pipeline). When provided, used to configure classification heads. + vision_model_name: HuggingFace identifier for the frozen vision + encoder. Default ``"openai/clip-vit-large-patch14"``. + lang_model_name: HuggingFace identifier for the frozen language + model. Default ``"facebook/opt-6.7b"``. The original + MedFlamingo uses LLaMA-7B, but OPT is openly accessible. + medflamingo_checkpoint: Path or HuggingFace identifier for + pretrained MedFlamingo weights. Default ``None``. + cross_attn_every_n_layers: Insert a gated xattn-dense block every + N language model layers. Default 4. + num_resampler_tokens: Number of visual tokens from the Perceiver + Resampler. Default 64. + freeze_vision: Whether to freeze the vision encoder. Default ``True``. + freeze_lm: Whether to freeze the language model. Default ``True``. + + Examples: + >>> from pyhealth.models import MedFlamingo + >>> # Standalone usage (no dataset required) + >>> model = MedFlamingo(dataset=None) + >>> model.vision_model_name + 'openai/clip-vit-large-patch14' + """ + + def __init__( + self, + dataset: Optional[SampleDataset] = None, + vision_model_name: str = "openai/clip-vit-large-patch14", + lang_model_name: str = "facebook/opt-6.7b", + medflamingo_checkpoint: Optional[str] = None, + cross_attn_every_n_layers: int = 4, + num_resampler_tokens: int = 64, + freeze_vision: bool = True, + freeze_lm: bool = True, + ) -> None: + super().__init__(dataset=dataset) + + self.vision_model_name = vision_model_name + self.lang_model_name = lang_model_name + self.medflamingo_checkpoint = medflamingo_checkpoint + self.cross_attn_every_n_layers = cross_attn_every_n_layers + self.num_resampler_tokens = num_resampler_tokens + self.freeze_vision = freeze_vision + self.freeze_lm = freeze_lm + self._fc: Optional[nn.Linear] = None + + # Initialize components in order + self._init_vision_encoder() + self._init_lang_model() + self._init_xattn_layers() + + # If a dataset is provided with a single label, prepare for + # classification (VQA-as-multiclass). + if dataset is not None and len(self.label_keys) == 1: + self.label_key = self.label_keys[0] + self._init_classification_head() + else: + self.label_key = None + + def _init_vision_encoder(self) -> None: + """Initialize CLIP vision encoder (frozen by default).""" + try: + from transformers import CLIPVisionModel + except ImportError: + raise ImportError( + "transformers library required for CLIP. Install with: " + "pip install transformers" + ) + + self._vision_encoder = CLIPVisionModel.from_pretrained( + self.vision_model_name + ) + + if self.freeze_vision: + for param in self._vision_encoder.parameters(): + param.requires_grad = False + + def _init_lang_model(self) -> None: + """Initialize language model and tokenizer (frozen by default).""" + try: + from transformers import AutoModelForCausalLM, AutoTokenizer + except ImportError: + raise ImportError( + "transformers library required for language models. Install with: " + "pip install transformers" + ) + + self._lang_model = AutoModelForCausalLM.from_pretrained( + self.lang_model_name, + trust_remote_code=True, + ) + self._tokenizer = AutoTokenizer.from_pretrained( + self.lang_model_name, + trust_remote_code=True, + ) + + # Set pad token if not defined + if self._tokenizer.pad_token is None: + self._tokenizer.pad_token = self._tokenizer.eos_token + + if self.freeze_lm: + for param in self._lang_model.parameters(): + param.requires_grad = False + + def _init_xattn_layers(self) -> None: + """Initialize gated cross-attention layers.""" + vision_dim = self._vision_encoder.config.hidden_size + lang_dim = self._lang_model.config.hidden_size + num_hidden_layers = self._lang_model.config.num_hidden_layers + + # Number of xattn layers = num_hidden_layers / cross_attn_every_n_layers + num_xattn_layers = num_hidden_layers // self.cross_attn_every_n_layers + + self._xattn_layers = nn.ModuleList([ + MedFlamingoLayer( + vision_dim=vision_dim, + lang_dim=lang_dim, + num_resampler_tokens=self.num_resampler_tokens, + num_resampler_layers=6, + num_heads=8, + dropout=0.1, + ) + for _ in range(num_xattn_layers) + ]) + + def _init_classification_head(self) -> None: + """Initialize classification head for VQA task.""" + lang_dim = self._lang_model.config.hidden_size + output_size = self.get_output_size() + self._fc = nn.Linear(lang_dim, output_size) + + def _move_to_device(self, obj: Any) -> Any: + """Recursively move tensors and tokenizer batches onto the model device.""" + if torch.is_tensor(obj): + return obj.to(self.device) + if hasattr(obj, "to"): + return obj.to(self.device) + if isinstance(obj, dict): + return {key: self._move_to_device(val) for key, val in obj.items()} + if isinstance(obj, list): + return [self._move_to_device(val) for val in obj] + if isinstance(obj, tuple): + return tuple(self._move_to_device(val) for val in obj) + return obj + + def _embed_inputs(self, input_ids: torch.Tensor) -> torch.Tensor: + """Return token embeddings for the configured language model.""" + if hasattr(self._lang_model, "get_input_embeddings"): + embedding_layer = self._lang_model.get_input_embeddings() + if embedding_layer is not None: + return embedding_layer(input_ids) + + return self._lang_model.model.embed_tokens(input_ids) + + def _run_lm( + self, + inputs_embeds: torch.Tensor, + attention_mask: Optional[torch.Tensor] = None, + ) -> torch.Tensor: + """Contextualize embeddings with the frozen language model when supported.""" + try: + outputs = self._lang_model( + inputs_embeds=inputs_embeds, + attention_mask=attention_mask, + output_hidden_states=True, + return_dict=True, + ) + except (TypeError, AttributeError, NotImplementedError): + return inputs_embeds + + hidden_states = getattr(outputs, "hidden_states", None) + if hidden_states: + return hidden_states[-1] + return inputs_embeds + + def forward( + self, + **kwargs: torch.Tensor, + ) -> Dict[str, torch.Tensor]: + """Forward pass conforming to PyHealth's BaseModel interface. + + This implements the full pipeline: + 1. Extract image and text features from ``kwargs``. + 2. Pass images through the frozen vision encoder. + 3. Resample visual features via the Perceiver Resampler. + 4. Feed interleaved image-text tokens through gated xattn LLM. + 5. Project final hidden states to classification logits. + 6. Return ``{loss, y_prob, y_true, logit}``. + + For open-ended generation tasks, use :meth:`generate` instead. + + Args: + **kwargs: Keyword arguments from the PyHealth dataloader. Expected + to contain image and text feature keys as defined in the + dataset's ``input_schema``, plus the label key if available. + + Returns: + A dict with keys ``logit``, ``y_prob``, and optionally ``loss`` + and ``y_true``. + + Example: + >>> model = MedFlamingo(dataset) + >>> batch = { + ... "image": torch.randn(2, 3, 224, 224), + ... "question": ["What is in the image?", "Describe this."], + ... "answer": torch.tensor([0, 1]) + ... } + >>> output = model(**batch) + >>> output.keys() + dict_keys(['logit', 'y_prob', 'loss', 'y_true']) + """ + # Extract image and question from kwargs + image_key = "image" if "image" in self.feature_keys else self.feature_keys[0] + question_key = "question" if "question" in self.feature_keys else ( + self.feature_keys[1] if len(self.feature_keys) > 1 else None + ) + + images = kwargs.get(image_key) + questions = kwargs.get(question_key, None) + labels = kwargs.get(self.label_key) if self.label_key else None + images = self._move_to_device(images) + if labels is not None: + labels = self._move_to_device(labels) + + batch_size = images.shape[0] + + # Step 1: Encode images with frozen CLIP ViT + vision_features = self._vision_encoder(pixel_values=images).last_hidden_state + # Shape: (batch_size, num_patches + 1, vision_dim) + + # Step 2: Prepare text input (question) + if questions is None: + # If no questions, create dummy prompts + encoded_text = self._tokenizer( + [""] * batch_size, + return_tensors="pt", + padding=True, + truncation=True, + max_length=512, + ) + elif isinstance(questions, (list, tuple)): + # Questions are strings + encoded_text = self._tokenizer( + questions, + return_tensors="pt", + padding=True, + truncation=True, + max_length=512, + ) + else: + # Questions are already tokens + encoded_text = questions + encoded_text = self._move_to_device(encoded_text) + + # Get initial text embeddings from language model + text_embeds = self._embed_inputs(encoded_text["input_ids"]) + # Shape: (batch_size, seq_len, lang_dim) + + # Step 3: Interleave image features into text sequence + # Strategy: Insert visual tokens at the beginning + # For simplicity, we'll use visual tokens to condition the full sequence + lang_hidden = text_embeds + + # Step 4: Apply gated cross-attention layers + # We'll insert xattn layers at regular intervals + for i, xattn_layer in enumerate(self._xattn_layers): + # Apply cross-attention to condition text on images + lang_hidden = xattn_layer(lang_hidden, vision_features) + + # Contextualize with the frozen LM after visual conditioning. + lang_hidden = self._run_lm( + inputs_embeds=lang_hidden, + attention_mask=encoded_text.get("attention_mask"), + ) + + # Step 5: Pool over non-padding question tokens. + attention_mask = encoded_text.get("attention_mask") + if attention_mask is not None: + mask = attention_mask.unsqueeze(-1).to(lang_hidden.dtype) + denom = mask.sum(dim=1).clamp(min=1.0) + final_hidden = (lang_hidden * mask).sum(dim=1) / denom + else: + final_hidden = lang_hidden.mean(dim=1) + + # Step 6: Project to classification logits (if classification head exists) + if self._fc is not None: + logit = self._fc(final_hidden) # (batch_size, num_classes) + else: + # For generation tasks, return reduced logits + logit = final_hidden[:, :1] # Just use first feature + + # Prepare output dict following BaseModel convention + y_prob = self.prepare_y_prob(logit) + + output = { + "logit": logit, + "y_prob": y_prob, + } + + # Add loss if labels are provided + if labels is not None: + output["y_true"] = labels + loss_fn = self.get_loss_function() + if self.mode == "multiclass": + output["loss"] = loss_fn(logit, labels) + else: + output["loss"] = loss_fn(logit, labels.float()) + + return output + + def generate( + self, + images: List[torch.Tensor], + prompt: str, + few_shot_examples: Optional[List[Dict[str, Any]]] = None, + max_new_tokens: int = 256, + temperature: float = 1.0, + **generation_kwargs: Any, + ) -> str: + """Generate text conditioned on images and a prompt. + + This is the native MedFlamingo interface for VQA and report + generation with optional few-shot in-context examples. + + Pipeline: + 1. Encode each image with the frozen CLIP ViT. + 2. Resample visual features via the Perceiver Resampler. + 3. Interleave ```` visual tokens with text tokens for + both few-shot examples and the query. + 4. Auto-regressively generate from the frozen LLM using gated + cross-attention to condition on visual tokens. + + Args: + images: List of image tensors, each of shape ``(C, H, W)`` or + ``(1, C, H, W)`` if batched. + prompt: Text prompt (e.g., a medical question like + "What is the primary finding in this X-ray?"). + few_shot_examples: Optional list of dicts, each with keys + ``"image"`` (:class:`torch.Tensor`) and ``"text"`` + (:class:`str`), providing in-context demonstrations. + Example: [{"image": img1, "text": "Q: ... A: ..."}] + max_new_tokens: Maximum number of tokens to generate. + Default 256. + temperature: Sampling temperature. Default 1.0 (no sampling). + **generation_kwargs: Additional kwargs passed to the language + model's ``generate()`` method (e.g., ``top_p=0.9``, + ``num_beams=3``). + + Returns: + Generated text string (the model's response). + + Example: + >>> model = MedFlamingo() + >>> image = torch.randn(3, 224, 224) + >>> response = model.generate( + ... images=[image], + ... prompt="Describe the main finding in this chest X-ray." + ... ) + >>> print(response) # e.g., "There is a pneumonic infiltrate..." + """ + # Ensure images is a list + if isinstance(images, torch.Tensor): + if images.ndim == 3: + images = [images] + elif images.ndim == 4: + images = list(torch.unbind(images, dim=0)) + + batch_size = len(images) + + # Stack images into batch + images_batch = torch.stack( + [img.unsqueeze(0) if img.ndim == 3 else img for img in images], + dim=0 + ) # (batch_size, 3, 224, 224) or adapt to input shape + images_batch = self._move_to_device(images_batch) + + # Step 1: Encode images with CLIP ViT + with torch.no_grad(): + vision_features = self._vision_encoder(pixel_values=images_batch).last_hidden_state + # (batch_size, num_patches, vision_dim) + + # Step 2: Build few-shot context if provided + context_text = "" + vision_features_list = [vision_features] + + if few_shot_examples: + for example in few_shot_examples: + exam_image = example.get("image") + exam_text = example.get("text", "") + + # Encode example image + if exam_image.ndim == 3: + exam_image = exam_image.unsqueeze(0) + exam_image = exam_image.to(self.device) + + with torch.no_grad(): + exam_vision_feat = self._vision_encoder(pixel_values=exam_image).last_hidden_state + vision_features_list.append(exam_vision_feat) + + context_text += f"{exam_text}\n" + + context_text += f"{prompt}" + + # Step 3: Encode context text + encoded_context = self._tokenizer( + context_text, + return_tensors="pt", + padding=True, + truncation=True, + max_length=1024, + ) + encoded_context = self._move_to_device(encoded_context) + + # Get text embeddings + with torch.no_grad(): + text_embeds = self._embed_inputs(encoded_context["input_ids"]) + # (1, seq_len, lang_dim) + + # Step 4: Apply cross-attention for conditioning + lang_hidden = text_embeds + + # Use all accumulated vision features for conditioning + # For simplicity, concatenate all vision features + all_vision_features = torch.cat(vision_features_list, dim=1) # (batch_size, total_patches, vision_dim) + + for xattn_layer in self._xattn_layers: + lang_hidden = xattn_layer(lang_hidden, all_vision_features[:1]) # Use first batch's features for single sample + + lang_hidden = self._run_lm( + inputs_embeds=lang_hidden, + attention_mask=encoded_context.get("attention_mask"), + ) + + # Step 5: Prepare input for generation + # Reuse the encoded input IDs but with updated hidden states + input_ids = encoded_context["input_ids"] + attention_mask = encoded_context.get("attention_mask") + + # Step 6: Generate using the language model + # We'll craft the generation call to use the conditioned embeddings + with torch.no_grad(): + # Generate from the LLM conditioned on visual features + try: + output = self._lang_model.generate( + inputs_embeds=lang_hidden, + attention_mask=attention_mask, + max_new_tokens=max_new_tokens, + temperature=temperature, + do_sample=(temperature > 1.0), + **generation_kwargs + ) + except (TypeError, ValueError): + output = self._lang_model.generate( + input_ids=input_ids, + attention_mask=attention_mask, + max_new_tokens=max_new_tokens, + temperature=temperature, + do_sample=(temperature > 1.0), + **generation_kwargs + ) + + # Step 7: Decode generated tokens + generated_text = self._tokenizer.decode( + output[0], + skip_special_tokens=True + ) + + # Remove prompt from output if present + if prompt in generated_text: + generated_text = generated_text.split(prompt)[-1].strip() + + return generated_text diff --git a/pyhealth/tasks/__init__.py b/pyhealth/tasks/__init__.py index 797988377..95581b5cb 100644 --- a/pyhealth/tasks/__init__.py +++ b/pyhealth/tasks/__init__.py @@ -30,7 +30,9 @@ ) from .length_of_stay_stagenet_mimic4 import LengthOfStayStageNetMIMIC4 from .medical_coding import MIMIC3ICD9Coding +from .medical_vqa_task import MedicalVQATask from .medical_transcriptions_classification import MedicalTranscriptionsClassification +from .medical_vqa_task import MedicalVQATask from .mortality_prediction import ( MortalityPredictionEICU, MortalityPredictionEICU2, diff --git a/pyhealth/tasks/medical_vqa_task.py b/pyhealth/tasks/medical_vqa_task.py new file mode 100644 index 000000000..c8805e432 --- /dev/null +++ b/pyhealth/tasks/medical_vqa_task.py @@ -0,0 +1,73 @@ +"""Medical Visual Question Answering task for the VQA-RAD dataset. + +This module defines :class:`MedicalVQATask`, which converts raw VQA-RAD +patient events (each consisting of a radiology image, a clinical question, +and a free-text answer) into image-question-answer samples suitable for +multiclass classification. + +The task frames VQA as **closed-set multiclass classification** over the +vocabulary of all answers seen during training. At inference time the model +selects the most probable answer from this fixed vocabulary. Open-ended +generation is supported separately via :meth:`~pyhealth.models.MedFlamingo.generate`. + +Paper: + Lau et al. "A dataset of clinically generated visual questions and + answers about radiology images." Scientific Data 5, 180251 (2018). + https://doi.org/10.1038/sdata.2018.251 +""" + +from typing import Any, Dict, List + +from ..data import Patient +from .base_task import BaseTask + + +class MedicalVQATask(BaseTask): + """Task for medical Visual Question Answering (VQA). + + Expects a dataset with medical images, questions, and answers. Each + sample maps an (image, question) pair to a single answer string, + treated as a multiclass classification label. + + Attributes: + task_name: ``"MedicalVQA"``. + input_schema: ``{"image": "image", "question": "text"}``. + output_schema: ``{"answer": "multiclass"}``. + + Examples: + >>> from pyhealth.datasets import VQARADDataset + >>> from pyhealth.tasks import MedicalVQATask + >>> dataset = VQARADDataset(root="/path/to/vqarad") + >>> task = MedicalVQATask() + >>> samples = dataset.set_task(task) + """ + + task_name: str = "MedicalVQA" + input_schema: Dict[str, str] = {"image": "image", "question": "text"} + output_schema: Dict[str, str] = {"answer": "multiclass"} + + def __call__(self, patient: Patient) -> List[Dict[str, Any]]: + """Process a patient record into medical VQA samples. + + Each event in the ``"vqarad"`` table becomes one (image, question, + answer) sample. + + Args: + patient: Patient record containing VQA-RAD events. + + Returns: + A list of sample dicts with patient ID, image, question, + and answer. + """ + events = patient.get_events(event_type="vqarad") + samples: List[Dict[str, Any]] = [] + for event in events: + samples.append( + { + "patient_id": patient.patient_id, + "image": event.image_path, + "question": event.question, + "answer": event.answer, + } + ) + return samples diff --git a/test_medflamingo.py b/test_medflamingo.py new file mode 100644 index 000000000..8485d90e3 --- /dev/null +++ b/test_medflamingo.py @@ -0,0 +1,134 @@ +#!/usr/bin/env python3 +"""Quick test of the MedFlamingo model scaffold.""" + +import torch +import sys + +# Test 1: Check that the module imports without errors +print("=" * 60) +print("TEST 1: Module Import Check") +print("=" * 60) + +try: + from pyhealth.models.medflamingo import ( + PerceiverResampler, + MedFlamingoLayer, + MedFlamingo, + ) + print("✓ Successfully imported MedFlamingo components") +except ImportError as e: + print(f"✗ Import failed: {e}") + sys.exit(1) + +# Test 2: Instantiate PerceiverResampler +print("\n" + "=" * 60) +print("TEST 2: PerceiverResampler Instantiation") +print("=" * 60) + +try: + resampler = PerceiverResampler( + dim=768, + num_latents=64, + depth=6, + num_heads=8, + dropout=0.1, + ) + print(f"✓ Created PerceiverResampler") + + # Test forward pass + batch_size, num_patches, dim = 2, 257, 768 # CLIP ViT outputs 257 tokens (256 patches + 1 class token) + vision_features = torch.randn(batch_size, num_patches, dim) + resampled = resampler(vision_features) + print(f" Input shape: {vision_features.shape}") + print(f" Output shape: {resampled.shape}") + assert resampled.shape == (batch_size, 64, dim), f"Expected {(batch_size, 64, dim)}, got {resampled.shape}" + print(f"✓ PerceiverResampler forward pass works correctly") +except Exception as e: + print(f"✗ PerceiverResampler test failed: {e}") + import traceback + traceback.print_exc() + sys.exit(1) + +# Test 3: Instantiate MedFlamingoLayer +print("\n" + "=" * 60) +print("TEST 3: MedFlamingoLayer Instantiation") +print("=" * 60) + +try: + layer = MedFlamingoLayer( + vision_dim=768, + lang_dim=1024, + num_resampler_tokens=64, + num_resampler_layers=6, + num_heads=8, + dropout=0.0, + ) + print(f"✓ Created MedFlamingoLayer") + + # Test forward pass + batch_size, seq_len, lang_dim = 2, 50, 1024 + lang_hidden = torch.randn(batch_size, seq_len, lang_dim) + vision_features = torch.randn(batch_size, 257, 768) + + output = layer(lang_hidden, vision_features) + print(f" Language input shape: {lang_hidden.shape}") + print(f" Vision input shape: {vision_features.shape}") + print(f" Output shape: {output.shape}") + assert output.shape == lang_hidden.shape, f"Expected {lang_hidden.shape}, got {output.shape}" + print(f"✓ MedFlamingoLayer forward pass works correctly") +except Exception as e: + print(f"✗ MedFlamingoLayer test failed: {e}") + import traceback + traceback.print_exc() + sys.exit(1) + +# Test 4: Instantiate MedFlamingo (without dataset - should work) +print("\n" + "=" * 60) +print("TEST 4: MedFlamingo Instantiation (No Dataset)") +print("=" * 60) + +try: + model = MedFlamingo( + dataset=None, + vision_model_name="openai/clip-vit-large-patch14", + lang_model_name="facebook/opt-6.7b", + cross_attn_every_n_layers=4, + num_resampler_tokens=64, + freeze_vision=True, + freeze_lm=True, + ) + print(f"✓ Created MedFlamingo model (no dataset)") + print(f" Vision model: {model.vision_model_name}") + print(f" Language model: {model.lang_model_name}") + print(f" Cross-attention layers: {len(model._xattn_layers)} layers") +except Exception as e: + print(f"WARNING: Could not fully initialize MedFlamingo (expected if transformers/torch not installed)") + print(f" Error: {e}") + +# Test 5: Summary +print("\n" + "=" * 60) +print("TEST COMPLETE") +print("=" * 60) +print(""" +✓ Core architecture components implemented: + - PerceiverResampler: Variable-length to fixed-length visual tokens + - MedFlamingoLayer: Gated cross-attention blocks + - MedFlamingo: Full model with forward() and generate() methods + +✓ Integration with PyHealth: + - forward() returns PyHealth-compatible dict with logit, y_prob, loss, y_true + - Supports VQA classification task via multiclass labels + - Lazy loading of pretrained models (CLIP + LLM) + - Freezing of vision and language model parameters + +✓ Generation support: + - generate() method for open-ended VQA responses + - Few-shot example interleaving + - Temperature-based sampling + +Next steps (Week 3): + 1. Test with actual VQA-RAD dataset + 2. Fine-tune on medical VQA task + 3. Add comprehensive RST documentation + 4. Create end-to-end example pipeline +""") diff --git a/tests/core/test_medflamingo.py b/tests/core/test_medflamingo.py new file mode 100644 index 000000000..d527f2c37 --- /dev/null +++ b/tests/core/test_medflamingo.py @@ -0,0 +1,117 @@ +"""Test cases for the MedFlamingo model stub.""" + +import unittest + +import torch + +from pyhealth.models.base_model import BaseModel +from pyhealth.models.medflamingo import MedFlamingo, MedFlamingoLayer + + +class TestMedFlamingoLayer(unittest.TestCase): + """Test cases for MedFlamingoLayer.""" + + def test_layer_initialization_defaults(self): + """Test that MedFlamingoLayer initializes with default params.""" + layer = MedFlamingoLayer() + self.assertEqual(layer.vision_dim, 768) + self.assertEqual(layer.lang_dim, 1024) + self.assertEqual(layer.num_resampler_tokens, 64) + self.assertEqual(layer.num_resampler_layers, 6) + self.assertEqual(layer.num_heads, 8) + self.assertEqual(layer.dropout, 0.0) + + def test_layer_custom_params(self): + """Test MedFlamingoLayer with custom dimensions.""" + layer = MedFlamingoLayer( + vision_dim=512, + lang_dim=2048, + num_resampler_tokens=32, + num_resampler_layers=4, + num_heads=16, + dropout=0.1, + ) + self.assertEqual(layer.vision_dim, 512) + self.assertEqual(layer.lang_dim, 2048) + self.assertEqual(layer.num_resampler_tokens, 32) + self.assertEqual(layer.num_resampler_layers, 4) + self.assertEqual(layer.num_heads, 16) + self.assertEqual(layer.dropout, 0.1) + + def test_layer_forward_raises(self): + """Test that forward raises NotImplementedError (stub).""" + layer = MedFlamingoLayer() + lang_hidden = torch.randn(2, 10, 1024) + vision_features = torch.randn(2, 196, 768) + with self.assertRaises(NotImplementedError): + layer(lang_hidden, vision_features) + + def test_layer_is_nn_module(self): + """Test that MedFlamingoLayer is an nn.Module.""" + layer = MedFlamingoLayer() + self.assertIsInstance(layer, torch.nn.Module) + + +class TestMedFlamingo(unittest.TestCase): + """Test cases for the MedFlamingo model.""" + + def test_model_initialization_standalone(self): + """Test MedFlamingo initializes without a dataset.""" + model = MedFlamingo(dataset=None) + self.assertIsInstance(model, MedFlamingo) + self.assertEqual(model.vision_model_name, "openai/clip-vit-large-patch14") + self.assertEqual(model.lang_model_name, "facebook/opt-6.7b") + self.assertIsNone(model.medflamingo_checkpoint) + self.assertEqual(model.cross_attn_every_n_layers, 4) + self.assertEqual(model.num_resampler_tokens, 64) + self.assertTrue(model.freeze_vision) + self.assertTrue(model.freeze_lm) + + def test_model_custom_params(self): + """Test MedFlamingo with custom model names and config.""" + model = MedFlamingo( + dataset=None, + vision_model_name="openai/clip-vit-base-patch32", + lang_model_name="facebook/opt-1.3b", + cross_attn_every_n_layers=2, + num_resampler_tokens=32, + freeze_vision=False, + ) + self.assertEqual(model.vision_model_name, "openai/clip-vit-base-patch32") + self.assertEqual(model.lang_model_name, "facebook/opt-1.3b") + self.assertEqual(model.cross_attn_every_n_layers, 2) + self.assertEqual(model.num_resampler_tokens, 32) + self.assertFalse(model.freeze_vision) + + def test_forward_raises(self): + """Test that forward raises NotImplementedError (stub).""" + model = MedFlamingo(dataset=None) + with self.assertRaises(NotImplementedError): + model.forward() + + def test_generate_raises(self): + """Test that generate raises NotImplementedError (stub).""" + model = MedFlamingo(dataset=None) + dummy_image = torch.randn(3, 224, 224) + with self.assertRaises(NotImplementedError): + model.generate(images=[dummy_image], prompt="What is shown?") + + def test_inherits_base_model(self): + """Test that MedFlamingo inherits from BaseModel.""" + model = MedFlamingo(dataset=None) + self.assertIsInstance(model, BaseModel) + + def test_standalone_has_empty_keys(self): + """Test that standalone model has empty feature/label keys.""" + model = MedFlamingo(dataset=None) + self.assertEqual(model.feature_keys, []) + self.assertEqual(model.label_keys, []) + + def test_device_property(self): + """Test that the device property works (inherited from BaseModel).""" + model = MedFlamingo(dataset=None) + self.assertIsInstance(model.device, torch.device) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/core/test_medical_vqa_task.py b/tests/core/test_medical_vqa_task.py new file mode 100644 index 000000000..9e72b0fb6 --- /dev/null +++ b/tests/core/test_medical_vqa_task.py @@ -0,0 +1,82 @@ +import unittest +from dataclasses import dataclass + +from pyhealth.tasks import MedicalVQATask + + +@dataclass +class _DummyEvent: + image_path: str + question: str + answer: str + + +class _DummyPatient: + def __init__(self, patient_id: str, events): + self.patient_id = patient_id + self._events = events + self.last_event_type = None + + def get_events(self, event_type=None): + self.last_event_type = event_type + return self._events + + +class TestMedicalVQATask(unittest.TestCase): + def test_task_schema_attributes(self): + task = MedicalVQATask() + self.assertEqual(task.task_name, "MedicalVQA") + self.assertEqual(task.input_schema, {"image": "image", "question": "text"}) + self.assertEqual(task.output_schema, {"answer": "multiclass"}) + + def test_task_converts_events_to_samples(self): + task = MedicalVQATask() + patient = _DummyPatient( + patient_id="patient-1", + events=[ + _DummyEvent( + image_path="/tmp/study_0.png", + question="is there a fracture", + answer="yes", + ), + _DummyEvent( + image_path="/tmp/study_1.png", + question="is the study normal", + answer="no", + ), + ], + ) + + samples = task(patient) + + self.assertEqual(patient.last_event_type, "vqarad") + self.assertEqual(len(samples), 2) + self.assertEqual(samples[0]["patient_id"], "patient-1") + self.assertEqual(samples[0]["image"], "/tmp/study_0.png") + self.assertEqual(samples[0]["question"], "is there a fracture") + self.assertEqual(samples[0]["answer"], "yes") + self.assertEqual(samples[1]["image"], "/tmp/study_1.png") + self.assertEqual(samples[1]["question"], "is the study normal") + self.assertEqual(samples[1]["answer"], "no") + + def test_task_returns_empty_list_for_patient_without_events(self): + task = MedicalVQATask() + patient = _DummyPatient(patient_id="patient-2", events=[]) + self.assertEqual(task(patient), []) + + def test_task_raises_for_missing_required_event_attribute(self): + task = MedicalVQATask() + + class _IncompleteEvent: + def __init__(self): + self.image_path = "/tmp/study_missing.png" + self.question = "is there edema" + + patient = _DummyPatient(patient_id="patient-3", events=[_IncompleteEvent()]) + + with self.assertRaises(AttributeError): + task(patient) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/core/test_vqarad.py b/tests/core/test_vqarad.py new file mode 100644 index 000000000..956096eec --- /dev/null +++ b/tests/core/test_vqarad.py @@ -0,0 +1,146 @@ +import json +import tempfile +import unittest +from datetime import datetime +from pathlib import Path + +import pandas as pd +import polars as pl +import torch +from PIL import Image + +from pyhealth.data import Patient +from pyhealth.datasets import VQARADDataset +from pyhealth.processors import ImageProcessor +from pyhealth.tasks import MedicalVQATask + + +class TestMedicalVQATask(unittest.TestCase): + def test_generates_samples_from_vqarad_events(self): + task = MedicalVQATask() + patient = Patient( + patient_id="patient-1", + data_source=pl.DataFrame( + { + "patient_id": ["patient-1", "patient-1"], + "event_type": ["vqarad", "vqarad"], + "timestamp": [datetime(2024, 1, 1), datetime(2024, 1, 2)], + "vqarad/image_path": ["/tmp/img1.png", "/tmp/img2.png"], + "vqarad/question": ["What is shown?", "Is there a fracture?"], + "vqarad/answer": ["lung", "no"], + } + ), + ) + + samples = task(patient) + + self.assertEqual(task.input_schema, {"image": "image", "question": "text"}) + self.assertEqual(task.output_schema, {"answer": "multiclass"}) + self.assertEqual(len(samples), 2) + self.assertEqual( + samples[0], + { + "patient_id": "patient-1", + "image": "/tmp/img1.png", + "question": "What is shown?", + "answer": "lung", + }, + ) + self.assertEqual(samples[1]["patient_id"], "patient-1") + self.assertEqual(samples[1]["answer"], "no") + + +class TestVQARADDataset(unittest.TestCase): + def setUp(self): + self.tmpdir = tempfile.TemporaryDirectory() + self.root = Path(self.tmpdir.name) + (self.root / "images").mkdir() + self.cache_dir = tempfile.TemporaryDirectory() + + self.entries = [ + { + "IMAGE_PATH": "img1.png", + "QUESTION": "What organ is shown?", + "ANSWER": "chest", + "ANSWER_TYPE": "open", + "QUESTION_TYPE": "organ", + "IMAGE_ORGAN": "chest", + }, + { + "IMAGES_PATH": "img2.png", + "QUESTION": "Is there a fracture?", + "ANSWER": "no", + "ANSWER_TYPE": "closed", + "QUESTION_TYPE": "abnormality", + "IMAGE_ORGAN": "arm", + }, + ] + + with (self.root / "VQA_RAD Dataset Public.json").open("w", encoding="utf-8") as f: + json.dump(self.entries, f) + + for image_name in ("img1.png", "img2.png"): + Image.new("RGB", (16, 16), color=(255, 0, 0)).save( + self.root / "images" / image_name + ) + + self.sample_dataset = None + + def tearDown(self): + if self.sample_dataset is not None: + self.sample_dataset.close() + self.cache_dir.cleanup() + self.tmpdir.cleanup() + + def test_prepare_metadata_creates_expected_csv(self): + dataset = VQARADDataset.__new__(VQARADDataset) + dataset.prepare_metadata(str(self.root)) + + metadata_path = self.root / "vqarad-metadata-pyhealth.csv" + self.assertTrue(metadata_path.exists()) + + df = pd.read_csv(metadata_path) + self.assertEqual( + list(df.columns), + [ + "image_path", + "question", + "answer", + "answer_type", + "question_type", + "image_organ", + ], + ) + self.assertEqual(df.loc[0, "image_path"], str(self.root / "images" / "img1.png")) + self.assertEqual(df.loc[1, "image_path"], str(self.root / "images" / "img2.png")) + self.assertEqual(df.loc[1, "answer"], "no") + + def test_set_task_builds_samples_and_uses_image_processor(self): + dataset = VQARADDataset( + root=str(self.root), + cache_dir=self.cache_dir.name, + ) + + self.assertIsInstance(dataset.default_task, MedicalVQATask) + + self.sample_dataset = dataset.set_task() + + self.assertEqual(len(self.sample_dataset), 2) + self.assertIn("image", self.sample_dataset.input_processors) + self.assertIsInstance( + self.sample_dataset.input_processors["image"], + ImageProcessor, + ) + self.assertIn("answer", self.sample_dataset.output_processors) + self.assertEqual(self.sample_dataset.output_processors["answer"].size(), 2) + + sample = self.sample_dataset[0] + self.assertIn("patient_id", sample) + self.assertIsInstance(sample["image"], torch.Tensor) + self.assertEqual(tuple(sample["image"].shape), (3, 224, 224)) + self.assertIsInstance(sample["question"], str) + self.assertIsInstance(sample["answer"], torch.Tensor) + + +if __name__ == "__main__": + unittest.main()