Skip to content

Instantly share code, notes, and snippets.

@minesh1291

minesh1291/vae_pytorch_lightning.ipynb

Created May 16, 2024 04:42
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save minesh1291/6ec9220e533210875eadf4c28f886c59 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save minesh1291/6ec9220e533210875eadf4c28f886c59 to your computer and use it in GitHub Desktop.
Download ZIP
Vae_pytorch_lightning.ipynb
Display the source blob
Display the rendered blob
Raw
vae_pytorch_lightning.ipynb
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": [],
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"accelerator": "GPU",
"widgets": {
"application/vnd.jupyter.widget-state+json": {
"3a869e786a044d87b3b36dbb6de6fa3a": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
"_model_name": "HBoxModel",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.5.0",
"box_style": "",
"layout": "IPY_MODEL_07be3131e53d4d1bbdd0979b4aed0668",
"_model_module": "@jupyter-widgets/controls",
"children": [
"IPY_MODEL_95ea9a25490a441baf656ed93c3e487f",
"IPY_MODEL_967093369a8a4437bc886d63c00a3629",
"IPY_MODEL_3645a6e6b9df4a34827174c923f8ef25"
]
}
},
"07be3131e53d4d1bbdd0979b4aed0668": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"95ea9a25490a441baf656ed93c3e487f": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_be294bf266e04ad0b30a4ec187685b4f",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": "",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_0e706c9f844147518da2708a0477fbfe"
}
},
"967093369a8a4437bc886d63c00a3629": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_4f0bc0e9a480485c874d702a903a0dfa",
"_dom_classes": [],
"description": "",
"_model_name": "FloatProgressModel",
"bar_style": "success",
"max": 26421880,
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": 26421880,
"_view_count": null,
"_view_module_version": "1.5.0",
"orientation": "horizontal",
"min": 0,
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_fb19a3fe8bb641aca5f80bfe4aa29a31"
}
},
"3645a6e6b9df4a34827174c923f8ef25": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_aaf66168e8f34b96a41a51e8259c38e1",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": " 26422272/? [00:00<00:00, 76247285.51it/s]",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_95e16954d44f4b4dbc60ef27e82bba67"
}
},
"be294bf266e04ad0b30a4ec187685b4f": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"0e706c9f844147518da2708a0477fbfe": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"4f0bc0e9a480485c874d702a903a0dfa": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"bar_color": null,
"_model_module": "@jupyter-widgets/controls"
}
},
"fb19a3fe8bb641aca5f80bfe4aa29a31": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"aaf66168e8f34b96a41a51e8259c38e1": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"95e16954d44f4b4dbc60ef27e82bba67": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"0be97dc9dca646c08f3fddf4edf34088": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
"_model_name": "HBoxModel",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.5.0",
"box_style": "",
"layout": "IPY_MODEL_f668666b1adb49cca3c8901fe902b4a6",
"_model_module": "@jupyter-widgets/controls",
"children": [
"IPY_MODEL_f1123689e7d54c1780bb23c237dc7ca3",
"IPY_MODEL_04046ff4c28c47ab95e0c4ece681233f",
"IPY_MODEL_d75b8340c6ed4bb7b11671e4185f65d8"
]
}
},
"f668666b1adb49cca3c8901fe902b4a6": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"f1123689e7d54c1780bb23c237dc7ca3": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_6098ff89ab8849429148d7bda98601d3",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": "",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_115d7534710a401dba3397fdf8a14e49"
}
},
"04046ff4c28c47ab95e0c4ece681233f": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_f35e2e98c9534c8f9e8d64656ba6932d",
"_dom_classes": [],
"description": "",
"_model_name": "FloatProgressModel",
"bar_style": "success",
"max": 29515,
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": 29515,
"_view_count": null,
"_view_module_version": "1.5.0",
"orientation": "horizontal",
"min": 0,
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_1d6c6c854ed14748b9c34b84fccc36ea"
}
},
"d75b8340c6ed4bb7b11671e4185f65d8": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_878937450c3641d888dcd832592fe81b",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": " 29696/? [00:00<00:00, 836202.61it/s]",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_92ccd4d95daa4d9599385dbed79a5441"
}
},
"6098ff89ab8849429148d7bda98601d3": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"115d7534710a401dba3397fdf8a14e49": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"f35e2e98c9534c8f9e8d64656ba6932d": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"bar_color": null,
"_model_module": "@jupyter-widgets/controls"
}
},
"1d6c6c854ed14748b9c34b84fccc36ea": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"878937450c3641d888dcd832592fe81b": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"92ccd4d95daa4d9599385dbed79a5441": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"320fe6172e09496e9999d57825720d73": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
"_model_name": "HBoxModel",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.5.0",
"box_style": "",
"layout": "IPY_MODEL_c20fe7fa92f84ccead8de8d2c24ffce5",
"_model_module": "@jupyter-widgets/controls",
"children": [
"IPY_MODEL_0ab18fc2351f468b8fd93f55ca4e916c",
"IPY_MODEL_15861fde52144d1d974ec9998bcbf4c3",
"IPY_MODEL_fcc739ddd0f941e4b3ef4b99a2dcd24f"
]
}
},
"c20fe7fa92f84ccead8de8d2c24ffce5": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"0ab18fc2351f468b8fd93f55ca4e916c": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_a24c1c20326c416b849fa7e9a4f396f3",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": "",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_2e33e9e3298d4e2485027a47c0abf878"
}
},
"15861fde52144d1d974ec9998bcbf4c3": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_76057466dadc421e86cfd5b515b06683",
"_dom_classes": [],
"description": "",
"_model_name": "FloatProgressModel",
"bar_style": "success",
"max": 4422102,
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": 4422102,
"_view_count": null,
"_view_module_version": "1.5.0",
"orientation": "horizontal",
"min": 0,
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_7b8f4e59f378422e9920a0b187db6de8"
}
},
"fcc739ddd0f941e4b3ef4b99a2dcd24f": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_d895f6e440024420aea1ee3f82ba2d04",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": " 4422656/? [00:00<00:00, 11209926.75it/s]",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_766e87e35d7c41b9aea8bc07990e5efa"
}
},
"a24c1c20326c416b849fa7e9a4f396f3": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"2e33e9e3298d4e2485027a47c0abf878": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"76057466dadc421e86cfd5b515b06683": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"bar_color": null,
"_model_module": "@jupyter-widgets/controls"
}
},
"7b8f4e59f378422e9920a0b187db6de8": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"d895f6e440024420aea1ee3f82ba2d04": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"766e87e35d7c41b9aea8bc07990e5efa": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"134afa401a8340649a5832b99a6ff660": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
"_model_name": "HBoxModel",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.5.0",
"box_style": "",
"layout": "IPY_MODEL_4b70cb0caf7b418586e644fef192c3cd",
"_model_module": "@jupyter-widgets/controls",
"children": [
"IPY_MODEL_9796f7de3c1449dc9c872b99cf1627d8",
"IPY_MODEL_75657d518d7d4cadaf1d31720341d03f",
"IPY_MODEL_357bc89aa9924d5f841c9694ea110818"
]
}
},
"4b70cb0caf7b418586e644fef192c3cd": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"9796f7de3c1449dc9c872b99cf1627d8": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_f922a08e96944525b5c82196a4827bfc",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": "",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_63cb9bfc03ae4d09aeae6d88efb0892b"
}
},
"75657d518d7d4cadaf1d31720341d03f": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_b6793d4c1d8f4461bcafb632106ae7d4",
"_dom_classes": [],
"description": "",
"_model_name": "FloatProgressModel",
"bar_style": "success",
"max": 5148,
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": 5148,
"_view_count": null,
"_view_module_version": "1.5.0",
"orientation": "horizontal",
"min": 0,
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_7b7753a780c6495da8bcdffe5853998b"
}
},
"357bc89aa9924d5f841c9694ea110818": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_e437fad5e40940c9928c52683786297d",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": " 6144/? [00:00<00:00, 164413.25it/s]",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_8d1070ad01ec440fbb8f7fb51920785d"
}
},
"f922a08e96944525b5c82196a4827bfc": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"63cb9bfc03ae4d09aeae6d88efb0892b": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"b6793d4c1d8f4461bcafb632106ae7d4": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"bar_color": null,
"_model_module": "@jupyter-widgets/controls"
}
},
"7b7753a780c6495da8bcdffe5853998b": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"e437fad5e40940c9928c52683786297d": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"8d1070ad01ec440fbb8f7fb51920785d": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"95477025b64742b88e58a718ca82dfe4": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HBoxModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HBoxView",
"_dom_classes": [],
"_model_name": "HBoxModel",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.5.0",
"box_style": "",
"layout": "IPY_MODEL_2785a8dc03064947be961552f474adb9",
"_model_module": "@jupyter-widgets/controls",
"children": [
"IPY_MODEL_c849673b610142e8a59d72e4d6d6d7ae",
"IPY_MODEL_e1472fff59a04d20b2618247014c6f2f",
"IPY_MODEL_03ee862493e0461bb410001cfca620bd"
]
}
},
"2785a8dc03064947be961552f474adb9": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": "row wrap",
"width": "100%",
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": "inline-flex",
"left": null
}
},
"c849673b610142e8a59d72e4d6d6d7ae": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_12d857003bd044eaa1cfec6778e5bc40",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": "Epoch 29: 100%",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_6b6229713e114f6895792b2e28d1a4de"
}
},
"e1472fff59a04d20b2618247014c6f2f": {
"model_module": "@jupyter-widgets/controls",
"model_name": "FloatProgressModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "ProgressView",
"style": "IPY_MODEL_6a30744037784040b892d14e7ebac107",
"_dom_classes": [],
"description": "",
"_model_name": "FloatProgressModel",
"bar_style": "success",
"max": 938,
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": 938,
"_view_count": null,
"_view_module_version": "1.5.0",
"orientation": "horizontal",
"min": 0,
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_148d934d8a1d4aca9fb41f902dc8e95c"
}
},
"03ee862493e0461bb410001cfca620bd": {
"model_module": "@jupyter-widgets/controls",
"model_name": "HTMLModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "HTMLView",
"style": "IPY_MODEL_ea81698552844ca7a579d290ac8bb47b",
"_dom_classes": [],
"description": "",
"_model_name": "HTMLModel",
"placeholder": "​",
"_view_module": "@jupyter-widgets/controls",
"_model_module_version": "1.5.0",
"value": " 938/938 [00:13<00:00, 68.05it/s, loss=0.492, v_num=0]",
"_view_count": null,
"_view_module_version": "1.5.0",
"description_tooltip": null,
"_model_module": "@jupyter-widgets/controls",
"layout": "IPY_MODEL_9362eff94d93488cad70eab9006866f7"
}
},
"12d857003bd044eaa1cfec6778e5bc40": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"6b6229713e114f6895792b2e28d1a4de": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"6a30744037784040b892d14e7ebac107": {
"model_module": "@jupyter-widgets/controls",
"model_name": "ProgressStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "ProgressStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"bar_color": null,
"_model_module": "@jupyter-widgets/controls"
}
},
"148d934d8a1d4aca9fb41f902dc8e95c": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": "2",
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
},
"ea81698552844ca7a579d290ac8bb47b": {
"model_module": "@jupyter-widgets/controls",
"model_name": "DescriptionStyleModel",
"model_module_version": "1.5.0",
"state": {
"_view_name": "StyleView",
"_model_name": "DescriptionStyleModel",
"description_width": "",
"_view_module": "@jupyter-widgets/base",
"_model_module_version": "1.5.0",
"_view_count": null,
"_view_module_version": "1.2.0",
"_model_module": "@jupyter-widgets/controls"
}
},
"9362eff94d93488cad70eab9006866f7": {
"model_module": "@jupyter-widgets/base",
"model_name": "LayoutModel",
"model_module_version": "1.2.0",
"state": {
"_view_name": "LayoutView",
"grid_template_rows": null,
"right": null,
"justify_content": null,
"_view_module": "@jupyter-widgets/base",
"overflow": null,
"_model_module_version": "1.2.0",
"_view_count": null,
"flex_flow": null,
"width": null,
"min_width": null,
"border": null,
"align_items": null,
"bottom": null,
"_model_module": "@jupyter-widgets/base",
"top": null,
"grid_column": null,
"overflow_y": null,
"overflow_x": null,
"grid_auto_flow": null,
"grid_area": null,
"grid_template_columns": null,
"flex": null,
"_model_name": "LayoutModel",
"justify_items": null,
"grid_row": null,
"max_height": null,
"align_content": null,
"visibility": null,
"align_self": null,
"height": null,
"min_height": null,
"padding": null,
"grid_auto_rows": null,
"grid_gap": null,
"max_width": null,
"order": null,
"_view_module_version": "1.2.0",
"grid_template_areas": null,
"object_position": null,
"object_fit": null,
"grid_auto_columns": null,
"margin": null,
"display": null,
"left": null
}
}
}
}
},
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/gist/minesh1291/6ec9220e533210875eadf4c28f886c59/vae_pytorch_lightning.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "code",
"metadata": {
"id": "ivhtHC97xXog"
},
"source": [
"## \"Auto-Encoding Variational Bayes\" https://arxiv.org/pdf/1312.6114.pdf"
],
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "1IQLPNTeQ-YD",
"colab": {
"base_uri": "https://localhost:8080/"
},
"outputId": "892acd1a-5dfe-4961-89a0-969d8b0b5493"
},
"source": [
"! pip install pytorch-lightning --quiet\n",
"import pytorch_lightning as pl\n",
"import torch\n",
"from torch import nn\n",
"from torch.nn import functional as F\n",
"import torchvision.transforms as transforms\n",
"from torchvision import datasets\n",
"from torch.utils.data import DataLoader"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"\u001b[?25l\r\u001b[K |▋ | 10 kB 28.7 MB/s eta 0:00:01\r\u001b[K |█▎ | 20 kB 30.8 MB/s eta 0:00:01\r\u001b[K |█▉ | 30 kB 33.7 MB/s eta 0:00:01\r\u001b[K |██▌ | 40 kB 36.2 MB/s eta 0:00:01\r\u001b[K |███▏ | 51 kB 36.9 MB/s eta 0:00:01\r\u001b[K |███▊ | 61 kB 32.1 MB/s eta 0:00:01\r\u001b[K |████▍ | 71 kB 18.4 MB/s eta 0:00:01\r\u001b[K |█████ | 81 kB 19.9 MB/s eta 0:00:01\r\u001b[K |█████▋ | 92 kB 15.9 MB/s eta 0:00:01\r\u001b[K |██████▎ | 102 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████▉ | 112 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████▌ | 122 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████ | 133 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████▊ | 143 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████▍ | 153 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████ | 163 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████▋ | 174 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████▏ | 184 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████▉ | 194 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████▌ | 204 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████ | 215 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████▊ | 225 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████▎ | 235 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████ | 245 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████▋ | 256 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████▏ | 266 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████▉ | 276 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████▍ | 286 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████████ | 296 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████████▊ | 307 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████████▎ | 317 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████████ | 327 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████████▌ | 337 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████████▏ | 348 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████████▉ | 358 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████████████▍ | 368 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████████████ | 378 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████████████▋ | 389 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████████████▎ | 399 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████████████ | 409 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████████████▌ | 419 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████████████████▏ | 430 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████████████████▊ | 440 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████████████████▍ | 450 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████████████████ | 460 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████████████████▋ | 471 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████████████████▎ | 481 kB 17.1 MB/s eta 0:00:01\r\u001b[K |█████████████████████████████▉ | 491 kB 17.1 MB/s eta 0:00:01\r\u001b[K |██████████████████████████████▌ | 501 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████████████████████▏| 512 kB 17.1 MB/s eta 0:00:01\r\u001b[K |███████████████████████████████▊| 522 kB 17.1 MB/s eta 0:00:01\r\u001b[K |████████████████████████████████| 526 kB 17.1 MB/s \n",
"\u001b[K |████████████████████████████████| 396 kB 54.9 MB/s \n",
"\u001b[K |████████████████████████████████| 596 kB 52.0 MB/s \n",
"\u001b[K |████████████████████████████████| 133 kB 64.2 MB/s \n",
"\u001b[K |████████████████████████████████| 829 kB 52.9 MB/s \n",
"\u001b[K |████████████████████████████████| 1.1 MB 47.9 MB/s \n",
"\u001b[K |████████████████████████████████| 271 kB 53.3 MB/s \n",
"\u001b[K |████████████████████████████████| 192 kB 48.7 MB/s \n",
"\u001b[K |████████████████████████████████| 160 kB 52.2 MB/s \n",
"\u001b[?25h Building wheel for future (setup.py) ... \u001b[?25l\u001b[?25hdone\n"
]
}
]
},
{
"cell_type": "code",
"metadata": {
"id": "Orjvsz6wQCo7"
},
"source": [
"class VAE(pl.LightningModule):\n",
" def __init__(self,enc_out_dim=256, latent_dim=16, input_height=784):\n",
" super(VAE, self).__init__()\n",
" self.save_hyperparameters()\n",
" self.encoder = nn.Sequential(nn.Linear(input_height, enc_out_dim), nn.ReLU())\n",
" self.decoder = nn.Sequential(nn.Linear(latent_dim, enc_out_dim), nn.ReLU(), nn.Linear(enc_out_dim, input_height))\n",
" self.fc_mu=nn.Linear(enc_out_dim, latent_dim)\n",
" self.fc_log_var=nn.Linear(enc_out_dim, latent_dim)\n",
"\n",
" def encode(self,x):\n",
" z = self.encoder(x)\n",
" mu = self.fc_mu(z)\n",
" log_var = self.fc_log_var(z)\n",
" return mu,log_var\n",
"\n",
" def sampling(self,mu, log_var):\n",
" std = torch.exp(log_var / 2)\n",
" q = torch.distributions.Normal(mu, std)\n",
" z = q.rsample()\n",
" return z\n",
"\n",
" def decode(self,z):\n",
" reconst=self.decoder(z)\n",
" return torch.sigmoid(reconst)\n",
"\n",
" def loss(self,recons,x, mu, logvar):\n",
" bce = F.binary_cross_entropy(recons, x)\n",
" kld = -0.5 * torch.sum(1 + logvar - mu.pow(2) - logvar.exp())\n",
" return bce + kld\n",
"\n",
" def training_step(self, batch, batch_idx):\n",
" x, _ = batch\n",
" x= x.view(-1,self.hparams.input_height)\n",
" mu,log_var = self.encode(x)\n",
" z = self.sampling(mu, log_var)\n",
"\n",
" recons = self.decode(z)\n",
"\n",
" vae_loss=self.loss(recons,x,mu,log_var)\n",
" self.log_dict({'vae_loss': vae_loss.mean()})\n",
" return vae_loss\n",
"\n",
" def configure_optimizers(self):\n",
" return torch.optim.Adam(self.parameters(), lr=1e-4)"
],
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "jdgNBDBmVD4H",
"colab": {
"base_uri": "https://localhost:8080/",
"height": 439,
"referenced_widgets": [
"3a869e786a044d87b3b36dbb6de6fa3a",
"07be3131e53d4d1bbdd0979b4aed0668",
"95ea9a25490a441baf656ed93c3e487f",
"967093369a8a4437bc886d63c00a3629",
"3645a6e6b9df4a34827174c923f8ef25",
"be294bf266e04ad0b30a4ec187685b4f",
"0e706c9f844147518da2708a0477fbfe",
"4f0bc0e9a480485c874d702a903a0dfa",
"fb19a3fe8bb641aca5f80bfe4aa29a31",
"aaf66168e8f34b96a41a51e8259c38e1",
"95e16954d44f4b4dbc60ef27e82bba67",
"0be97dc9dca646c08f3fddf4edf34088",
"f668666b1adb49cca3c8901fe902b4a6",
"f1123689e7d54c1780bb23c237dc7ca3",
"04046ff4c28c47ab95e0c4ece681233f",
"d75b8340c6ed4bb7b11671e4185f65d8",
"6098ff89ab8849429148d7bda98601d3",
"115d7534710a401dba3397fdf8a14e49",
"f35e2e98c9534c8f9e8d64656ba6932d",
"1d6c6c854ed14748b9c34b84fccc36ea",
"878937450c3641d888dcd832592fe81b",
"92ccd4d95daa4d9599385dbed79a5441",
"320fe6172e09496e9999d57825720d73",
"c20fe7fa92f84ccead8de8d2c24ffce5",
"0ab18fc2351f468b8fd93f55ca4e916c",
"15861fde52144d1d974ec9998bcbf4c3",
"fcc739ddd0f941e4b3ef4b99a2dcd24f",
"a24c1c20326c416b849fa7e9a4f396f3",
"2e33e9e3298d4e2485027a47c0abf878",
"76057466dadc421e86cfd5b515b06683",
"7b8f4e59f378422e9920a0b187db6de8",
"d895f6e440024420aea1ee3f82ba2d04",
"766e87e35d7c41b9aea8bc07990e5efa",
"134afa401a8340649a5832b99a6ff660",
"4b70cb0caf7b418586e644fef192c3cd",
"9796f7de3c1449dc9c872b99cf1627d8",
"75657d518d7d4cadaf1d31720341d03f",
"357bc89aa9924d5f841c9694ea110818",
"f922a08e96944525b5c82196a4827bfc",
"63cb9bfc03ae4d09aeae6d88efb0892b",
"b6793d4c1d8f4461bcafb632106ae7d4",
"7b7753a780c6495da8bcdffe5853998b",
"e437fad5e40940c9928c52683786297d",
"8d1070ad01ec440fbb8f7fb51920785d"
]
},
"outputId": "f6b27501-3430-40fe-f7a5-c3a0a67a109b"
},
"source": [
"# train data\n",
"train_data = datasets.FashionMNIST(root='../input/data',train=True,download=True,transform=transforms.ToTensor())\n",
"\n",
"# training and validation data loaders\n",
"train_loader = DataLoader(train_data,batch_size=64,shuffle=True)"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gz\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gz to ../input/data/FashionMNIST/raw/train-images-idx3-ubyte.gz\n"
]
},
{
"output_type": "display_data",
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "3a869e786a044d87b3b36dbb6de6fa3a",
"version_minor": 0,
"version_major": 2
},
"text/plain": [
" 0%| | 0/26421880 [00:00<?, ?it/s]"
]
},
"metadata": {}
},
{
"output_type": "stream",
"name": "stdout",
"text": [
"Extracting ../input/data/FashionMNIST/raw/train-images-idx3-ubyte.gz to ../input/data/FashionMNIST/raw\n",
"\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-labels-idx1-ubyte.gz\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-labels-idx1-ubyte.gz to ../input/data/FashionMNIST/raw/train-labels-idx1-ubyte.gz\n"
]
},
{
"output_type": "display_data",
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "0be97dc9dca646c08f3fddf4edf34088",
"version_minor": 0,
"version_major": 2
},
"text/plain": [
" 0%| | 0/29515 [00:00<?, ?it/s]"
]
},
"metadata": {}
},
{
"output_type": "stream",
"name": "stdout",
"text": [
"Extracting ../input/data/FashionMNIST/raw/train-labels-idx1-ubyte.gz to ../input/data/FashionMNIST/raw\n",
"\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-images-idx3-ubyte.gz\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-images-idx3-ubyte.gz to ../input/data/FashionMNIST/raw/t10k-images-idx3-ubyte.gz\n"
]
},
{
"output_type": "display_data",
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "320fe6172e09496e9999d57825720d73",
"version_minor": 0,
"version_major": 2
},
"text/plain": [
" 0%| | 0/4422102 [00:00<?, ?it/s]"
]
},
"metadata": {}
},
{
"output_type": "stream",
"name": "stdout",
"text": [
"Extracting ../input/data/FashionMNIST/raw/t10k-images-idx3-ubyte.gz to ../input/data/FashionMNIST/raw\n",
"\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-labels-idx1-ubyte.gz\n",
"Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-labels-idx1-ubyte.gz to ../input/data/FashionMNIST/raw/t10k-labels-idx1-ubyte.gz\n"
]
},
{
"output_type": "display_data",
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "134afa401a8340649a5832b99a6ff660",
"version_minor": 0,
"version_major": 2
},
"text/plain": [
" 0%| | 0/5148 [00:00<?, ?it/s]"
]
},
"metadata": {}
},
{
"output_type": "stream",
"name": "stdout",
"text": [
"Extracting ../input/data/FashionMNIST/raw/t10k-labels-idx1-ubyte.gz to ../input/data/FashionMNIST/raw\n",
"\n"
]
}
]
},
{
"cell_type": "code",
"metadata": {
"id": "O6ZKHm0vUvYM",
"colab": {
"base_uri": "https://localhost:8080/",
"height": 399,
"referenced_widgets": [
"95477025b64742b88e58a718ca82dfe4",
"2785a8dc03064947be961552f474adb9",
"c849673b610142e8a59d72e4d6d6d7ae",
"e1472fff59a04d20b2618247014c6f2f",
"03ee862493e0461bb410001cfca620bd",
"12d857003bd044eaa1cfec6778e5bc40",
"6b6229713e114f6895792b2e28d1a4de",
"6a30744037784040b892d14e7ebac107",
"148d934d8a1d4aca9fb41f902dc8e95c",
"ea81698552844ca7a579d290ac8bb47b",
"9362eff94d93488cad70eab9006866f7"
]
},
"outputId": "e8c8197f-576f-43eb-89bc-c79544fd0f3b"
},
"source": [
"vae = VAE()\n",
"trainer = pl.Trainer(gpus=1, max_epochs=30, progress_bar_refresh_rate=25)\n",
"trainer.fit(vae, train_loader)"
],
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stderr",
"text": [
"/usr/local/lib/python3.7/dist-packages/pytorch_lightning/trainer/connectors/callback_connector.py:91: LightningDeprecationWarning: Setting `Trainer(progress_bar_refresh_rate=25)` is deprecated in v1.5 and will be removed in v1.7. Please pass `pytorch_lightning.callbacks.progress.TQDMProgressBar` with `refresh_rate` directly to the Trainer's `callbacks` argument instead. Or, to disable the progress bar pass `enable_progress_bar = False` to the Trainer.\n",
" f\"Setting `Trainer(progress_bar_refresh_rate={progress_bar_refresh_rate})` is deprecated in v1.5 and\"\n",
"GPU available: True, used: True\n",
"TPU available: False, using: 0 TPU cores\n",
"IPU available: False, using: 0 IPUs\n",
"LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]\n",
"\n",
" | Name | Type | Params\n",
"------------------------------------------\n",
"0 | encoder | Sequential | 200 K \n",
"1 | decoder | Sequential | 205 K \n",
"2 | fc_mu | Linear | 4.1 K \n",
"3 | fc_log_var | Linear | 4.1 K \n",
"------------------------------------------\n",
"415 K Trainable params\n",
"0 Non-trainable params\n",
"415 K Total params\n",
"1.660 Total estimated model params size (MB)\n"
]
},
{
"output_type": "display_data",
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "95477025b64742b88e58a718ca82dfe4",
"version_minor": 0,
"version_major": 2
},
"text/plain": [
"Training: 0it [00:00, ?it/s]"
]
},
"metadata": {}
}
]
}
]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment