ch03_2_3_학습률_스케줄링_적용.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "machine_shape": "hm",
      "gpuType": "T4",
      "authorship_tag": "ABX9TyPvDE1fA7OB2WARabgoLhVR",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU",
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "2e681bf0b9e24b8f80d6116a2d274660": {
          "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_28dba86a0f714b44b99f98049e870c90",
              "IPY_MODEL_95fe96f3a27a4babb1ebfdbe7d438947",
              "IPY_MODEL_1c3e86fc780a4a70a621be1b231327a6"
            ],
            "layout": "IPY_MODEL_6b0e915a14e24039ae20959cce8f2361"
          }
        },
        "28dba86a0f714b44b99f98049e870c90": {
          "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_fe1acc6d003740e18182b998b40ce1a5",
            "placeholder": "​",
            "style": "IPY_MODEL_4bf6bd1ba2b545bebd6afd3d951a11b4",
            "value": "Downloading builder script: 100%"
          }
        },
        "95fe96f3a27a4babb1ebfdbe7d438947": {
          "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_cec22bf3b1e14c9aa7cf07ca557bfc80",
            "max": 3981,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_440089cd56384a16a6bdf58aa275b9ff",
            "value": 3981
          }
        },
        "1c3e86fc780a4a70a621be1b231327a6": {
          "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_9d940bf040624a28a1e64d69884604eb",
            "placeholder": "​",
            "style": "IPY_MODEL_12db35e33e774767b2c652c2c2904162",
            "value": " 3.98k/3.98k [00:00&lt;00:00, 305kB/s]"
          }
        },
        "6b0e915a14e24039ae20959cce8f2361": {
          "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
          }
        },
        "fe1acc6d003740e18182b998b40ce1a5": {
          "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
          }
        },
        "4bf6bd1ba2b545bebd6afd3d951a11b4": {
          "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": ""
          }
        },
        "cec22bf3b1e14c9aa7cf07ca557bfc80": {
          "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
          }
        },
        "440089cd56384a16a6bdf58aa275b9ff": {
          "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": ""
          }
        },
        "9d940bf040624a28a1e64d69884604eb": {
          "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
          }
        },
        "12db35e33e774767b2c652c2c2904162": {
          "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": ""
          }
        },
        "75575b7ec7d04a318833fa7e94de32b9": {
          "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_22b87c9c11eb441fae3ca586d8026451",
              "IPY_MODEL_933c4b13661641ce902e348fa366c450",
              "IPY_MODEL_1d44095165f64f2b9d17c99050ec4647"
            ],
            "layout": "IPY_MODEL_1f735c881ef8403885c2b12ab564276f"
          }
        },
        "22b87c9c11eb441fae3ca586d8026451": {
          "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_b5eb9c734a504f09b2785f3bce2f6a51",
            "placeholder": "​",
            "style": "IPY_MODEL_c4f9b243aff34f1fb7efbf7410297191",
            "value": "Downloading readme: 100%"
          }
        },
        "933c4b13661641ce902e348fa366c450": {
          "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_5ff2496902634221a01d2aa70a2bd50c",
            "max": 6825,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_4f6d494d591c4928bb99f006587be5d7",
            "value": 6825
          }
        },
        "1d44095165f64f2b9d17c99050ec4647": {
          "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_b3a87b66d6314fb4889bb49b4e909e31",
            "placeholder": "​",
            "style": "IPY_MODEL_e84f7e31112a476b83bd86e4edc5b2b4",
            "value": " 6.83k/6.83k [00:00&lt;00:00, 490kB/s]"
          }
        },
        "1f735c881ef8403885c2b12ab564276f": {
          "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
          }
        },
        "b5eb9c734a504f09b2785f3bce2f6a51": {
          "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
          }
        },
        "c4f9b243aff34f1fb7efbf7410297191": {
          "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": ""
          }
        },
        "5ff2496902634221a01d2aa70a2bd50c": {
          "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
          }
        },
        "4f6d494d591c4928bb99f006587be5d7": {
          "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": ""
          }
        },
        "b3a87b66d6314fb4889bb49b4e909e31": {
          "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
          }
        },
        "e84f7e31112a476b83bd86e4edc5b2b4": {
          "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": ""
          }
        },
        "0586c707ed1848dba080e457cfa23a0a": {
          "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_3db7b35d2b29417ea027772a853040db",
              "IPY_MODEL_fb7b608d55524911911951b240f77cbd",
              "IPY_MODEL_1aeeb4ba924a4d2092286376cb8bc92c"
            ],
            "layout": "IPY_MODEL_224e21e22dfb4301ae1288ff3aa04b3f"
          }
        },
        "3db7b35d2b29417ea027772a853040db": {
          "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_492e0a80d555461fadf425c535f0d44e",
            "placeholder": "​",
            "style": "IPY_MODEL_8feb6bc1a02b478bb68e1617c9660d08",
            "value": "Downloading data: 100%"
          }
        },
        "fb7b608d55524911911951b240f77cbd": {
          "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_9f0cb08d9da743ecb11c5aa19294c92f",
            "max": 9912422,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_c0f3c0fc815741c8b198ea64dc5a16a6",
            "value": 9912422
          }
        },
        "1aeeb4ba924a4d2092286376cb8bc92c": {
          "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_a1845d30ad484f81841579c381536395",
            "placeholder": "​",
            "style": "IPY_MODEL_3c70b2e5471e41229e215a31ea2826b4",
            "value": " 9.91M/9.91M [00:00&lt;00:00, 19.7MB/s]"
          }
        },
        "224e21e22dfb4301ae1288ff3aa04b3f": {
          "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
          }
        },
        "492e0a80d555461fadf425c535f0d44e": {
          "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
          }
        },
        "8feb6bc1a02b478bb68e1617c9660d08": {
          "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": ""
          }
        },
        "9f0cb08d9da743ecb11c5aa19294c92f": {
          "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
          }
        },
        "c0f3c0fc815741c8b198ea64dc5a16a6": {
          "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": ""
          }
        },
        "a1845d30ad484f81841579c381536395": {
          "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
          }
        },
        "3c70b2e5471e41229e215a31ea2826b4": {
          "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": ""
          }
        },
        "f6178b580da04707b03163d0914ee73f": {
          "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_2f246525418649feac1938ed87294d7d",
              "IPY_MODEL_217749a6e4994b98a88f3663304d64f5",
              "IPY_MODEL_800284a99d344164bbc72071d777c9b6"
            ],
            "layout": "IPY_MODEL_ec2f8f3dacaa449092bfb4a4d3b7cbef"
          }
        },
        "2f246525418649feac1938ed87294d7d": {
          "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_3d1b7002bb4e46d8920d02cf0e9851e0",
            "placeholder": "​",
            "style": "IPY_MODEL_8b55aafb81964224afebd5f4df76814f",
            "value": "Downloading data: 100%"
          }
        },
        "217749a6e4994b98a88f3663304d64f5": {
          "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_bfea677be7b14abf870ec2f838697e5e",
            "max": 28881,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_71101171f0284d98ac94035204f24bfb",
            "value": 28881
          }
        },
        "800284a99d344164bbc72071d777c9b6": {
          "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_2348ee0b6b4a4f16b0fe52e032fc0915",
            "placeholder": "​",
            "style": "IPY_MODEL_64f1882a7e5843d7943baa2d84c65709",
            "value": " 28.9k/28.9k [00:00&lt;00:00, 1.86MB/s]"
          }
        },
        "ec2f8f3dacaa449092bfb4a4d3b7cbef": {
          "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
          }
        },
        "3d1b7002bb4e46d8920d02cf0e9851e0": {
          "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
          }
        },
        "8b55aafb81964224afebd5f4df76814f": {
          "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": ""
          }
        },
        "bfea677be7b14abf870ec2f838697e5e": {
          "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
          }
        },
        "71101171f0284d98ac94035204f24bfb": {
          "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": ""
          }
        },
        "2348ee0b6b4a4f16b0fe52e032fc0915": {
          "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
          }
        },
        "64f1882a7e5843d7943baa2d84c65709": {
          "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": ""
          }
        },
        "eff3f5882bb1498e9692a0fc09702227": {
          "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_ea31d5271571473db7e393616bb6efe0",
              "IPY_MODEL_2aade86919044c60bcedc46a417afaa9",
              "IPY_MODEL_44ef26482dca4c51a104ea4c2a14779a"
            ],
            "layout": "IPY_MODEL_13b9e53e2ffd408498087501ee1856dc"
          }
        },
        "ea31d5271571473db7e393616bb6efe0": {
          "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_f5a2ca2a1053492d8f4c02a0b6408e1a",
            "placeholder": "​",
            "style": "IPY_MODEL_885ac34100e249aab74faa2fc530b5bc",
            "value": "Downloading data: 100%"
          }
        },
        "2aade86919044c60bcedc46a417afaa9": {
          "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_8cfad128757f428294ae5e79a62d4173",
            "max": 1648877,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_2d80eef0187843b9b2654787dc7d2e90",
            "value": 1648877
          }
        },
        "44ef26482dca4c51a104ea4c2a14779a": {
          "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_09510694789f48bbbf9fbb6dcd2e97b5",
            "placeholder": "​",
            "style": "IPY_MODEL_23bafa5a163f4d299de4f63ea6bb11e5",
            "value": " 1.65M/1.65M [00:00&lt;00:00, 5.32MB/s]"
          }
        },
        "13b9e53e2ffd408498087501ee1856dc": {
          "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
          }
        },
        "f5a2ca2a1053492d8f4c02a0b6408e1a": {
          "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
          }
        },
        "885ac34100e249aab74faa2fc530b5bc": {
          "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": ""
          }
        },
        "8cfad128757f428294ae5e79a62d4173": {
          "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
          }
        },
        "2d80eef0187843b9b2654787dc7d2e90": {
          "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": ""
          }
        },
        "09510694789f48bbbf9fbb6dcd2e97b5": {
          "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
          }
        },
        "23bafa5a163f4d299de4f63ea6bb11e5": {
          "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": ""
          }
        },
        "93d41767799e41ea8618af0210ec55b6": {
          "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_af5055e502f44b8aaaaa22de9881e508",
              "IPY_MODEL_cafe95ad16c84f45b183302d581e1eed",
              "IPY_MODEL_adc01a3cae4746f28a5be43a0edaa56b"
            ],
            "layout": "IPY_MODEL_4502ffb7073f4e21b1c2d60fce243561"
          }
        },
        "af5055e502f44b8aaaaa22de9881e508": {
          "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_0d0d2bbe902d41d3865053de27541eaf",
            "placeholder": "​",
            "style": "IPY_MODEL_6c02fad1b7c8445fbfba694750884ff5",
            "value": "Downloading data: 100%"
          }
        },
        "cafe95ad16c84f45b183302d581e1eed": {
          "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_688aee4457a34c399138443f09e16834",
            "max": 4542,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_6562f3ac438b491ba07a110a1aab6023",
            "value": 4542
          }
        },
        "adc01a3cae4746f28a5be43a0edaa56b": {
          "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_cafcee19eb944a6f88b3b7fc2a97c5e6",
            "placeholder": "​",
            "style": "IPY_MODEL_cb820b8b322a4326b73af3a92b3c9372",
            "value": " 4.54k/4.54k [00:00&lt;00:00, 311kB/s]"
          }
        },
        "4502ffb7073f4e21b1c2d60fce243561": {
          "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
          }
        },
        "0d0d2bbe902d41d3865053de27541eaf": {
          "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
          }
        },
        "6c02fad1b7c8445fbfba694750884ff5": {
          "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": ""
          }
        },
        "688aee4457a34c399138443f09e16834": {
          "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
          }
        },
        "6562f3ac438b491ba07a110a1aab6023": {
          "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": ""
          }
        },
        "cafcee19eb944a6f88b3b7fc2a97c5e6": {
          "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
          }
        },
        "cb820b8b322a4326b73af3a92b3c9372": {
          "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": ""
          }
        },
        "a9414d496315480eade1694d9b66400b": {
          "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_6d3591d3701848708edf04e2bc1206df",
              "IPY_MODEL_6ce0f372543846f992fcf93428c705f6",
              "IPY_MODEL_2d24d0f475834289ae6ab9e6719cc7d3"
            ],
            "layout": "IPY_MODEL_75b2d9ec0bbd4624a2fbf72e539b848a"
          }
        },
        "6d3591d3701848708edf04e2bc1206df": {
          "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_a323df42ae2644e4a3f1484b215d326a",
            "placeholder": "​",
            "style": "IPY_MODEL_4fce455f84064a349cd0b351be0897b0",
            "value": "Generating train split: 100%"
          }
        },
        "6ce0f372543846f992fcf93428c705f6": {
          "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_6a3e4ba30c6742d9b21f1625e29f2b40",
            "max": 60000,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_0a1727ec6e0f4b6cb9488fc6c0eda863",
            "value": 60000
          }
        },
        "2d24d0f475834289ae6ab9e6719cc7d3": {
          "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_d79de4e55fdd4ea6946ee96549f3afcb",
            "placeholder": "​",
            "style": "IPY_MODEL_170bd9351f2e4b12b1d8adf352e09993",
            "value": " 60000/60000 [00:11&lt;00:00, 5337.35 examples/s]"
          }
        },
        "75b2d9ec0bbd4624a2fbf72e539b848a": {
          "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
          }
        },
        "a323df42ae2644e4a3f1484b215d326a": {
          "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
          }
        },
        "4fce455f84064a349cd0b351be0897b0": {
          "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": ""
          }
        },
        "6a3e4ba30c6742d9b21f1625e29f2b40": {
          "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
          }
        },
        "0a1727ec6e0f4b6cb9488fc6c0eda863": {
          "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": ""
          }
        },
        "d79de4e55fdd4ea6946ee96549f3afcb": {
          "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
          }
        },
        "170bd9351f2e4b12b1d8adf352e09993": {
          "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": ""
          }
        },
        "0fdc678859824eea96315be5dda89c59": {
          "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_095cff2e90ec4a6e8e1a91b1ddcffb91",
              "IPY_MODEL_f79bac358c404c5ca5fade049229e63b",
              "IPY_MODEL_3fed2f153b684ba08e10c18b36d5e1fc"
            ],
            "layout": "IPY_MODEL_8aae4c7a70f349cb9c6201b1909a39bc"
          }
        },
        "095cff2e90ec4a6e8e1a91b1ddcffb91": {
          "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_d1765676885148368d04aab6d88bb9a5",
            "placeholder": "​",
            "style": "IPY_MODEL_77af331945874750aa8d2efc9fad490e",
            "value": "Generating test split: 100%"
          }
        },
        "f79bac358c404c5ca5fade049229e63b": {
          "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_2bead3491af44d2db006d394f616d041",
            "max": 10000,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_9f414d2470b04516bf700867b8de4f8a",
            "value": 10000
          }
        },
        "3fed2f153b684ba08e10c18b36d5e1fc": {
          "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_c0310e887cd24a37b6eb642730c74381",
            "placeholder": "​",
            "style": "IPY_MODEL_7466100cd17449559f41c5eefa23cc09",
            "value": " 10000/10000 [00:01&lt;00:00, 4861.26 examples/s]"
          }
        },
        "8aae4c7a70f349cb9c6201b1909a39bc": {
          "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
          }
        },
        "d1765676885148368d04aab6d88bb9a5": {
          "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
          }
        },
        "77af331945874750aa8d2efc9fad490e": {
          "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": ""
          }
        },
        "2bead3491af44d2db006d394f616d041": {
          "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
          }
        },
        "9f414d2470b04516bf700867b8de4f8a": {
          "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": ""
          }
        },
        "c0310e887cd24a37b6eb642730c74381": {
          "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
          }
        },
        "7466100cd17449559f41c5eefa23cc09": {
          "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": ""
          }
        }
      }
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/yblee110/jax-flax-book/blob/main/ch03_2_3_%ED%95%99%EC%8A%B5%EB%A5%A0_%EC%8A%A4%EC%BC%80%EC%A4%84%EB%A7%81_%EC%A0%81%EC%9A%A9.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 1,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "AQNnxuGxhnop",
        "outputId": "700c8474-b50a-43fd-b6f4-11bbcc6beddb"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Collecting datasets\n",
            "  Downloading datasets-2.20.0-py3-none-any.whl (547 kB)\n",
            "\u001b[?25l     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m0.0/547.8 kB\u001b[0m \u001b[31m?\u001b[0m eta \u001b[36m-:--:--\u001b[0m\r\u001b[2K     \u001b[91m━━━━━━━━━━━\u001b[0m\u001b[90m╺\u001b[0m\u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m153.6/547.8 kB\u001b[0m \u001b[31m4.4 MB/s\u001b[0m eta \u001b[36m0:00:01\u001b[0m\r\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m547.8/547.8 kB\u001b[0m \u001b[31m8.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from datasets) (3.15.4)\n",
            "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from datasets) (1.25.2)\n",
            "Collecting pyarrow>=15.0.0 (from datasets)\n",
            "  Downloading pyarrow-17.0.0-cp310-cp310-manylinux_2_28_x86_64.whl (39.9 MB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m39.9/39.9 MB\u001b[0m \u001b[31m31.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: pyarrow-hotfix in /usr/local/lib/python3.10/dist-packages (from datasets) (0.6)\n",
            "Collecting dill<0.3.9,>=0.3.0 (from datasets)\n",
            "  Downloading dill-0.3.8-py3-none-any.whl (116 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m116.3/116.3 kB\u001b[0m \u001b[31m12.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from datasets) (2.0.3)\n",
            "Collecting requests>=2.32.2 (from datasets)\n",
            "  Downloading requests-2.32.3-py3-none-any.whl (64 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m64.9/64.9 kB\u001b[0m \u001b[31m6.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: tqdm>=4.66.3 in /usr/local/lib/python3.10/dist-packages (from datasets) (4.66.4)\n",
            "Collecting xxhash (from datasets)\n",
            "  Downloading xxhash-3.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (194 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m194.1/194.1 kB\u001b[0m \u001b[31m9.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hCollecting multiprocess (from datasets)\n",
            "  Downloading multiprocess-0.70.16-py310-none-any.whl (134 kB)\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m134.8/134.8 kB\u001b[0m \u001b[31m10.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25hRequirement already satisfied: fsspec[http]<=2024.5.0,>=2023.1.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (2023.6.0)\n",
            "Requirement already satisfied: aiohttp in /usr/local/lib/python3.10/dist-packages (from datasets) (3.9.5)\n",
            "Requirement already satisfied: huggingface-hub>=0.21.2 in /usr/local/lib/python3.10/dist-packages (from datasets) (0.23.4)\n",
            "Requirement already satisfied: packaging in /usr/local/lib/python3.10/dist-packages (from datasets) (24.1)\n",
            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from datasets) (6.0.1)\n",
            "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.3.1)\n",
            "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (23.2.0)\n",
            "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.4.1)\n",
            "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (6.0.5)\n",
            "Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.9.4)\n",
            "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (4.0.3)\n",
            "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.21.2->datasets) (4.12.2)\n",
            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests>=2.32.2->datasets) (3.3.2)\n",
            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests>=2.32.2->datasets) (3.7)\n",
            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests>=2.32.2->datasets) (2.0.7)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests>=2.32.2->datasets) (2024.7.4)\n",
            "Requirement already satisfied: python-dateutil>=2.8.2 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2.8.2)\n",
            "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2023.4)\n",
            "Requirement already satisfied: tzdata>=2022.1 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2024.1)\n",
            "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.8.2->pandas->datasets) (1.16.0)\n",
            "Installing collected packages: xxhash, requests, pyarrow, dill, multiprocess, datasets\n",
            "  Attempting uninstall: requests\n",
            "    Found existing installation: requests 2.31.0\n",
            "    Uninstalling requests-2.31.0:\n",
            "      Successfully uninstalled requests-2.31.0\n",
            "  Attempting uninstall: pyarrow\n",
            "    Found existing installation: pyarrow 14.0.2\n",
            "    Uninstalling pyarrow-14.0.2:\n",
            "      Successfully uninstalled pyarrow-14.0.2\n",
            "\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n",
            "cudf-cu12 24.4.1 requires pyarrow<15.0.0a0,>=14.0.1, but you have pyarrow 17.0.0 which is incompatible.\n",
            "google-colab 1.0.0 requires requests==2.31.0, but you have requests 2.32.3 which is incompatible.\n",
            "ibis-framework 8.0.0 requires pyarrow<16,>=2, but you have pyarrow 17.0.0 which is incompatible.\u001b[0m\u001b[31m\n",
            "\u001b[0mSuccessfully installed datasets-2.20.0 dill-0.3.8 multiprocess-0.70.16 pyarrow-17.0.0 requests-2.32.3 xxhash-3.4.1\n",
            "\u001b[31mERROR: Invalid requirement: '==0.4.4'\u001b[0m\u001b[31m\n",
            "\u001b[0m"
          ]
        }
      ],
      "source": [
        "!pip install datasets\n",
        "!pip install orbax-checkpoint ==0.4.4"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import jax\n",
        "import flax\n",
        "import optax\n",
        "import datasets\n",
        "\n",
        "import jax.numpy as jnp\n",
        "from datasets import load_dataset\n",
        "import flax.linen as nn\n",
        "\n",
        "print(\"JAX Version : {}\".format(jax.__version__)) #출력 JAX Version : 0.4.20\n",
        "print(\"FLAX Version : {}\".format(flax.__version__)) #출력 FLAX Version : 0.7.5\n",
        "print(\"OPTAX Version : {}\".format(optax.__version__))#출력 OPTAX Version : 0.1.7"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "Sm46RyL2iy3f",
        "outputId": "b3c054ce-7304-4e73-c414-c88abbc11551"
      },
      "execution_count": 2,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "JAX Version : 0.4.26\n",
            "FLAX Version : 0.8.4\n",
            "OPTAX Version : 0.2.2\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "def get_datasets():\n",
        "  datasets = load_dataset(\"mnist\")\n",
        "  datasets = datasets.with_format(\"jax\")\n",
        "  datasets = {\n",
        "    \"train\": {\n",
        "      \"image\": datasets[\"train\"][\"image\"][...,None].astype(jnp.float32)/255,\n",
        "      \"label\": datasets[\"train\"][\"label\"],\n",
        "    },\n",
        "    \"test\": {\n",
        "      \"image\": datasets[\"test\"][\"image\"][...,None].astype(jnp.float32)/255,\n",
        "      \"label\": datasets[\"test\"][\"label\"],\n",
        "    },\n",
        "  }\n",
        "  return datasets['train'], datasets['test']\n"
      ],
      "metadata": {
        "id": "W7ET-NqZiy57"
      },
      "execution_count": 3,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "train_ds, test_ds = get_datasets()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 467,
          "referenced_widgets": [
            "2e681bf0b9e24b8f80d6116a2d274660",
            "28dba86a0f714b44b99f98049e870c90",
            "95fe96f3a27a4babb1ebfdbe7d438947",
            "1c3e86fc780a4a70a621be1b231327a6",
            "6b0e915a14e24039ae20959cce8f2361",
            "fe1acc6d003740e18182b998b40ce1a5",
            "4bf6bd1ba2b545bebd6afd3d951a11b4",
            "cec22bf3b1e14c9aa7cf07ca557bfc80",
            "440089cd56384a16a6bdf58aa275b9ff",
            "9d940bf040624a28a1e64d69884604eb",
            "12db35e33e774767b2c652c2c2904162",
            "75575b7ec7d04a318833fa7e94de32b9",
            "22b87c9c11eb441fae3ca586d8026451",
            "933c4b13661641ce902e348fa366c450",
            "1d44095165f64f2b9d17c99050ec4647",
            "1f735c881ef8403885c2b12ab564276f",
            "b5eb9c734a504f09b2785f3bce2f6a51",
            "c4f9b243aff34f1fb7efbf7410297191",
            "5ff2496902634221a01d2aa70a2bd50c",
            "4f6d494d591c4928bb99f006587be5d7",
            "b3a87b66d6314fb4889bb49b4e909e31",
            "e84f7e31112a476b83bd86e4edc5b2b4",
            "0586c707ed1848dba080e457cfa23a0a",
            "3db7b35d2b29417ea027772a853040db",
            "fb7b608d55524911911951b240f77cbd",
            "1aeeb4ba924a4d2092286376cb8bc92c",
            "224e21e22dfb4301ae1288ff3aa04b3f",
            "492e0a80d555461fadf425c535f0d44e",
            "8feb6bc1a02b478bb68e1617c9660d08",
            "9f0cb08d9da743ecb11c5aa19294c92f",
            "c0f3c0fc815741c8b198ea64dc5a16a6",
            "a1845d30ad484f81841579c381536395",
            "3c70b2e5471e41229e215a31ea2826b4",
            "f6178b580da04707b03163d0914ee73f",
            "2f246525418649feac1938ed87294d7d",
            "217749a6e4994b98a88f3663304d64f5",
            "800284a99d344164bbc72071d777c9b6",
            "ec2f8f3dacaa449092bfb4a4d3b7cbef",
            "3d1b7002bb4e46d8920d02cf0e9851e0",
            "8b55aafb81964224afebd5f4df76814f",
            "bfea677be7b14abf870ec2f838697e5e",
            "71101171f0284d98ac94035204f24bfb",
            "2348ee0b6b4a4f16b0fe52e032fc0915",
            "64f1882a7e5843d7943baa2d84c65709",
            "eff3f5882bb1498e9692a0fc09702227",
            "ea31d5271571473db7e393616bb6efe0",
            "2aade86919044c60bcedc46a417afaa9",
            "44ef26482dca4c51a104ea4c2a14779a",
            "13b9e53e2ffd408498087501ee1856dc",
            "f5a2ca2a1053492d8f4c02a0b6408e1a",
            "885ac34100e249aab74faa2fc530b5bc",
            "8cfad128757f428294ae5e79a62d4173",
            "2d80eef0187843b9b2654787dc7d2e90",
            "09510694789f48bbbf9fbb6dcd2e97b5",
            "23bafa5a163f4d299de4f63ea6bb11e5",
            "93d41767799e41ea8618af0210ec55b6",
            "af5055e502f44b8aaaaa22de9881e508",
            "cafe95ad16c84f45b183302d581e1eed",
            "adc01a3cae4746f28a5be43a0edaa56b",
            "4502ffb7073f4e21b1c2d60fce243561",
            "0d0d2bbe902d41d3865053de27541eaf",
            "6c02fad1b7c8445fbfba694750884ff5",
            "688aee4457a34c399138443f09e16834",
            "6562f3ac438b491ba07a110a1aab6023",
            "cafcee19eb944a6f88b3b7fc2a97c5e6",
            "cb820b8b322a4326b73af3a92b3c9372",
            "a9414d496315480eade1694d9b66400b",
            "6d3591d3701848708edf04e2bc1206df",
            "6ce0f372543846f992fcf93428c705f6",
            "2d24d0f475834289ae6ab9e6719cc7d3",
            "75b2d9ec0bbd4624a2fbf72e539b848a",
            "a323df42ae2644e4a3f1484b215d326a",
            "4fce455f84064a349cd0b351be0897b0",
            "6a3e4ba30c6742d9b21f1625e29f2b40",
            "0a1727ec6e0f4b6cb9488fc6c0eda863",
            "d79de4e55fdd4ea6946ee96549f3afcb",
            "170bd9351f2e4b12b1d8adf352e09993",
            "0fdc678859824eea96315be5dda89c59",
            "095cff2e90ec4a6e8e1a91b1ddcffb91",
            "f79bac358c404c5ca5fade049229e63b",
            "3fed2f153b684ba08e10c18b36d5e1fc",
            "8aae4c7a70f349cb9c6201b1909a39bc",
            "d1765676885148368d04aab6d88bb9a5",
            "77af331945874750aa8d2efc9fad490e",
            "2bead3491af44d2db006d394f616d041",
            "9f414d2470b04516bf700867b8de4f8a",
            "c0310e887cd24a37b6eb642730c74381",
            "7466100cd17449559f41c5eefa23cc09"
          ]
        },
        "id": "a6fDppGLiy8b",
        "outputId": "78f8ae1d-9b4b-4d60-cad5-0e3fcb179231"
      },
      "execution_count": 4,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_token.py:89: UserWarning: \n",
            "The secret `HF_TOKEN` does not exist in your Colab secrets.\n",
            "To authenticate with the Hugging Face Hub, create a token in your settings tab (https://huggingface.co/settings/tokens), set it as secret in your Google Colab and restart your session.\n",
            "You will be able to reuse this secret in all of your notebooks.\n",
            "Please note that authentication is recommended but still optional to access public models or datasets.\n",
            "  warnings.warn(\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading builder script:   0%|          | 0.00/3.98k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "2e681bf0b9e24b8f80d6116a2d274660"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading readme:   0%|          | 0.00/6.83k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "75575b7ec7d04a318833fa7e94de32b9"
            }
          },
          "metadata": {}
        },
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "The repository for mnist contains custom code which must be executed to correctly load the dataset. You can inspect the repository content at https://hf.co/datasets/mnist.\n",
            "You can avoid this prompt in future by passing the argument `trust_remote_code=True`.\n",
            "\n",
            "Do you wish to run the custom code? [y/N] y\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading data:   0%|          | 0.00/9.91M [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "0586c707ed1848dba080e457cfa23a0a"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading data:   0%|          | 0.00/28.9k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "f6178b580da04707b03163d0914ee73f"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading data:   0%|          | 0.00/1.65M [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "eff3f5882bb1498e9692a0fc09702227"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading data:   0%|          | 0.00/4.54k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "93d41767799e41ea8618af0210ec55b6"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Generating train split:   0%|          | 0/60000 [00:00<?, ? examples/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "a9414d496315480eade1694d9b66400b"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Generating test split:   0%|          | 0/10000 [00:00<?, ? examples/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "0fdc678859824eea96315be5dda89c59"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "class CNN(nn.Module):\n",
        "    num_classes: int\n",
        "\n",
        "\n",
        "    @nn.compact\n",
        "    def __call__(self, x, train: bool):\n",
        "        x = nn.Conv(features=16, kernel_size=(5, 5), strides=(2, 2),\n",
        "                    padding='VALID')(x)\n",
        "        x = nn.BatchNorm(use_running_average=not train)(x)\n",
        "        x = nn.relu(x)\n",
        "        x = nn.Conv(features=32, kernel_size=(5, 5), strides=(2, 2),\n",
        "                    padding='VALID')(x)\n",
        "        x = nn.BatchNorm(use_running_average=not train)(x)\n",
        "        x = nn.relu(x)\n",
        "        x = nn.Dropout(rate=0.5, deterministic=not train)(x)\n",
        "        x = jnp.mean(x, axis=(1, 2))\n",
        "        x = nn.Dense(features=self.num_classes)(x)\n",
        "        return x\n"
      ],
      "metadata": {
        "id": "vbyXCzQdiy-v"
      },
      "execution_count": 5,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "rng = jax.random.PRNGKey(0)\n",
        "model = CNN(num_classes=10)\n",
        "main_key, params_key, dropout_key = jax.random.split(key=rng, num=3)\n",
        "variables = model.init(params_key, jnp.ones((1, 28, 28, 1)),train=False)\n",
        "params = variables['params']\n",
        "batch_stats = variables['batch_stats']"
      ],
      "metadata": {
        "id": "3MozFm1LizBX"
      },
      "execution_count": 6,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "jax.tree_util.tree_map(jnp.shape, variables)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "Abl-0xVLizD7",
        "outputId": "ce3f09e5-fd85-4c95-b496-b74597a7eee0"
      },
      "execution_count": 7,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "{'batch_stats': {'BatchNorm_0': {'mean': (16,), 'var': (16,)},\n",
              "  'BatchNorm_1': {'mean': (32,), 'var': (32,)}},\n",
              " 'params': {'BatchNorm_0': {'bias': (16,), 'scale': (16,)},\n",
              "  'BatchNorm_1': {'bias': (32,), 'scale': (32,)},\n",
              "  'Conv_0': {'bias': (16,), 'kernel': (5, 5, 1, 16)},\n",
              "  'Conv_1': {'bias': (32,), 'kernel': (5, 5, 16, 32)},\n",
              "  'Dense_0': {'bias': (10,), 'kernel': (32, 10)}}}"
            ]
          },
          "metadata": {},
          "execution_count": 7
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "def compute_metrics(logits, labels):\n",
        "  loss = jnp.mean(optax.softmax_cross_entropy(logits, jax.nn.one_hot(labels, num_classes=10)))\n",
        "  accuracy = jnp.mean(jnp.argmax(logits, -1) == labels)\n",
        "  metrics = {\n",
        "    'loss': loss,\n",
        "    'accuracy': accuracy\n",
        "  }\n",
        "  return metrics"
      ],
      "metadata": {
        "id": "MwClXA6xizGX"
      },
      "execution_count": 8,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def create_learning_rate_fn(num_epochs, warmup_epochs, base_learning_rate, steps_per_epoch):\n",
        "    warmup_fn = optax.linear_schedule(\n",
        "        init_value=0., end_value=base_learning_rate,\n",
        "        transition_steps=warmup_epochs * steps_per_epoch)\n",
        "    cosine_epochs = max(num_epochs - warmup_epochs, 1)\n",
        "\n",
        "    cosine_fn = optax.cosine_decay_schedule(\n",
        "        init_value=base_learning_rate,\n",
        "        decay_steps=cosine_epochs * steps_per_epoch)\n",
        "    schedule_fn = optax.join_schedules(\n",
        "        schedules=[warmup_fn, cosine_fn],\n",
        "        boundaries=[warmup_epochs * steps_per_epoch])\n",
        "\n",
        "    return schedule_fn"
      ],
      "metadata": {
        "id": "dujuZb-EizLw"
      },
      "execution_count": 9,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "from flax.training import train_state\n",
        "from typing import Any\n",
        "\n",
        "train_epoch = 10\n",
        "warmup_epoch = 2\n",
        "learning_rate = 0.01\n",
        "batch_size = 64\n",
        "eval_batch_size = 100\n",
        "train_ds_size = len(train_ds['image'])\n",
        "steps_per_epoch = train_ds_size // batch_size\n",
        "learning_rate_fn = create_learning_rate_fn(train_epoch, warmup_epoch, learning_rate, steps_per_epoch)\n",
        "\n",
        "class TrainState(train_state.TrainState):\n",
        "  batch_stats: Any\n",
        "  key: jax.random.key\n",
        "\n",
        "tx = optax.adam(learning_rate=learning_rate_fn)\n",
        "state = TrainState.create(\n",
        "  apply_fn=model.apply,\n",
        "  params=params,\n",
        "  batch_stats=batch_stats,\n",
        "  key=dropout_key,\n",
        "  tx=tx,\n",
        ")"
      ],
      "metadata": {
        "id": "yYtWYoG1izJD"
      },
      "execution_count": 10,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "import orbax.checkpoint\n",
        "from flax.training import orbax_utils\n",
        "import os\n",
        "import shutil"
      ],
      "metadata": {
        "id": "A-YyJE1ftv76"
      },
      "execution_count": 11,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "train_path = os.getenv('HOME') + '/cnn/'\n",
        "\n",
        "\n",
        "ckpt_mgr = orbax.checkpoint.CheckpointManager(\n",
        "       train_path,\n",
        "       orbax.checkpoint.Checkpointer(orbax.checkpoint.PyTreeCheckpointHandler()),\n",
        "       orbax.checkpoint.CheckpointManagerOptions(\n",
        "           create=not (os.path.isdir(train_path)),\n",
        "           max_to_keep=3,\n",
        "           step_prefix=\"model_epoch\",\n",
        "       ),\n",
        "   )\n"
      ],
      "metadata": {
        "id": "P9bhIiiQqN0i"
      },
      "execution_count": 12,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "import functools\n",
        "@functools.partial(jax.jit, static_argnums=2)\n",
        "def train_step(state, batch, learning_rate_fn):\n",
        "  dropout_train_key = jax.random.fold_in(key=dropout_key,\n",
        "    data=state.step)\n",
        "  def loss_fn(params):\n",
        "    logits, updates = state.apply_fn(\n",
        "    {'params': params, 'batch_stats': state.batch_stats},\n",
        "    batch['image'], train=True, mutable=['batch_stats'],\n",
        "    rngs={'dropout': dropout_train_key})\n",
        "    loss = jnp.mean(optax.softmax_cross_entropy(\n",
        "      logits=logits,\n",
        "      labels=jax.nn.one_hot(batch['label'],num_classes=10)))\n",
        "    return loss, (logits, updates)\n",
        "  grad_fn = jax.value_and_grad(loss_fn, has_aux=True)\n",
        "  (_, (logits, updates)), grads = grad_fn(state.params)\n",
        "  state = state.apply_gradients(grads=grads)\n",
        "  state = state.replace(batch_stats=updates['batch_stats'])\n",
        "  metrics = compute_metrics(logits, batch['label'])\n",
        "  lr = learning_rate_fn(state.step)\n",
        "  metrics['learning_rate'] = lr\n",
        "  return state, metrics\n"
      ],
      "metadata": {
        "id": "bGd2MP3xizOY"
      },
      "execution_count": 13,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "@jax.jit\n",
        "def eval_step(state, batch):\n",
        "  logits = state.apply_fn(\n",
        "     {'params': state.params, 'batch_stats': state.batch_stats},\n",
        "     batch['image'], train=False)\n",
        "  return compute_metrics(logits, batch['label'])"
      ],
      "metadata": {
        "id": "Gj3O8hB9lXRU"
      },
      "execution_count": 14,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def train_loop(state, train_ds, batch_size, epoch, learning_rate_fn, rng):\n",
        "  train_ds_size = len(train_ds['image'])\n",
        "  steps_per_epoch = train_ds_size // batch_size\n",
        "  perms = jax.random.permutation(rng, train_ds_size)\n",
        "  perms = perms[:steps_per_epoch * batch_size]\n",
        "  perms = perms.reshape((steps_per_epoch, batch_size))\n",
        "  batch_metrics = []\n",
        "  for perm in perms:\n",
        "    batch = {k: v[perm, ...] for k, v in train_ds.items()}\n",
        "    state, metrics = train_step(state, batch, learning_rate_fn)\n",
        "    batch_metrics.append(metrics)\n",
        "  training_batch_metrics = jax.device_get(batch_metrics)\n",
        "  training_epoch_metrics = {k: sum([metrics[k] for metrics in\n",
        "    training_batch_metrics])/steps_per_epoch\n",
        "    for k in training_batch_metrics[0]}\n",
        "  print('EPOCH: %d\\nTraining loss: %.4f, accuracy: %.2f' % (epoch,\n",
        "    training_epoch_metrics['loss'],\n",
        "    training_epoch_metrics['accuracy'] * 100))\n",
        "  return state\n"
      ],
      "metadata": {
        "id": "SyaKKJpxizQ8"
      },
      "execution_count": 15,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def eval_loop(state, test_ds, batch_size):\n",
        "  eval_ds_size = test_ds['image'].shape[0]\n",
        "  steps_per_epoch = eval_ds_size // batch_size\n",
        "  batch_metrics = []\n",
        "  for i in range(steps_per_epoch):\n",
        "    batch = {k: v[i*batch_size:(i+1)*batch_size, ...] for k, v in test_ds.items()}\n",
        "    metrics = eval_step(state, batch)\n",
        "    batch_metrics.append(metrics)\n",
        "  eval_batch_metrics = jax.device_get(batch_metrics)\n",
        "  eval_batch_metrics = {\n",
        "    k: sum([metrics[k] for metrics in eval_batch_metrics])/steps_per_epoch\n",
        "    for k in eval_batch_metrics[0]}\n",
        "\n",
        "  print('    Eval loss: %.4f, accuracy: %.2f' % (eval_batch_metrics['loss'], eval_batch_metrics['accuracy'] * 100))"
      ],
      "metadata": {
        "id": "rBvhCSm6izTT"
      },
      "execution_count": 16,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "for epoch in range(train_epoch):\n",
        "  rng, key = jax.random.split(rng)\n",
        "  state = train_loop(state, train_ds, batch_size, epoch,\n",
        "    learning_rate_fn, rng)\n",
        "  eval_loop(state, test_ds, eval_batch_size)\n",
        "  save_args = orbax_utils.save_args_from_target(state)\n",
        "  ckpt_mgr.save(epoch, state, save_kwargs={\"save_args\": save_args})"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "3sQYsaspizWE",
        "outputId": "51beff66-ea37-4d70-d699-9ae64855c245"
      },
      "execution_count": 17,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "EPOCH: 0\n",
            "Training loss: 0.9390, accuracy: 72.74\n",
            "    Eval loss: 0.3019, accuracy: 90.66\n",
            "EPOCH: 1\n",
            "Training loss: 0.2484, accuracy: 92.45\n",
            "    Eval loss: 0.1795, accuracy: 94.44\n",
            "EPOCH: 2\n",
            "Training loss: 0.1893, accuracy: 94.27\n",
            "    Eval loss: 0.1090, accuracy: 96.86\n",
            "EPOCH: 3\n",
            "Training loss: 0.1559, accuracy: 95.26\n",
            "    Eval loss: 0.0948, accuracy: 97.32\n",
            "EPOCH: 4\n",
            "Training loss: 0.1393, accuracy: 95.73\n",
            "    Eval loss: 0.1138, accuracy: 96.61\n",
            "EPOCH: 5\n",
            "Training loss: 0.1211, accuracy: 96.20\n",
            "    Eval loss: 0.0641, accuracy: 98.09\n",
            "EPOCH: 6\n",
            "Training loss: 0.1069, accuracy: 96.67\n",
            "    Eval loss: 0.0626, accuracy: 98.14\n",
            "EPOCH: 7\n",
            "Training loss: 0.0972, accuracy: 96.98\n",
            "    Eval loss: 0.0504, accuracy: 98.49\n",
            "EPOCH: 8\n",
            "Training loss: 0.0882, accuracy: 97.24\n",
            "    Eval loss: 0.0479, accuracy: 98.55\n",
            "EPOCH: 9\n",
            "Training loss: 0.0841, accuracy: 97.41\n",
            "    Eval loss: 0.0475, accuracy: 98.56\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "raw_restored = ckpt_mgr.restore('9')\n",
        "raw_restored"
      ],
      "metadata": {
        "id": "RjKOqDPhqf-W",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "6bd6a894-cf5e-49c4-9af4-ae34f934e17f"
      },
      "execution_count": 18,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "{'batch_stats': {'BatchNorm_0': {'mean': array([ 0.59678304,  0.34670404, -0.32364425, -1.5403948 , -0.41289315,\n",
              "          -0.9029092 , -1.0194184 , -1.4013318 , -0.2755111 , -0.40960672,\n",
              "          -0.2929568 , -1.3420625 ,  0.9907112 , -1.3641517 , -0.05670249,\n",
              "          -0.9666436 ], dtype=float32),\n",
              "   'var': array([2.754574  , 0.4765729 , 0.79068047, 3.139733  , 1.9464167 ,\n",
              "          3.3058364 , 3.4047148 , 3.8393342 , 1.8070192 , 2.9429402 ,\n",
              "          0.3669262 , 3.5045764 , 2.1112766 , 2.0194216 , 0.6417175 ,\n",
              "          2.8159597 ], dtype=float32)},\n",
              "  'BatchNorm_1': {'mean': array([-3.7710893 ,  1.0635827 , -5.265705  , -0.9515563 ,  5.7551737 ,\n",
              "          -7.405269  , -0.0529795 , -0.88988537, -2.9906244 ,  5.9144897 ,\n",
              "          -1.8900927 , -3.976402  , -2.58967   , -6.522824  , -2.9571164 ,\n",
              "           1.7212024 ,  0.36768922,  0.04797452, -0.2436241 , -7.0267615 ,\n",
              "           5.0057926 , -4.0950956 ,  1.439795  , -5.6158085 , -4.036963  ,\n",
              "           4.252638  , -0.54707086, -4.9110804 , -3.3844934 , -1.3179133 ,\n",
              "          -4.579265  ,  5.8077955 ], dtype=float32),\n",
              "   'var': array([ 7.0015135,  5.698587 , 10.192861 ,  8.713224 ,  5.925345 ,\n",
              "           7.763242 , 12.205926 ,  8.603589 , 12.3140745,  7.4565024,\n",
              "           8.95226  ,  7.0730014, 10.870935 ,  9.424113 ,  8.020228 ,\n",
              "           4.798607 ,  9.679846 ,  9.174238 , 12.396536 ,  8.061559 ,\n",
              "          10.566014 ,  8.434447 ,  7.7500076, 10.797337 ,  9.642636 ,\n",
              "           8.652558 ,  7.0093684,  7.1579623, 10.251053 , 12.818575 ,\n",
              "           9.238743 ,  8.465704 ], dtype=float32)}},\n",
              " 'key': array([2562233961, 1946702221], dtype=uint32),\n",
              " 'opt_state': [{'count': array(9370, dtype=int32),\n",
              "   'mu': {'BatchNorm_0': {'bias': array([ 0.00011218,  0.00393522,  0.00268865,  0.00872598,  0.00256785,\n",
              "             0.00846969, -0.00938905,  0.03185908, -0.0007597 ,  0.00070556,\n",
              "            -0.00117019, -0.02785125, -0.00064779,  0.01464934, -0.00451334,\n",
              "            -0.00747119], dtype=float32),\n",
              "     'scale': array([-0.0027057 ,  0.00090396,  0.00317749,  0.00622377,  0.01451865,\n",
              "             0.00984382, -0.01228256,  0.01718505, -0.00625001,  0.00326924,\n",
              "             0.00148809, -0.01958428,  0.00911643,  0.00902811, -0.01890563,\n",
              "            -0.00911766], dtype=float32)},\n",
              "    'BatchNorm_1': {'bias': array([ 1.0234136e-03,  7.6868746e-05, -3.3956405e-03, -1.5561990e-03,\n",
              "            -1.2846587e-03, -9.4913441e-04,  3.7964791e-04, -6.5858127e-04,\n",
              "             1.8359930e-03,  2.1006891e-03, -9.8427339e-04,  3.3746858e-03,\n",
              "            -2.2691891e-03, -1.4523538e-03, -1.0055257e-03,  7.6265819e-04,\n",
              "             6.2770257e-04,  7.9935300e-05,  5.7409785e-04, -5.3198484e-04,\n",
              "             3.5397045e-03, -2.3542109e-04,  2.4526514e-04, -2.0634686e-03,\n",
              "            -4.1270757e-04, -1.2505902e-03,  4.1771224e-03, -4.5893551e-04,\n",
              "            -1.9916310e-04, -4.3075311e-04,  2.1294078e-03,  1.9148360e-03],\n",
              "           dtype=float32),\n",
              "     'scale': array([ 1.9946832e-03,  6.6299527e-04, -4.1994359e-03, -2.0279009e-03,\n",
              "            -2.0682735e-03, -8.0172432e-04, -3.6106270e-05, -2.0919568e-03,\n",
              "             6.7025336e-04,  1.4917095e-03, -1.1616402e-03,  4.7715516e-03,\n",
              "            -2.8496461e-03, -1.0655548e-03, -2.0052085e-03,  1.3321586e-03,\n",
              "            -2.4441164e-04, -4.8973365e-04,  5.2599801e-04, -3.1203878e-04,\n",
              "             3.1512436e-03,  5.5528362e-06,  2.5895861e-04, -2.6188961e-03,\n",
              "            -6.3351390e-04, -1.1359939e-03,  4.0924526e-03, -1.7035348e-04,\n",
              "            -5.7142670e-04, -1.4482257e-03,  2.2642897e-03,  1.9485857e-03],\n",
              "           dtype=float32)},\n",
              "    'Conv_0': {'bias': array([ 3.8337111e-10,  1.1119576e-09, -5.2016891e-10,  1.5100583e-09,\n",
              "             3.0273506e-10,  9.0963148e-10,  1.6262542e-09,  1.1559359e-09,\n",
              "            -2.3942959e-10,  2.5832497e-10, -7.5719075e-10,  3.8713934e-09,\n",
              "            -6.8728306e-10,  3.9595274e-09, -4.6106055e-10, -8.8214352e-10],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[-2.96848884e-04, -1.82235567e-03, -1.01171620e-03,\n",
              "               -3.50947259e-04,  6.89605763e-03,  9.26060043e-03,\n",
              "               -1.25300395e-03, -6.71855058e-04,  1.05820538e-03,\n",
              "                1.95391802e-03, -4.82726702e-03,  1.57979364e-03,\n",
              "                9.78429685e-04,  2.70817336e-03, -2.14182219e-04,\n",
              "               -2.38754367e-03]],\n",
              "     \n",
              "             [[ 3.36391124e-04, -1.20795565e-04,  1.23632024e-03,\n",
              "               -7.53012719e-04,  7.13743037e-03,  9.10865609e-03,\n",
              "               -1.43528532e-03, -1.04250107e-03,  1.47956726e-03,\n",
              "                3.37508973e-04, -3.01246904e-03,  1.44770613e-03,\n",
              "                1.81476411e-03,  2.39824108e-03, -9.19770624e-04,\n",
              "                3.16749210e-04]],\n",
              "     \n",
              "             [[-8.99199746e-04, -7.01596437e-04,  1.03088410e-03,\n",
              "               -1.33303553e-03,  6.97366288e-03,  8.24263040e-03,\n",
              "               -1.56531099e-03, -1.53717119e-03,  1.11995486e-03,\n",
              "               -7.93748477e-04, -8.88962299e-04,  1.20532548e-03,\n",
              "                2.23594811e-03,  3.28022637e-03, -1.06628682e-03,\n",
              "                3.11028445e-04]],\n",
              "     \n",
              "             [[-2.28290679e-03,  2.29788318e-04, -3.05839611e-04,\n",
              "               -2.39662593e-03,  6.51510479e-03,  7.76547100e-03,\n",
              "               -1.36503647e-03, -1.89448555e-03, -5.46038034e-04,\n",
              "               -7.17488991e-04,  2.45883432e-03,  6.70706140e-05,\n",
              "                3.30166565e-03,  2.36966577e-03, -8.04435986e-04,\n",
              "               -2.90883647e-04]],\n",
              "     \n",
              "             [[-2.31370004e-03,  4.40599257e-03,  2.12413725e-05,\n",
              "               -2.10534153e-03,  4.48246859e-03,  6.62992243e-03,\n",
              "               -1.11698965e-03, -2.46614311e-03, -1.60908396e-03,\n",
              "               -1.76038255e-03,  4.64955391e-03, -1.35204216e-04,\n",
              "                3.21591063e-03,  1.41347956e-03, -1.08023451e-05,\n",
              "               -5.53606427e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[ 2.10628728e-04,  1.86706893e-06,  9.42490587e-04,\n",
              "                1.81696098e-03,  4.53603733e-03,  4.22004936e-03,\n",
              "               -3.46384104e-03, -1.29151228e-03,  4.66487574e-04,\n",
              "                7.21966673e-04, -3.04496102e-03,  3.60009912e-03,\n",
              "                2.07994366e-03,  1.21391262e-04, -4.88317863e-04,\n",
              "               -1.31198857e-03]],\n",
              "     \n",
              "             [[ 6.86390325e-04,  2.85311649e-03,  1.76083448e-03,\n",
              "                1.00955949e-03,  5.07141557e-03,  5.34067117e-03,\n",
              "               -4.45502065e-03, -2.57594837e-03,  1.05525716e-03,\n",
              "               -1.29814795e-03, -2.52578803e-03,  3.58945015e-03,\n",
              "                2.33875471e-03,  4.56012785e-05, -6.15677680e-04,\n",
              "               -1.34221488e-03]],\n",
              "     \n",
              "             [[-4.06072591e-04,  2.35722051e-04,  3.92474059e-04,\n",
              "                9.88892396e-04,  4.40715067e-03,  4.82882233e-03,\n",
              "               -4.36806120e-03, -1.78444060e-03,  1.02243177e-03,\n",
              "               -9.63906641e-05, -2.92450981e-03,  4.14713658e-03,\n",
              "                2.46251561e-03, -2.48568947e-04, -1.02097297e-03,\n",
              "               -1.47316419e-03]],\n",
              "     \n",
              "             [[-1.56264647e-03,  1.02772028e-03, -1.13148312e-03,\n",
              "                4.48494626e-04,  3.36440955e-03,  6.05505379e-03,\n",
              "               -4.46543749e-03, -8.66374467e-04, -4.94409469e-04,\n",
              "               -4.56432317e-04, -3.75981326e-05,  1.83933391e-03,\n",
              "                3.86102009e-03,  2.06313387e-04, -6.88367581e-04,\n",
              "               -1.32422359e-03]],\n",
              "     \n",
              "             [[-1.14505668e-03,  2.93598324e-03,  3.60786536e-04,\n",
              "                4.43961384e-04,  2.67292815e-03,  4.35424224e-03,\n",
              "               -3.62533424e-03, -1.26196491e-03, -2.09251943e-04,\n",
              "                8.22215341e-04,  3.09079210e-03,  8.53063364e-04,\n",
              "                3.32497479e-03, -1.56928785e-04, -1.42349571e-03,\n",
              "               -1.14572444e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[-4.20978206e-04,  2.02089129e-03,  7.12846173e-04,\n",
              "                2.67367414e-03,  3.55608854e-03,  4.25342098e-03,\n",
              "               -3.69000179e-03, -2.17419118e-03, -5.26792544e-04,\n",
              "               -2.85987568e-04, -1.12841395e-03,  7.23175984e-03,\n",
              "                1.41481229e-03, -1.07917935e-03,  4.36288799e-04,\n",
              "               -1.21425302e-03]],\n",
              "     \n",
              "             [[ 5.36327658e-04,  2.35502189e-03,  2.58084293e-03,\n",
              "                1.81021704e-03,  3.48159065e-03,  3.42876278e-03,\n",
              "               -5.77986520e-03, -2.56331312e-03,  4.00672929e-04,\n",
              "               -1.76932453e-03, -2.24187691e-03,  7.38749234e-03,\n",
              "                1.66437100e-03, -2.30407284e-04,  1.93427259e-04,\n",
              "               -1.44386897e-04]],\n",
              "     \n",
              "             [[-2.42639973e-04,  1.86177215e-03, -7.84953532e-04,\n",
              "                4.01724741e-04,  3.37032671e-03,  3.77496984e-03,\n",
              "               -6.30620169e-03, -1.95908034e-03,  1.21230574e-03,\n",
              "               -1.34899013e-03, -1.25399732e-03,  6.50181482e-03,\n",
              "                2.44445936e-03, -1.40970806e-04, -7.67748919e-04,\n",
              "               -7.73097156e-04]],\n",
              "     \n",
              "             [[-7.28420098e-04,  3.56603065e-03, -5.70201897e-04,\n",
              "                1.65158679e-04,  2.73926603e-03,  4.32698382e-03,\n",
              "               -6.01168210e-03, -7.22099678e-04,  9.51970636e-04,\n",
              "                1.07393949e-04, -4.97556990e-04,  4.15379927e-03,\n",
              "                2.24152789e-03,  1.77513040e-03, -2.68754410e-03,\n",
              "               -8.94938421e-04]],\n",
              "     \n",
              "             [[-5.62098227e-04,  5.92336291e-03,  1.00490893e-03,\n",
              "               -3.00380372e-04,  2.38642888e-03,  4.18860465e-03,\n",
              "               -5.13678323e-03, -1.07836840e-03,  2.46456824e-03,\n",
              "                2.98419618e-04,  3.81559879e-03,  2.09104014e-03,\n",
              "                2.48142495e-03,  3.30497336e-04, -2.58909026e-03,\n",
              "               -1.17367087e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[ 1.45226149e-04,  3.24689946e-03,  1.11087307e-03,\n",
              "                1.49485760e-03,  1.69978209e-03,  3.44044808e-03,\n",
              "               -2.89280107e-03,  5.09201142e-04, -4.29836626e-04,\n",
              "               -1.88653986e-03, -9.74601251e-04,  6.07332634e-03,\n",
              "                4.03685000e-04, -6.61353231e-04,  2.08185811e-05,\n",
              "               -2.27894401e-03]],\n",
              "     \n",
              "             [[ 1.06983818e-03,  3.04943370e-03,  2.79848720e-03,\n",
              "                8.59202584e-04,  2.35487008e-03,  3.86350602e-03,\n",
              "               -5.02936728e-03,  3.19601008e-04,  6.37328310e-04,\n",
              "               -1.77071313e-03, -1.53440307e-03,  6.16503600e-03,\n",
              "                6.82582031e-04, -4.73065884e-04, -2.92282610e-04,\n",
              "               -7.85905868e-05]],\n",
              "     \n",
              "             [[ 4.38261748e-04,  2.99327215e-03,  7.88566365e-04,\n",
              "                2.47201242e-05,  2.32854951e-03,  2.39719916e-03,\n",
              "               -5.89687331e-03, -4.41182259e-04,  1.07513706e-03,\n",
              "               -1.03774783e-03, -3.09451483e-03,  5.32757118e-03,\n",
              "                4.40885255e-04,  4.18478565e-04, -1.16085913e-03,\n",
              "               -7.35967187e-04]],\n",
              "     \n",
              "             [[-2.88543990e-04,  4.60772216e-03,  7.86329270e-04,\n",
              "                5.00949682e-07,  2.24070810e-03,  2.11548805e-03,\n",
              "               -6.00133464e-03,  1.91354728e-03,  1.26067828e-03,\n",
              "                4.94028092e-04, -1.04709099e-04,  1.81203044e-03,\n",
              "                5.47574949e-04,  1.06105162e-03, -2.64088134e-03,\n",
              "               -8.99355276e-04]],\n",
              "     \n",
              "             [[ 1.02134887e-04,  5.85604832e-03,  7.10962107e-04,\n",
              "               -1.86746113e-03,  2.30373163e-03,  2.77752080e-03,\n",
              "               -6.35959813e-03, -7.10555818e-04,  4.07691300e-03,\n",
              "                2.61772890e-04,  1.47207710e-03,  1.92602514e-04,\n",
              "                6.99150318e-04, -1.41346082e-03, -2.80387164e-03,\n",
              "               -2.09122710e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[ 7.17369781e-04,  1.69137132e-03,  1.43994112e-03,\n",
              "               -3.23556451e-04,  1.76859507e-03,  2.51535466e-03,\n",
              "               -4.39248746e-03,  1.01209749e-04,  3.21246102e-04,\n",
              "               -2.17845896e-03,  1.19511597e-03,  3.99510935e-03,\n",
              "               -7.93229148e-04, -1.19540433e-03, -6.95473747e-04,\n",
              "               -6.18221704e-03]],\n",
              "     \n",
              "             [[ 1.09639228e-03,  1.91119651e-03,  1.90337119e-03,\n",
              "               -4.92034364e-04,  1.80853182e-03,  1.19770737e-03,\n",
              "               -7.11822510e-03,  6.35278120e-04,  6.02800807e-04,\n",
              "               -1.10981870e-03, -7.43359618e-04,  3.61470180e-03,\n",
              "               -8.66549322e-04, -1.90420006e-03, -6.86248459e-05,\n",
              "               -3.30044236e-03]],\n",
              "     \n",
              "             [[-2.29325451e-05,  1.12205755e-03,  2.44274526e-03,\n",
              "                3.76092212e-04,  1.80439674e-03,  2.09183912e-04,\n",
              "               -9.18548927e-03,  1.19185681e-03,  1.44636561e-03,\n",
              "               -6.60017715e-04, -2.02961219e-03,  3.54592572e-03,\n",
              "               -6.04091852e-04, -1.16006727e-03, -2.05886690e-03,\n",
              "               -8.15851614e-04]],\n",
              "     \n",
              "             [[-1.67922262e-04,  1.91549747e-03,  2.93884799e-03,\n",
              "               -9.07139096e-04,  2.26230570e-03, -1.75220557e-05,\n",
              "               -1.12855323e-02,  9.15110402e-04,  8.21084541e-04,\n",
              "               -8.69847281e-05,  1.01887842e-03,  9.50310612e-04,\n",
              "               -6.25683169e-04, -1.90275977e-03, -2.47131288e-03,\n",
              "                6.06529880e-04]],\n",
              "     \n",
              "             [[-4.14940936e-04,  4.16685967e-03,  3.17479204e-03,\n",
              "               -7.07376399e-04,  1.46284839e-03,  6.33068150e-04,\n",
              "               -1.04531404e-02, -2.42429087e-04,  2.71723932e-03,\n",
              "                6.24050008e-05,  7.09235319e-04, -4.33081470e-04,\n",
              "               -2.21650727e-04, -3.59851122e-03, -4.02120082e-03,\n",
              "               -2.95640435e-04]]]], dtype=float32)},\n",
              "    'Conv_1': {'bias': array([ 1.6833404e-10, -6.2878071e-11, -8.9434604e-10, -4.2032230e-10,\n",
              "            -3.3347308e-10,  2.1256097e-10,  1.7294936e-10, -9.6940650e-11,\n",
              "             8.9777248e-11,  3.0720310e-10, -4.1880666e-12, -3.8123962e-10,\n",
              "             2.6087233e-11, -8.9846908e-11,  3.2923600e-10, -2.6533684e-10,\n",
              "             7.7515730e-11, -2.8084637e-11, -9.8188346e-11,  2.0060734e-10,\n",
              "            -6.0148671e-11,  1.2979215e-10, -1.9278119e-10, -8.1351662e-12,\n",
              "            -1.1877860e-12, -9.7488413e-11, -2.5172619e-11, -1.0348088e-09,\n",
              "            -2.6827940e-10,  1.5170598e-10, -4.2758330e-10,  3.8968906e-10],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[ 2.65180046e-04, -1.58748808e-04, -7.65517398e-05, ...,\n",
              "               -1.38355404e-04,  2.18768313e-04, -3.47222194e-05],\n",
              "              [-1.63499237e-04, -2.88425828e-04, -3.79343517e-04, ...,\n",
              "                7.84159056e-05,  7.23809862e-05, -2.29914629e-04],\n",
              "              [ 2.04533790e-04,  7.95665197e-04, -3.68506699e-05, ...,\n",
              "                1.12390017e-05,  6.98538497e-04,  4.05922852e-04],\n",
              "              ...,\n",
              "              [ 1.21325269e-04,  9.98987467e-04, -3.22674983e-04, ...,\n",
              "                7.10227541e-05, -3.66870430e-04,  2.42481983e-04],\n",
              "              [-1.50810316e-04,  1.36578936e-04, -1.13024536e-04, ...,\n",
              "               -3.68598521e-05,  2.71208235e-04, -6.99009615e-05],\n",
              "              [-2.46579493e-05,  4.70872212e-04, -5.49526558e-05, ...,\n",
              "               -2.57760803e-05,  2.27394994e-04,  5.60447574e-04]],\n",
              "     \n",
              "             [[ 9.99519980e-05,  8.78508072e-05,  5.40248448e-06, ...,\n",
              "               -1.44907375e-04,  6.02625252e-04,  1.46625433e-04],\n",
              "              [-2.29845755e-04,  7.47657541e-05, -3.70538008e-04, ...,\n",
              "                1.92358522e-04,  2.45147123e-04,  2.53945211e-04],\n",
              "              [ 1.26309547e-04,  1.34414295e-04, -9.21049214e-05, ...,\n",
              "                1.69229079e-05,  5.49524484e-05, -1.61320233e-04],\n",
              "              ...,\n",
              "              [ 1.16136791e-04,  8.56697443e-04, -8.97426726e-05, ...,\n",
              "               -3.16342703e-05, -3.04014917e-04, -3.34955577e-04],\n",
              "              [-1.53030705e-04, -1.80924762e-04, -5.32274717e-04, ...,\n",
              "               -3.81387290e-05,  2.47882854e-04,  5.65848677e-05],\n",
              "              [ 5.15010033e-04,  9.31412287e-05,  7.06920400e-06, ...,\n",
              "               -2.27483994e-04,  4.38830059e-04, -1.11040936e-05]],\n",
              "     \n",
              "             [[ 6.43997737e-06,  2.02209150e-04,  1.01977261e-04, ...,\n",
              "               -1.50092601e-04,  4.17725038e-04,  1.58478826e-04],\n",
              "              [-7.60889452e-05, -6.69687870e-05,  2.48451455e-04, ...,\n",
              "                7.24714846e-05, -1.99989125e-04, -1.24066617e-04],\n",
              "              [ 2.05952805e-04, -3.20764142e-04, -3.10781325e-04, ...,\n",
              "               -1.18719821e-04, -1.88086269e-04,  3.81325692e-04],\n",
              "              ...,\n",
              "              [ 2.03306612e-04,  9.75052943e-04, -6.67968008e-04, ...,\n",
              "                1.32132584e-04, -2.33074068e-04, -8.88053910e-06],\n",
              "              [-2.58253654e-04, -3.76267999e-04, -5.44821844e-04, ...,\n",
              "               -9.16923018e-05,  2.12021376e-04, -1.09969929e-04],\n",
              "              [ 7.08086664e-05, -2.96231701e-05, -6.47709676e-05, ...,\n",
              "               -6.03940716e-05, -1.11382426e-04,  3.76316137e-04]],\n",
              "     \n",
              "             [[-1.80052266e-05,  3.11816584e-05, -3.84119252e-04, ...,\n",
              "                1.01265206e-04,  3.89911846e-04,  6.43840758e-06],\n",
              "              [ 2.41836402e-04,  3.07005306e-04, -4.58410359e-04, ...,\n",
              "                2.57401058e-04, -5.91273245e-04,  1.38837131e-04],\n",
              "              [-2.16405620e-04, -5.81283704e-04, -5.52660669e-04, ...,\n",
              "               -1.54683599e-04,  1.96801135e-04,  4.85336321e-04],\n",
              "              ...,\n",
              "              [ 1.14720962e-04,  2.83418631e-04, -2.81256624e-04, ...,\n",
              "               -3.45654225e-05, -1.48446998e-04, -1.31723718e-04],\n",
              "              [-2.24870135e-04, -6.80025405e-05,  1.52861219e-04, ...,\n",
              "               -1.13608541e-04,  1.63753575e-04, -2.36240405e-04],\n",
              "              [ 3.35107150e-04, -4.93427797e-04, -4.00470250e-04, ...,\n",
              "               -1.50999986e-05,  5.76562597e-05, -6.27599948e-05]],\n",
              "     \n",
              "             [[ 7.13637637e-05,  1.17138508e-04, -4.58287715e-04, ...,\n",
              "               -2.99455187e-05,  1.39277385e-04,  1.15127536e-04],\n",
              "              [-1.10418856e-04,  2.72324280e-04, -9.03263572e-05, ...,\n",
              "                2.28339952e-04, -2.68911215e-04, -2.44501163e-04],\n",
              "              [ 1.54584501e-04,  5.04062104e-04,  1.91984989e-04, ...,\n",
              "               -9.36427750e-05, -7.90263875e-05,  1.19567121e-05],\n",
              "              ...,\n",
              "              [-8.65850889e-05, -9.08394650e-05,  3.50212358e-05, ...,\n",
              "               -4.03087688e-06, -3.43029169e-05,  2.64024740e-04],\n",
              "              [-1.08902903e-04,  1.05102343e-04,  1.47360843e-04, ...,\n",
              "               -8.58535350e-05, -8.51812074e-05, -1.37923838e-04],\n",
              "              [ 3.38473066e-04,  2.10120925e-06, -3.25624918e-04, ...,\n",
              "               -3.76124663e-05, -2.47129472e-04,  4.52382315e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[-5.18782326e-05,  5.04504715e-06, -1.63804114e-04, ...,\n",
              "               -7.93366198e-05,  1.54454210e-05,  4.58485796e-04],\n",
              "              [ 6.35700198e-05,  3.69854213e-04, -1.03871935e-04, ...,\n",
              "                1.12186593e-04,  1.02979050e-03, -1.11079797e-04],\n",
              "              [-1.89305531e-04,  4.30047403e-05, -3.19992942e-05, ...,\n",
              "                1.29746215e-04, -4.24646896e-05,  3.26832727e-04],\n",
              "              ...,\n",
              "              [ 1.53593923e-04,  2.40229390e-04,  1.03397742e-05, ...,\n",
              "               -6.00188541e-05, -3.06362199e-04, -2.90820492e-04],\n",
              "              [ 3.30803712e-04,  1.40129152e-04, -6.39469712e-04, ...,\n",
              "               -1.36608389e-04,  3.15662124e-04,  2.54645041e-04],\n",
              "              [-9.71967529e-05,  1.55744303e-04,  9.81413614e-05, ...,\n",
              "               -2.02167157e-05, -5.56196574e-06,  2.55753519e-04]],\n",
              "     \n",
              "             [[-8.48815398e-05, -1.60873649e-04, -2.89056712e-04, ...,\n",
              "               -9.00360537e-05,  4.11963352e-04,  3.28942202e-04],\n",
              "              [ 1.15508570e-04,  6.95192284e-05,  2.42685255e-05, ...,\n",
              "               -3.71903552e-05, -2.10544487e-04, -5.25360010e-05],\n",
              "              [ 4.34820686e-04,  4.15774120e-05,  1.16700343e-04, ...,\n",
              "               -1.62397337e-04,  4.68295366e-05, -5.23099734e-05],\n",
              "              ...,\n",
              "              [-7.11073953e-05,  3.23338842e-04,  7.51097687e-05, ...,\n",
              "               -6.06883223e-05, -1.35862836e-04, -1.80344796e-04],\n",
              "              [ 6.24761567e-04,  2.96585669e-04, -4.51960310e-04, ...,\n",
              "                2.56789383e-04,  3.43549822e-04,  4.38903604e-04],\n",
              "              [-1.86446181e-04,  1.94605265e-04,  2.65515642e-04, ...,\n",
              "                4.15844639e-04,  4.96856053e-04,  2.78200459e-05]],\n",
              "     \n",
              "             [[-1.13287475e-04, -2.03540621e-05, -1.21618767e-04, ...,\n",
              "                9.87497333e-05, -9.73453643e-06, -9.10045856e-07],\n",
              "              [ 3.69070505e-04, -1.84672797e-04, -4.48773702e-04, ...,\n",
              "                6.30201976e-05,  7.05638449e-05,  6.65805070e-04],\n",
              "              [ 3.61572427e-04, -1.12230213e-04, -2.82090972e-04, ...,\n",
              "                2.40201189e-05, -1.20400087e-04, -9.66462685e-05],\n",
              "              ...,\n",
              "              [-2.95854989e-04,  4.82371222e-04, -1.52617664e-04, ...,\n",
              "               -1.71662068e-05, -9.57882439e-05, -3.19776911e-04],\n",
              "              [ 7.33862515e-04,  3.36096506e-04, -2.73012440e-04, ...,\n",
              "                4.34905465e-04,  1.52350738e-04,  5.43636677e-04],\n",
              "              [ 5.32674836e-04, -3.05522466e-04,  2.28793710e-04, ...,\n",
              "               -1.70645973e-04, -4.74395019e-05, -2.11634528e-04]],\n",
              "     \n",
              "             [[-3.39200429e-04, -1.81415220e-04, -2.87208997e-04, ...,\n",
              "                1.01514088e-04,  4.52427448e-05,  4.44584148e-05],\n",
              "              [-2.10817394e-04, -2.11603488e-04, -7.15095084e-04, ...,\n",
              "               -9.57893571e-05,  5.56620507e-05, -6.32646406e-05],\n",
              "              [ 3.16935562e-04,  3.86969245e-04,  2.97371647e-04, ...,\n",
              "               -1.70360639e-04, -1.43950107e-04,  2.30268764e-04],\n",
              "              ...,\n",
              "              [-1.44799589e-04,  3.18534760e-04,  1.67194317e-04, ...,\n",
              "               -7.94512162e-05, -8.83119901e-06, -1.95113127e-04],\n",
              "              [ 5.78768493e-04,  9.79002361e-05, -1.98232941e-04, ...,\n",
              "                3.02420463e-04,  1.56482929e-05,  4.15915449e-04],\n",
              "              [ 2.14205618e-04, -4.89884987e-04, -1.27113599e-04, ...,\n",
              "                1.00990495e-04,  7.13038462e-05, -2.43296992e-04]],\n",
              "     \n",
              "             [[-8.83791072e-05,  2.01433868e-04, -6.83841179e-04, ...,\n",
              "               -8.72450910e-05, -8.60171858e-05,  3.22297565e-04],\n",
              "              [ 4.28768108e-04,  2.04850614e-04, -2.26252232e-04, ...,\n",
              "                2.29568905e-05, -3.76320895e-05,  8.48763302e-05],\n",
              "              [ 4.90282080e-04, -7.49875384e-04, -1.59536023e-04, ...,\n",
              "                9.98784672e-05,  2.28838311e-04,  4.55915637e-04],\n",
              "              ...,\n",
              "              [ 2.85972856e-05, -6.99395532e-05,  1.58955008e-04, ...,\n",
              "               -1.06599677e-04,  7.70006300e-05, -1.25886174e-04],\n",
              "              [-4.51036613e-06,  2.52237951e-04, -2.66127754e-04, ...,\n",
              "                8.19615670e-05, -2.15669366e-04,  2.82778667e-04],\n",
              "              [-6.13977172e-05,  1.48849271e-04, -5.10077807e-04, ...,\n",
              "               -4.36788483e-04,  4.50217689e-04,  1.01831730e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[ 4.08642576e-04, -7.71950727e-05,  1.22419835e-04, ...,\n",
              "                2.43673232e-04,  5.13271603e-04,  2.25442956e-04],\n",
              "              [ 9.45465436e-05,  1.24460566e-04, -1.22099853e-04, ...,\n",
              "               -2.31189870e-06,  3.96248011e-04,  3.84611340e-04],\n",
              "              [-7.20905882e-05,  3.90149740e-04, -1.21951434e-05, ...,\n",
              "               -2.07648583e-04,  3.22536012e-04,  8.07479410e-07],\n",
              "              ...,\n",
              "              [ 4.70696395e-05,  3.95116396e-04, -4.46754799e-04, ...,\n",
              "               -4.26487415e-04, -1.37604366e-06, -1.12302012e-04],\n",
              "              [ 1.48897394e-04,  1.52900902e-05, -3.29992326e-04, ...,\n",
              "                1.61859876e-04,  6.63097366e-04, -5.91918724e-05],\n",
              "              [ 2.12001069e-05,  4.92469349e-04,  6.03380395e-05, ...,\n",
              "               -4.87709010e-04,  7.36611837e-05, -5.19848472e-05]],\n",
              "     \n",
              "             [[ 2.85595830e-04, -1.54454159e-04, -1.94159846e-04, ...,\n",
              "                2.42987910e-04,  2.96054553e-04,  4.11949877e-04],\n",
              "              [ 7.15494854e-04, -2.61184934e-04, -1.00422272e-04, ...,\n",
              "               -1.43988043e-04,  4.88353427e-04, -4.44041158e-04],\n",
              "              [-2.32302351e-04, -4.36013434e-05,  2.48924607e-05, ...,\n",
              "                1.35396214e-04, -6.66283595e-05,  4.00961988e-04],\n",
              "              ...,\n",
              "              [ 2.00607959e-04,  5.87661285e-04, -2.29466314e-04, ...,\n",
              "               -7.03109545e-05, -1.02218350e-04, -1.10729365e-04],\n",
              "              [ 5.72392455e-05, -1.44624846e-05, -2.94497178e-04, ...,\n",
              "                9.87954118e-05,  5.27000986e-04, -7.69374237e-05],\n",
              "              [-1.17887830e-05, -1.14024060e-05,  6.01678767e-05, ...,\n",
              "                1.23160600e-04, -5.95156162e-05,  1.49196509e-04]],\n",
              "     \n",
              "             [[ 2.26597811e-04, -2.31745391e-04, -1.72392509e-04, ...,\n",
              "                3.45644017e-04,  2.80312292e-04, -3.48110072e-04],\n",
              "              [-2.80001696e-04,  1.53586021e-04, -2.63112481e-04, ...,\n",
              "               -1.02565144e-04,  4.95593049e-05,  6.88726723e-04],\n",
              "              [ 2.42191425e-04,  1.93237051e-04, -3.99693963e-05, ...,\n",
              "               -1.39817603e-05,  1.43253346e-04, -4.64644909e-05],\n",
              "              ...,\n",
              "              [ 1.59307965e-04,  4.05334140e-04, -9.23877014e-05, ...,\n",
              "               -3.69149726e-04, -9.81054691e-05,  7.90680861e-05],\n",
              "              [-1.58165873e-04,  6.72968163e-05, -2.20969465e-04, ...,\n",
              "               -3.25302717e-05,  7.06752879e-04, -5.48198150e-05],\n",
              "              [ 3.57915706e-04, -2.22995805e-04,  3.10060947e-04, ...,\n",
              "               -2.85272894e-04, -5.60154876e-05, -2.93513469e-04]],\n",
              "     \n",
              "             [[ 6.20724750e-05,  2.46390591e-05, -3.76819407e-05, ...,\n",
              "               -1.25866107e-04, -2.17374050e-04,  1.71410109e-04],\n",
              "              [ 7.95459258e-04, -1.99404763e-04,  1.18164258e-04, ...,\n",
              "               -2.30268255e-04,  1.04949824e-04,  4.62523487e-04],\n",
              "              [ 5.06938610e-04, -2.33641680e-04,  5.14042622e-05, ...,\n",
              "                2.01389121e-05,  1.97199712e-04,  6.01934553e-05],\n",
              "              ...,\n",
              "              [-6.32087322e-05,  1.65054749e-04, -8.54613827e-05, ...,\n",
              "               -1.98512193e-04,  1.10675108e-04, -2.66220450e-05],\n",
              "              [-3.38908198e-04,  2.96076149e-04, -3.63144733e-04, ...,\n",
              "               -1.46827515e-04,  6.01085019e-04,  2.10582366e-05],\n",
              "              [ 5.17729670e-04, -1.95933098e-04,  2.76757055e-06, ...,\n",
              "               -2.91475269e-04, -1.29073829e-04, -2.92928657e-04]],\n",
              "     \n",
              "             [[ 1.18547876e-04, -4.82274882e-05, -3.75596399e-04, ...,\n",
              "               -2.05693621e-04, -1.74615940e-04,  3.96647200e-04],\n",
              "              [ 1.12499278e-04, -4.24244790e-05, -5.57090680e-05, ...,\n",
              "               -8.15243911e-05, -8.45086615e-05,  2.36208914e-04],\n",
              "              [-4.33543813e-04,  6.59635814e-04,  2.54847895e-04, ...,\n",
              "               -6.74877592e-05,  1.45080674e-04,  7.33515044e-05],\n",
              "              ...,\n",
              "              [ 2.71231402e-05,  2.47074029e-04,  5.34318315e-05, ...,\n",
              "               -1.60735726e-04, -6.34449534e-05, -1.09377535e-04],\n",
              "              [ 1.83025084e-04,  1.82753021e-04, -2.27552635e-04, ...,\n",
              "                4.09729982e-05,  6.24609413e-04,  1.89607526e-04],\n",
              "              [-8.99556617e-05,  4.23338817e-04,  4.28921456e-04, ...,\n",
              "               -4.08097869e-04,  1.15954055e-04,  5.30838093e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[ 1.58783572e-04,  1.29858206e-04,  4.61305171e-05, ...,\n",
              "                1.69565828e-04,  1.23154881e-04,  1.11788628e-04],\n",
              "              [ 2.25827825e-04,  3.47984838e-04, -4.96189226e-04, ...,\n",
              "               -2.52993806e-04,  4.13090194e-04,  5.20556467e-04],\n",
              "              [ 4.43298486e-04,  2.68617907e-04,  1.23226346e-04, ...,\n",
              "               -1.08556480e-04,  3.19560786e-05, -9.02212632e-05],\n",
              "              ...,\n",
              "              [ 2.91490054e-04, -5.73075667e-04, -5.30100369e-04, ...,\n",
              "                1.92515290e-04, -1.71936597e-04, -4.95878921e-05],\n",
              "              [ 3.16339720e-04,  1.87182843e-04, -3.36626108e-04, ...,\n",
              "               -4.06117470e-05, -1.94361171e-04, -8.80751613e-05],\n",
              "              [-9.27299043e-05,  9.75651637e-05,  4.70470404e-05, ...,\n",
              "               -5.85299276e-05, -7.14938069e-05,  1.46507346e-05]],\n",
              "     \n",
              "             [[-9.07502545e-06,  9.37773002e-05, -5.51674457e-04, ...,\n",
              "                3.57969606e-04,  6.55593351e-04,  2.28378995e-04],\n",
              "              [-4.04851926e-05, -2.28632576e-04, -2.52940372e-05, ...,\n",
              "               -1.09322595e-04,  4.59987641e-05,  2.21229238e-05],\n",
              "              [ 1.69889914e-04,  1.56360366e-05, -7.82044226e-05, ...,\n",
              "               -1.17098760e-04, -5.60033717e-04,  4.05705578e-05],\n",
              "              ...,\n",
              "              [ 1.73520806e-04, -2.15005173e-04,  6.02458749e-06, ...,\n",
              "               -4.45166952e-05, -1.32497116e-05, -2.98469269e-04],\n",
              "              [ 3.20475374e-04,  2.30372796e-04, -2.58548534e-04, ...,\n",
              "                2.08676211e-05, -1.63055622e-04,  5.50163059e-05],\n",
              "              [-5.33191196e-05, -1.52751367e-04, -2.48567244e-06, ...,\n",
              "                9.54969801e-05,  8.56761108e-05, -4.30031432e-05]],\n",
              "     \n",
              "             [[ 1.28648448e-04, -7.79822512e-05, -3.08168674e-04, ...,\n",
              "               -1.11130503e-05,  2.08851212e-04, -1.12577975e-04],\n",
              "              [ 8.82086693e-04, -5.57523570e-04, -1.34431015e-04, ...,\n",
              "               -1.55372501e-04,  6.05049543e-04, -1.02429374e-04],\n",
              "              [ 8.66940478e-04,  1.36165909e-04,  2.27040109e-05, ...,\n",
              "                2.63500260e-04, -5.85156959e-05,  1.60158961e-05],\n",
              "              ...,\n",
              "              [-4.61916497e-04, -8.43941380e-05, -3.01542386e-05, ...,\n",
              "               -2.65487852e-05,  2.69298180e-05,  2.53203267e-04],\n",
              "              [ 1.39061565e-04,  2.38228095e-05, -6.43177889e-04, ...,\n",
              "                1.43609956e-04,  1.44081248e-04,  9.49710957e-05],\n",
              "              [ 3.75706557e-04, -4.06305335e-04,  6.74381081e-05, ...,\n",
              "               -2.88049894e-04,  3.67025059e-05, -1.02114805e-04]],\n",
              "     \n",
              "             [[-6.37027406e-05, -3.74368043e-04, -6.16456207e-04, ...,\n",
              "               -2.11878767e-04, -2.93404246e-05,  3.68092587e-04],\n",
              "              [-5.27194527e-04, -4.45285725e-04,  6.31441799e-05, ...,\n",
              "               -1.62296463e-04, -6.72743918e-05,  3.03120032e-04],\n",
              "              [ 1.89721453e-04,  1.09776083e-04,  7.70942934e-05, ...,\n",
              "               -3.96040909e-04, -3.11696385e-05,  4.23147540e-05],\n",
              "              ...,\n",
              "              [-2.71365978e-04,  3.49016103e-04, -4.25016515e-05, ...,\n",
              "                7.10157037e-05,  5.23951196e-04,  3.59463331e-04],\n",
              "              [ 1.09962399e-04, -2.13607433e-04, -3.84937273e-04, ...,\n",
              "                2.23276555e-04,  1.57104834e-04,  1.39997501e-05],\n",
              "              [ 5.54146318e-05,  1.39950440e-04,  1.70257554e-04, ...,\n",
              "               -5.24368079e-04, -2.09902777e-04, -3.42912361e-04]],\n",
              "     \n",
              "             [[-1.04663719e-04, -3.21642554e-04, -5.49714896e-04, ...,\n",
              "               -8.35694373e-05,  2.53943494e-04,  3.97505093e-04],\n",
              "              [ 1.86109508e-04,  5.03748714e-04,  1.19068813e-04, ...,\n",
              "               -6.34352255e-05,  5.47025120e-04,  3.51667753e-04],\n",
              "              [ 5.76688675e-04,  3.08276532e-04, -1.81762603e-04, ...,\n",
              "                1.27267442e-04,  1.90171282e-04,  1.34892798e-06],\n",
              "              ...,\n",
              "              [ 2.93688936e-04,  1.27402498e-04, -9.49770110e-05, ...,\n",
              "                1.24656493e-04,  2.56061088e-04,  1.78371876e-04],\n",
              "              [-3.26998328e-04, -2.97222665e-04, -2.80270644e-04, ...,\n",
              "                2.76813400e-04,  6.92257017e-05, -7.04839040e-05],\n",
              "              [ 3.23777815e-04,  3.48883332e-04,  1.64421552e-04, ...,\n",
              "               -5.34887542e-04,  5.56760991e-04,  2.69383658e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[ 3.35284349e-05,  5.83102974e-07, -4.86237637e-04, ...,\n",
              "               -8.79198633e-05,  3.07771435e-04,  2.16757166e-04],\n",
              "              [ 5.43992093e-04,  6.24357563e-06, -2.37118220e-04, ...,\n",
              "               -2.54089944e-04,  3.80044687e-04,  4.37676063e-04],\n",
              "              [-2.15629960e-04,  4.30949614e-04, -3.21664229e-05, ...,\n",
              "               -5.46277224e-05,  1.67727903e-05,  9.13465847e-05],\n",
              "              ...,\n",
              "              [ 4.81250026e-06,  5.95011457e-04, -3.63009676e-05, ...,\n",
              "                3.94741539e-04, -3.38662328e-04,  2.62201356e-04],\n",
              "              [ 6.86822968e-05,  4.06465879e-05, -3.05528054e-04, ...,\n",
              "               -3.12692064e-05,  6.95644412e-04,  3.03944762e-05],\n",
              "              [-7.83935757e-05,  2.14583459e-04,  3.00274114e-05, ...,\n",
              "               -2.57378651e-05, -3.63482250e-05,  3.94498711e-05]],\n",
              "     \n",
              "             [[ 1.42611476e-04, -3.32841330e-04, -3.83944425e-04, ...,\n",
              "               -3.70847643e-04,  4.33624286e-04,  3.12048855e-04],\n",
              "              [ 2.85010727e-04,  2.19734320e-05, -3.47685622e-04, ...,\n",
              "                1.53126224e-04, -1.37208510e-04, -1.20888544e-04],\n",
              "              [ 4.08394408e-04,  2.94701858e-05, -1.50876986e-06, ...,\n",
              "                1.46594641e-04,  1.34236452e-05, -5.25834512e-05],\n",
              "              ...,\n",
              "              [-8.99713268e-05,  4.80330782e-04,  6.49777940e-05, ...,\n",
              "               -6.88147993e-05, -4.21753066e-04,  6.30933209e-06],\n",
              "              [ 2.10485625e-04,  2.21073729e-04, -1.95356421e-04, ...,\n",
              "                1.67904655e-05,  5.53763879e-04,  1.37377472e-04],\n",
              "              [ 7.12296169e-05, -3.36095487e-04,  4.05699247e-05, ...,\n",
              "               -5.68484247e-05,  7.22257537e-05, -5.19562018e-05]],\n",
              "     \n",
              "             [[-1.81407668e-05, -2.06030993e-04, -6.93643116e-04, ...,\n",
              "               -2.55800900e-04, -1.69752253e-04,  2.05778837e-04],\n",
              "              [ 4.93684493e-04,  2.91508914e-04, -1.89470142e-04, ...,\n",
              "                1.65192789e-04, -1.84516059e-04,  6.65977539e-04],\n",
              "              [-9.87883686e-06, -7.31149385e-06, -5.24177267e-05, ...,\n",
              "               -2.09369973e-04,  8.04626761e-05,  1.07269778e-04],\n",
              "              ...,\n",
              "              [-2.17980836e-04,  2.81202549e-04, -2.53918377e-04, ...,\n",
              "               -4.59800722e-05, -3.12798074e-04,  3.04890767e-04],\n",
              "              [ 2.82170979e-04,  7.10579858e-04, -1.24606056e-04, ...,\n",
              "               -2.18124769e-04,  5.73777128e-04,  2.30041333e-05],\n",
              "              [ 1.06333711e-04, -8.28604316e-05, -2.83478090e-04, ...,\n",
              "               -2.34608873e-04, -1.83090800e-04, -9.19354352e-06]],\n",
              "     \n",
              "             [[-1.84245204e-04, -1.70761981e-04, -4.49031359e-04, ...,\n",
              "               -2.42324080e-04, -2.37617405e-05,  1.61793840e-04],\n",
              "              [ 1.01771287e-03,  1.18201016e-04, -1.52463093e-04, ...,\n",
              "               -1.27067935e-04, -3.04890593e-04, -8.27634358e-05],\n",
              "              [ 4.19860240e-04,  1.00601741e-04, -6.46690660e-06, ...,\n",
              "               -1.01076497e-04, -8.68151692e-05, -1.85349432e-04],\n",
              "              ...,\n",
              "              [-1.86259116e-04,  2.73823971e-05, -1.74067784e-04, ...,\n",
              "                1.89333041e-05, -1.14873757e-04, -2.26058037e-04],\n",
              "              [ 3.10033094e-04,  6.03427354e-04, -3.01974098e-04, ...,\n",
              "               -7.84345088e-04,  3.63066385e-04,  1.14594208e-04],\n",
              "              [ 6.21634186e-04,  5.11347898e-05,  3.02486675e-04, ...,\n",
              "                2.48995202e-04, -1.39704352e-04, -3.19267623e-04]],\n",
              "     \n",
              "             [[-2.28739591e-05,  3.32264288e-04, -2.95899023e-04, ...,\n",
              "               -1.74266781e-04,  2.60023575e-04,  2.43017625e-04],\n",
              "              [-1.73280569e-04,  4.96249122e-04, -2.37474422e-04, ...,\n",
              "               -6.47584748e-05, -2.19667563e-04,  3.77228662e-06],\n",
              "              [-1.57551578e-04,  8.22974325e-05, -6.49462600e-05, ...,\n",
              "               -1.39177413e-04,  2.35752101e-04, -4.14299575e-05],\n",
              "              ...,\n",
              "              [-3.32510128e-04, -4.06030995e-05, -1.42801538e-04, ...,\n",
              "                8.27382828e-05,  6.21649888e-05, -1.90370745e-04],\n",
              "              [ 3.13787168e-04,  5.26859425e-04, -3.58685327e-04, ...,\n",
              "               -1.00001786e-03,  4.89010883e-04,  9.49606037e-05],\n",
              "              [ 5.14475454e-04, -1.49429892e-04,  3.94167437e-04, ...,\n",
              "               -1.84563178e-05, -2.59016961e-04, -1.80607356e-04]]]],\n",
              "           dtype=float32)},\n",
              "    'Dense_0': {'bias': array([-2.4973224e-03,  6.7123416e-04,  2.6150723e-03, -2.0085187e-03,\n",
              "            -1.1565093e-03,  3.6970475e-03, -7.1615788e-05, -6.9848995e-04,\n",
              "            -1.2473250e-03,  6.9642608e-04], dtype=float32),\n",
              "     'kernel': array([[-1.48731610e-03, -2.77985018e-05,  4.16548783e-03,\n",
              "             -7.68440659e-04, -3.98487377e-04,  1.91142107e-03,\n",
              "             -2.69848853e-04, -1.00215699e-03, -2.21613655e-03,\n",
              "              9.32750991e-05],\n",
              "            [-1.62777491e-03,  7.54272332e-04,  2.03767628e-03,\n",
              "             -1.47845177e-03, -1.16145436e-03,  2.64588953e-03,\n",
              "              3.71136237e-04, -2.29287194e-03,  1.31406006e-04,\n",
              "              6.20172359e-04],\n",
              "            [-6.86337124e-04, -4.13556438e-04,  4.40467149e-03,\n",
              "              1.33405416e-03, -1.33484066e-03,  9.57296463e-04,\n",
              "              1.08275701e-04, -1.15025532e-03, -2.85982527e-03,\n",
              "             -3.59483762e-04],\n",
              "            [-2.47209158e-04,  3.67065251e-04,  3.26095941e-03,\n",
              "             -1.22702273e-03, -6.86115294e-04,  3.68906581e-03,\n",
              "             -3.25533852e-04, -3.42037971e-03, -1.74347265e-03,\n",
              "              3.32639436e-04],\n",
              "            [ 2.99240619e-05, -5.00901253e-04,  3.36248707e-03,\n",
              "             -2.52732029e-03, -1.06302358e-03,  2.20118277e-03,\n",
              "              9.35666249e-05, -2.04662560e-03,  7.80069167e-05,\n",
              "              3.72703042e-04],\n",
              "            [-1.79705117e-03,  3.66690540e-04,  2.73697195e-04,\n",
              "             -2.62281028e-05, -4.90776147e-04,  2.02716421e-03,\n",
              "             -8.56634811e-04,  1.42198650e-03, -1.40249985e-03,\n",
              "              4.83649143e-04],\n",
              "            [-1.40690128e-03, -5.80933229e-05,  5.26450993e-03,\n",
              "             -1.15257339e-03,  8.56866973e-05,  3.01267323e-03,\n",
              "              8.86828857e-05, -3.65723460e-03, -2.33395281e-03,\n",
              "              1.57201255e-04],\n",
              "            [-9.58415098e-04,  4.29068838e-04,  2.22653197e-03,\n",
              "             -2.71549099e-03, -7.23558187e-04,  3.13881878e-03,\n",
              "             -1.12293975e-03, -3.23026150e-04, -9.33050935e-04,\n",
              "              9.82059049e-04],\n",
              "            [-1.78874715e-03,  5.88588591e-04,  3.83691909e-03,\n",
              "             -1.22039369e-03, -6.99158059e-04,  2.51330435e-03,\n",
              "             -8.44190770e-04, -9.85682200e-05, -2.79339217e-03,\n",
              "              5.05635981e-04],\n",
              "            [-1.49224629e-03,  5.69712429e-04,  1.50130573e-03,\n",
              "             -2.16791173e-03, -5.66096103e-04,  3.83969722e-03,\n",
              "             -5.15992113e-04, -1.47135041e-04, -4.34558810e-04,\n",
              "             -5.86775481e-04],\n",
              "            [-2.22089561e-03,  2.61654292e-04,  1.71519152e-03,\n",
              "              5.99833787e-04, -5.46647585e-04,  1.44127384e-03,\n",
              "              1.41463635e-04, -1.72063359e-03, -3.92447662e-04,\n",
              "              7.21208286e-04],\n",
              "            [-9.85085964e-04,  3.85559892e-04,  1.97384786e-03,\n",
              "             -2.12855008e-03, -5.65390976e-04,  2.01515248e-03,\n",
              "             -1.46155362e-04, -1.85025303e-04, -4.23405669e-04,\n",
              "              5.90516429e-05],\n",
              "            [-2.44088448e-03, -3.09987023e-04,  1.23459799e-03,\n",
              "              3.75277188e-04, -6.46382105e-04,  1.32017594e-03,\n",
              "              3.24573950e-04,  5.80878987e-06, -1.19220116e-04,\n",
              "              2.56039231e-04],\n",
              "            [-1.54703110e-03,  2.56817701e-04,  9.11945361e-04,\n",
              "             -2.13196807e-04, -2.58358719e-04,  1.11243920e-03,\n",
              "             -1.40286924e-04,  5.09927340e-04, -6.78396143e-04,\n",
              "              4.61389136e-05],\n",
              "            [-7.68614002e-04, -8.61590088e-05,  1.26642885e-03,\n",
              "             -1.79284706e-03, -6.56416814e-04,  2.69785197e-03,\n",
              "              1.72224740e-04,  3.46675050e-04, -1.23411464e-03,\n",
              "              5.49707911e-05],\n",
              "            [-1.28161197e-03,  9.90264234e-05,  1.64454326e-03,\n",
              "             -1.14600116e-03, -1.31412398e-03,  2.61259777e-03,\n",
              "             -4.90985462e-04, -2.98799889e-04, -6.67527318e-04,\n",
              "              8.42879526e-04],\n",
              "            [-2.99899676e-03,  1.01122621e-03,  2.22213939e-03,\n",
              "             -1.47662079e-03, -1.95053010e-03,  2.89933011e-03,\n",
              "             -4.19334712e-04, -2.34769177e-04,  2.98903586e-04,\n",
              "              6.48648827e-04],\n",
              "            [-5.68731106e-04, -3.62690407e-05,  3.00736679e-03,\n",
              "             -2.81353178e-03, -3.12945363e-03,  4.42836061e-03,\n",
              "              2.28328499e-04, -1.09879533e-04, -1.65555021e-03,\n",
              "              6.49358379e-04],\n",
              "            [-1.04495906e-03,  1.14828697e-04,  1.39649527e-03,\n",
              "             -1.70309376e-03, -1.93435140e-03,  2.71155848e-03,\n",
              "             -2.80452718e-04,  4.18123673e-04, -8.60053638e-04,\n",
              "              1.18190143e-03],\n",
              "            [-3.13145312e-04,  4.14907903e-04,  2.02818122e-03,\n",
              "              2.69656884e-05, -9.76303883e-04,  4.10453998e-04,\n",
              "             -3.91295209e-04, -4.00419958e-04, -1.10643567e-03,\n",
              "              3.07090522e-04],\n",
              "            [-8.07978970e-04,  1.63070648e-03, -8.90036579e-04,\n",
              "             -1.10395800e-03, -1.84089283e-03,  2.39393534e-03,\n",
              "             -9.55584634e-04,  9.71515779e-04, -2.30862759e-04,\n",
              "              8.33152328e-04],\n",
              "            [-1.26279378e-03,  4.41962620e-05,  1.78748858e-03,\n",
              "             -9.22488922e-04, -1.10592763e-03,  2.25606025e-03,\n",
              "             -5.23395895e-04,  1.15729083e-04, -1.27722847e-03,\n",
              "              8.88357754e-04],\n",
              "            [-1.96867529e-03,  9.73261369e-04,  1.55687798e-03,\n",
              "             -1.45418709e-03, -2.84072303e-04,  1.85476313e-03,\n",
              "              6.29440940e-04, -6.11223688e-04, -1.70006533e-04,\n",
              "             -5.26177522e-04],\n",
              "            [-3.12658888e-03,  6.96551229e-04,  2.88152427e-04,\n",
              "             -9.32161347e-04, -8.88012437e-06,  2.06604856e-03,\n",
              "              7.70174665e-04,  4.17361473e-04, -3.25931993e-04,\n",
              "              1.55270594e-04],\n",
              "            [-1.05083047e-03,  2.33270708e-04,  2.19589635e-03,\n",
              "             -3.13130347e-03,  7.35043068e-05,  3.26320436e-03,\n",
              "              2.20186965e-04, -1.39880603e-05, -1.48027600e-03,\n",
              "             -3.09665658e-04],\n",
              "            [-6.93936192e-04,  4.10693057e-04,  1.81289716e-03,\n",
              "             -3.20841908e-03, -1.84890896e-03,  4.95979004e-03,\n",
              "             -6.88030195e-05, -8.79705651e-04,  1.78752758e-04,\n",
              "             -6.62361737e-04],\n",
              "            [-1.12702255e-03, -2.28004705e-04,  1.28644984e-03,\n",
              "             -1.32495805e-03, -1.38475763e-04,  2.49337917e-03,\n",
              "              5.99985942e-04, -7.44629360e-04, -5.28421660e-04,\n",
              "             -2.88301526e-04],\n",
              "            [-7.67885824e-04,  5.17686596e-04,  3.42002977e-03,\n",
              "             -1.10946456e-03, -3.48728237e-04,  1.20564899e-03,\n",
              "             -1.53049652e-04, -2.01125583e-03, -1.13708538e-03,\n",
              "              3.84105253e-04],\n",
              "            [-2.59864540e-03,  4.06491890e-04,  1.30082283e-03,\n",
              "             -9.43018822e-04, -1.14628361e-04,  1.56978914e-03,\n",
              "              4.95719432e-04,  1.98360969e-04, -7.52196065e-05,\n",
              "             -2.39672925e-04],\n",
              "            [-2.60437909e-03, -3.76146054e-04,  2.17644777e-03,\n",
              "              3.27274611e-04, -6.00389147e-04,  1.51273923e-03,\n",
              "              4.57231654e-05,  9.48567758e-05, -1.23452232e-03,\n",
              "              6.58391567e-04],\n",
              "            [-1.67279947e-03,  9.29660455e-05,  1.58472569e-03,\n",
              "             -8.06469470e-04, -2.09968319e-04,  1.16490154e-03,\n",
              "             -1.26735496e-04, -9.15755576e-04,  5.92704804e-04,\n",
              "              2.96429003e-04],\n",
              "            [-8.97705497e-04, -3.16556252e-05, -4.28597385e-04,\n",
              "             -4.83497570e-04,  1.11538247e-05,  2.69695814e-03,\n",
              "              2.43210976e-04, -5.13377716e-04, -1.01780356e-03,\n",
              "              4.21313627e-04]], dtype=float32)}},\n",
              "   'nu': {'BatchNorm_0': {'bias': array([0.00101139, 0.00073047, 0.0005574 , 0.00392027, 0.00126213,\n",
              "            0.00135049, 0.00178212, 0.00649406, 0.00089966, 0.00162631,\n",
              "            0.00140204, 0.00443221, 0.00142618, 0.00380787, 0.00134232,\n",
              "            0.001241  ], dtype=float32),\n",
              "     'scale': array([0.005206  , 0.00545367, 0.00071883, 0.00151108, 0.00142804,\n",
              "            0.00117757, 0.0017718 , 0.0031161 , 0.00336226, 0.00132863,\n",
              "            0.00442754, 0.00228052, 0.00558872, 0.00159105, 0.00482422,\n",
              "            0.00110281], dtype=float32)},\n",
              "    'BatchNorm_1': {'bias': array([1.0130052e-04, 8.6864165e-05, 9.1456022e-05, 7.1520648e-05,\n",
              "            9.8731631e-05, 1.0031465e-04, 9.1163311e-05, 1.2768224e-04,\n",
              "            8.5281543e-05, 8.5903295e-05, 1.0165682e-04, 6.2871150e-05,\n",
              "            8.8750174e-05, 8.9115434e-05, 9.5840369e-05, 9.8070464e-05,\n",
              "            1.0910749e-04, 6.8171772e-05, 9.8025012e-05, 7.6197102e-05,\n",
              "            7.8218480e-05, 8.6962289e-05, 6.4860164e-05, 7.6427299e-05,\n",
              "            8.9037843e-05, 7.7504548e-05, 1.0445524e-04, 9.7961376e-05,\n",
              "            9.4355295e-05, 9.1443973e-05, 8.7844826e-05, 9.1353802e-05],\n",
              "           dtype=float32),\n",
              "     'scale': array([1.14510993e-04, 7.08603256e-05, 8.45457835e-05, 6.42467567e-05,\n",
              "            9.40625177e-05, 7.79249531e-05, 8.37739863e-05, 1.08421671e-04,\n",
              "            7.79901093e-05, 7.19922464e-05, 1.00503355e-04, 1.02730759e-04,\n",
              "            8.13975130e-05, 6.73692266e-05, 1.06988438e-04, 8.94192126e-05,\n",
              "            8.18244007e-05, 3.52624738e-05, 7.63886128e-05, 9.31179529e-05,\n",
              "            7.60773401e-05, 9.98369942e-05, 7.54975263e-05, 8.60409127e-05,\n",
              "            9.75254952e-05, 5.64062138e-05, 1.22228608e-04, 1.46436272e-04,\n",
              "            1.26084167e-04, 7.42758784e-05, 7.36813090e-05, 7.99041227e-05],\n",
              "           dtype=float32)},\n",
              "    'Conv_0': {'bias': array([7.5595590e-18, 3.3558989e-17, 3.1486167e-17, 2.8415655e-16,\n",
              "            2.6370811e-17, 4.8783208e-17, 8.4870921e-17, 6.3300164e-16,\n",
              "            1.6688483e-17, 5.1249732e-17, 1.3506875e-16, 3.4762026e-16,\n",
              "            7.4673581e-18, 4.5009954e-16, 2.4117832e-17, 4.6492990e-17],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[2.17553807e-05, 1.56926995e-04, 1.10942980e-04,\n",
              "               3.80651996e-04, 1.38571791e-04, 2.84411974e-04,\n",
              "               2.73982350e-05, 1.52390840e-05, 2.84822127e-05,\n",
              "               2.20491827e-04, 2.48716504e-04, 5.20604262e-05,\n",
              "               2.69853335e-05, 2.48752476e-04, 6.92595640e-05,\n",
              "               2.47153366e-04]],\n",
              "     \n",
              "             [[1.02198355e-05, 1.47596671e-04, 6.82614482e-05,\n",
              "               3.42318963e-04, 1.31515713e-04, 2.90435477e-04,\n",
              "               2.58711698e-05, 4.46222875e-05, 4.24700083e-05,\n",
              "               2.05234130e-04, 2.33479863e-04, 1.86857487e-05,\n",
              "               3.16315636e-05, 3.27240530e-04, 6.59364159e-05,\n",
              "               1.98383204e-04]],\n",
              "     \n",
              "             [[1.35699920e-05, 1.03929640e-04, 4.07984662e-05,\n",
              "               2.86029506e-04, 1.26875224e-04, 2.95335922e-04,\n",
              "               2.79171163e-05, 1.50565262e-04, 6.02963300e-05,\n",
              "               1.73092529e-04, 2.37789223e-04, 9.12566338e-06,\n",
              "               3.53315954e-05, 4.51573811e-04, 6.93487091e-05,\n",
              "               1.15867588e-04]],\n",
              "     \n",
              "             [[2.60287052e-05, 1.41609606e-04, 3.58211000e-05,\n",
              "               2.23875031e-04, 1.23854828e-04, 2.87491333e-04,\n",
              "               2.49183249e-05, 3.09297524e-04, 5.66879426e-05,\n",
              "               9.69232860e-05, 2.92727666e-04, 8.01480473e-06,\n",
              "               4.22518242e-05, 5.75260026e-04, 7.30354295e-05,\n",
              "               2.40450663e-05]],\n",
              "     \n",
              "             [[3.60032136e-05, 1.84338787e-04, 4.09309941e-05,\n",
              "               1.65861537e-04, 1.25244565e-04, 2.65414361e-04,\n",
              "               2.10914222e-05, 4.59069939e-04, 5.82339053e-05,\n",
              "               3.89077322e-05, 2.83294154e-04, 5.55882207e-06,\n",
              "               4.12305817e-05, 6.57361932e-04, 7.43802884e-05,\n",
              "               2.00288741e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[1.62210417e-05, 1.33281865e-04, 1.76450849e-04,\n",
              "               3.08491552e-04, 9.30126407e-05, 2.61369423e-04,\n",
              "               7.52506166e-05, 2.84777416e-05, 2.76757783e-05,\n",
              "               1.95900066e-04, 2.19873429e-04, 8.67206109e-05,\n",
              "               2.45947431e-05, 2.32000442e-04, 8.53684178e-05,\n",
              "               2.70517456e-04]],\n",
              "     \n",
              "             [[8.08313689e-06, 1.38300224e-04, 1.36472110e-04,\n",
              "               2.25841388e-04, 8.52569065e-05, 2.57973268e-04,\n",
              "               5.98847582e-05, 7.95262094e-05, 4.10433968e-05,\n",
              "               1.62938348e-04, 2.10889048e-04, 6.03375011e-05,\n",
              "               2.59072312e-05, 3.11306940e-04, 8.48857235e-05,\n",
              "               2.23202369e-04]],\n",
              "     \n",
              "             [[1.45022677e-05, 1.03196202e-04, 7.56759327e-05,\n",
              "               1.46486185e-04, 8.34713792e-05, 2.48304976e-04,\n",
              "               5.03013944e-05, 1.93152795e-04, 5.26780532e-05,\n",
              "               1.12629546e-04, 2.40546244e-04, 4.53662251e-05,\n",
              "               3.10658688e-05, 3.93179245e-04, 8.92893877e-05,\n",
              "               1.31073059e-04]],\n",
              "     \n",
              "             [[2.54541992e-05, 9.52034170e-05, 4.10120556e-05,\n",
              "               7.54180219e-05, 8.83360990e-05, 2.45777046e-04,\n",
              "               4.71865169e-05, 3.64197796e-04, 4.41422599e-05,\n",
              "               4.42062228e-05, 2.54170242e-04, 2.47170738e-05,\n",
              "               3.96175892e-05, 4.70307510e-04, 9.21074679e-05,\n",
              "               3.40524966e-05]],\n",
              "     \n",
              "             [[3.44854416e-05, 1.66179045e-04, 2.84929592e-05,\n",
              "               4.49585677e-05, 9.72454291e-05, 2.21315218e-04,\n",
              "               5.09184356e-05, 4.97851637e-04, 5.88062321e-05,\n",
              "               1.63402383e-05, 2.10416503e-04, 9.05501292e-06,\n",
              "               3.85157437e-05, 5.26044576e-04, 9.35738062e-05,\n",
              "               2.09356258e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[1.85524423e-05, 1.02177743e-04, 2.03906500e-04,\n",
              "               1.92455132e-04, 3.95503776e-05, 2.17860797e-04,\n",
              "               1.57825314e-04, 5.82208013e-05, 2.37063832e-05,\n",
              "               1.75909561e-04, 2.16102431e-04, 1.48587671e-04,\n",
              "               2.15307591e-05, 1.30910121e-04, 7.41691329e-05,\n",
              "               2.80315144e-04]],\n",
              "     \n",
              "             [[1.22814445e-05, 1.26005820e-04, 2.25981668e-04,\n",
              "               9.64743886e-05, 3.87695254e-05, 1.96721769e-04,\n",
              "               1.36200266e-04, 1.22448997e-04, 3.69333793e-05,\n",
              "               1.30994536e-04, 2.07230754e-04, 1.20801109e-04,\n",
              "               2.02016890e-05, 1.35872528e-04, 7.93515137e-05,\n",
              "               2.39104746e-04]],\n",
              "     \n",
              "             [[1.76956855e-05, 1.27541338e-04, 1.61972799e-04,\n",
              "               3.52585630e-05, 4.42637320e-05, 1.71758613e-04,\n",
              "               1.22675789e-04, 2.24004005e-04, 4.18176896e-05,\n",
              "               7.25754944e-05, 2.24453033e-04, 9.69439061e-05,\n",
              "               2.71129229e-05, 1.55936912e-04, 8.67588096e-05,\n",
              "               1.47129336e-04]],\n",
              "     \n",
              "             [[2.86602808e-05, 8.90282172e-05, 9.43506893e-05,\n",
              "               1.53635847e-05, 5.36801490e-05, 1.54258174e-04,\n",
              "               1.40348420e-04, 3.23231041e-04, 3.67223911e-05,\n",
              "               2.55407158e-05, 2.22465853e-04, 4.97689252e-05,\n",
              "               3.38624668e-05, 2.02094365e-04, 8.35114042e-05,\n",
              "               4.96891880e-05]],\n",
              "     \n",
              "             [[3.49572474e-05, 1.42783058e-04, 5.26404656e-05,\n",
              "               4.86293939e-05, 6.86979474e-05, 1.30166765e-04,\n",
              "               1.67948907e-04, 4.14308219e-04, 6.36083787e-05,\n",
              "               2.05979268e-05, 2.34703795e-04, 2.03355794e-05,\n",
              "               3.44305372e-05, 2.68095435e-04, 8.58290368e-05,\n",
              "               2.18300374e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[2.40587324e-05, 9.39479069e-05, 1.59450414e-04,\n",
              "               8.66098126e-05, 1.70397525e-05, 1.12261274e-04,\n",
              "               2.45697622e-04, 5.17076114e-05, 2.58722412e-05,\n",
              "               1.62839206e-04, 1.89832659e-04, 2.05224977e-04,\n",
              "               1.77304537e-05, 3.89422348e-05, 8.61685330e-05,\n",
              "               2.92778888e-04]],\n",
              "     \n",
              "             [[2.06979003e-05, 1.23962469e-04, 2.30030768e-04,\n",
              "               2.41666512e-05, 1.86241632e-05, 8.50731958e-05,\n",
              "               2.32323655e-04, 7.66332159e-05, 4.55980298e-05,\n",
              "               1.12288450e-04, 1.94477558e-04, 1.51144850e-04,\n",
              "               1.62523156e-05, 2.49209825e-05, 8.84720794e-05,\n",
              "               2.57852516e-04]],\n",
              "     \n",
              "             [[2.71974441e-05, 1.66063706e-04, 2.35202286e-04,\n",
              "               1.27666563e-05, 2.40488662e-05, 6.36129917e-05,\n",
              "               2.29896061e-04, 1.11906855e-04, 4.77706708e-05,\n",
              "               4.64742843e-05, 2.37972868e-04, 1.00449804e-04,\n",
              "               2.11591887e-05, 2.80864879e-05, 8.73843383e-05,\n",
              "               1.67686085e-04]],\n",
              "     \n",
              "             [[3.68203728e-05, 1.38905103e-04, 1.75944369e-04,\n",
              "               4.45920014e-05, 2.86677296e-05, 5.31483529e-05,\n",
              "               2.58356507e-04, 1.74971457e-04, 4.65428311e-05,\n",
              "               1.94825443e-05, 2.42646784e-04, 3.54085605e-05,\n",
              "               2.60928864e-05, 5.40963374e-05, 8.59271167e-05,\n",
              "               7.52206834e-05]],\n",
              "     \n",
              "             [[3.75628988e-05, 1.44293459e-04, 1.24249476e-04,\n",
              "               1.25885912e-04, 4.14483111e-05, 4.09333734e-05,\n",
              "               2.92536482e-04, 2.79560219e-04, 7.14347334e-05,\n",
              "               3.09232528e-05, 2.45297648e-04, 7.94808329e-06,\n",
              "               2.96312664e-05, 1.03705373e-04, 9.64839128e-05,\n",
              "               2.36916439e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[3.22187625e-05, 1.04889434e-04, 9.13152107e-05,\n",
              "               3.49735419e-05, 1.46328612e-05, 2.80992808e-05,\n",
              "               3.00966785e-04, 3.99171768e-05, 3.77300639e-05,\n",
              "               1.63741308e-04, 1.96294961e-04, 2.41357047e-04,\n",
              "               2.15153032e-05, 1.69014398e-04, 7.86294113e-05,\n",
              "               3.20727879e-04]],\n",
              "     \n",
              "             [[3.09621428e-05, 1.41234646e-04, 1.44223028e-04,\n",
              "               2.25875392e-05, 1.72733744e-05, 1.31291663e-05,\n",
              "               3.04876710e-04, 2.26790726e-05, 6.89796507e-05,\n",
              "               1.09058354e-04, 2.51745485e-04, 1.66458180e-04,\n",
              "               1.93368651e-05, 1.53633344e-04, 7.51127664e-05,\n",
              "               2.90485506e-04]],\n",
              "     \n",
              "             [[3.89136767e-05, 1.80785952e-04, 1.90779392e-04,\n",
              "               5.92770084e-05, 1.84036471e-05, 1.80006355e-05,\n",
              "               3.11873126e-04, 3.07017144e-05, 6.79078948e-05,\n",
              "               3.78467266e-05, 2.61860550e-04, 8.90512092e-05,\n",
              "               2.22884191e-05, 1.63271121e-04, 7.34556888e-05,\n",
              "               2.30640057e-04]],\n",
              "     \n",
              "             [[4.32706947e-05, 1.78141825e-04, 2.01409508e-04,\n",
              "               1.45480459e-04, 1.77116763e-05, 1.72270466e-05,\n",
              "               3.37932666e-04, 9.42432889e-05, 5.71111595e-05,\n",
              "               2.06155109e-05, 2.61940731e-04, 2.28259014e-05,\n",
              "               2.50256689e-05, 2.06126584e-04, 7.83806245e-05,\n",
              "               1.49136802e-04]],\n",
              "     \n",
              "             [[3.71751157e-05, 1.68148079e-04, 1.73353503e-04,\n",
              "               2.41057758e-04, 2.46487616e-05, 1.15173616e-05,\n",
              "               3.61212500e-04, 2.18563262e-04, 7.74297660e-05,\n",
              "               5.61804809e-05, 2.87889387e-04, 2.06387303e-05,\n",
              "               2.78351126e-05, 2.76670384e-04, 8.63972018e-05,\n",
              "               7.33213019e-05]]]], dtype=float32)},\n",
              "    'Conv_1': {'bias': array([1.4738033e-18, 8.2453762e-19, 1.9965005e-18, 8.3805132e-19,\n",
              "            3.1968215e-18, 2.5789610e-18, 5.9042777e-19, 1.3029746e-18,\n",
              "            1.0570425e-18, 1.5025055e-18, 1.2287711e-18, 1.5226551e-18,\n",
              "            7.0519189e-19, 1.4929031e-18, 1.7486625e-18, 1.2624759e-18,\n",
              "            8.8870682e-19, 7.1771768e-19, 7.8910399e-19, 2.2977760e-18,\n",
              "            7.2541754e-19, 1.9135370e-18, 4.6440427e-19, 1.5561282e-18,\n",
              "            1.3385546e-18, 1.0623657e-18, 7.8485688e-19, 3.8601703e-18,\n",
              "            1.0199614e-18, 6.3067305e-19, 1.1293457e-18, 1.6840298e-18],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[2.8610174e-07, 5.0465201e-07, 8.4211666e-07, ...,\n",
              "               1.6617160e-07, 1.7124954e-07, 3.0630079e-07],\n",
              "              [6.8292098e-07, 8.1452390e-07, 2.0196990e-06, ...,\n",
              "               2.6442964e-07, 1.2363732e-06, 1.2104858e-06],\n",
              "              [9.7704140e-07, 7.0382242e-07, 1.1282133e-06, ...,\n",
              "               2.3505252e-07, 8.4228145e-07, 1.1342238e-06],\n",
              "              ...,\n",
              "              [6.0204246e-07, 1.1971181e-06, 2.2046804e-06, ...,\n",
              "               6.0622062e-07, 9.2031462e-07, 1.0933817e-06],\n",
              "              [1.6879949e-06, 1.7395898e-06, 3.4639388e-06, ...,\n",
              "               4.9283130e-07, 7.1435949e-07, 1.8324724e-06],\n",
              "              [1.0915494e-06, 7.1735258e-07, 1.5369504e-06, ...,\n",
              "               4.2575277e-07, 7.2065603e-07, 1.3305580e-06]],\n",
              "     \n",
              "             [[6.5596504e-07, 7.9022556e-07, 8.6776714e-07, ...,\n",
              "               3.7014129e-07, 6.3142170e-07, 5.1085004e-07],\n",
              "              [8.2255758e-07, 7.7006888e-07, 2.4250558e-06, ...,\n",
              "               3.9156808e-07, 1.1801134e-06, 1.4750742e-06],\n",
              "              [1.1797617e-06, 6.9033945e-07, 1.2354448e-06, ...,\n",
              "               4.3956103e-07, 8.7670725e-07, 1.4867433e-06],\n",
              "              ...,\n",
              "              [7.5781264e-07, 1.6223945e-06, 2.0534358e-06, ...,\n",
              "               5.0984062e-07, 8.1574854e-07, 1.3792339e-06],\n",
              "              [2.1541912e-06, 1.9830379e-06, 4.4462563e-06, ...,\n",
              "               3.7103993e-07, 6.8486963e-07, 2.4484532e-06],\n",
              "              [1.8075516e-06, 1.9003926e-06, 1.5476693e-06, ...,\n",
              "               7.4183413e-07, 1.7448089e-06, 2.4775788e-06]],\n",
              "     \n",
              "             [[6.2444514e-07, 6.8260817e-07, 1.3802569e-06, ...,\n",
              "               4.7526501e-07, 8.4599793e-07, 6.7687375e-07],\n",
              "              [7.7591335e-07, 1.7702297e-06, 3.3301806e-06, ...,\n",
              "               3.7061983e-07, 1.4096659e-06, 1.3331359e-06],\n",
              "              [1.5165334e-06, 1.4070288e-06, 2.1340268e-06, ...,\n",
              "               3.7517441e-07, 8.5994998e-07, 2.5054019e-06],\n",
              "              ...,\n",
              "              [8.9606999e-07, 1.7076989e-06, 1.7257299e-06, ...,\n",
              "               5.6228919e-07, 7.1960659e-07, 1.9341971e-06],\n",
              "              [2.4944254e-06, 1.7915729e-06, 4.3307746e-06, ...,\n",
              "               3.4770346e-07, 7.1961171e-07, 3.1569921e-06],\n",
              "              [2.5046493e-06, 1.9573067e-06, 2.2929614e-06, ...,\n",
              "               1.0908664e-06, 1.9684219e-06, 4.0859445e-06]],\n",
              "     \n",
              "             [[6.6243109e-07, 8.2372071e-07, 1.8620798e-06, ...,\n",
              "               6.1979000e-07, 8.8056805e-07, 6.2675269e-07],\n",
              "              [9.3574999e-07, 1.6962111e-06, 4.2634110e-06, ...,\n",
              "               3.0420028e-07, 1.1035296e-06, 1.8931788e-06],\n",
              "              [1.4163213e-06, 1.6837876e-06, 2.7761298e-06, ...,\n",
              "               1.9463343e-07, 8.6559936e-07, 2.5096522e-06],\n",
              "              ...,\n",
              "              [9.3614705e-07, 1.4464878e-06, 1.3112569e-06, ...,\n",
              "               6.2639600e-07, 4.8773899e-07, 1.8568799e-06],\n",
              "              [2.3811808e-06, 1.4288971e-06, 3.7798077e-06, ...,\n",
              "               3.2772215e-07, 7.4276937e-07, 3.6411288e-06],\n",
              "              [2.9767618e-06, 3.0816229e-06, 4.9175064e-06, ...,\n",
              "               1.5211803e-06, 2.3880298e-06, 5.4749553e-06]],\n",
              "     \n",
              "             [[6.9850483e-07, 7.7586134e-07, 2.1051428e-06, ...,\n",
              "               6.1727582e-07, 8.0996165e-07, 6.3026283e-07],\n",
              "              [1.3708191e-06, 1.6180832e-06, 3.5912306e-06, ...,\n",
              "               2.9813441e-07, 5.6772672e-07, 1.4056030e-06],\n",
              "              [1.8928781e-06, 1.2775943e-06, 2.9703695e-06, ...,\n",
              "               2.5070236e-07, 7.5016368e-07, 2.7763733e-06],\n",
              "              ...,\n",
              "              [9.6877727e-07, 9.6420672e-07, 6.3679011e-07, ...,\n",
              "               5.5107029e-07, 4.7317417e-07, 1.6057645e-06],\n",
              "              [2.1766998e-06, 1.1007963e-06, 3.4257805e-06, ...,\n",
              "               3.9364508e-07, 7.7424221e-07, 3.6949978e-06],\n",
              "              [3.3416170e-06, 2.8928073e-06, 5.8674036e-06, ...,\n",
              "               1.3256594e-06, 2.2520007e-06, 4.9887994e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[6.3219727e-07, 5.7446061e-07, 9.7072871e-07, ...,\n",
              "               2.1055119e-07, 2.3398539e-07, 5.8322610e-07],\n",
              "              [1.4576360e-06, 9.0813683e-07, 2.0520138e-06, ...,\n",
              "               2.8865222e-07, 1.5476151e-06, 1.7522293e-06],\n",
              "              [7.0030654e-07, 4.5599396e-07, 1.1546233e-06, ...,\n",
              "               2.1831434e-07, 4.8035912e-07, 6.2678248e-07],\n",
              "              ...,\n",
              "              [1.2439048e-06, 1.5984234e-06, 2.6660659e-06, ...,\n",
              "               3.9721706e-07, 6.9344657e-07, 1.4332178e-06],\n",
              "              [2.0122204e-06, 7.9767443e-07, 3.9064366e-06, ...,\n",
              "               6.8287625e-07, 5.7840617e-07, 1.6695807e-06],\n",
              "              [9.5047892e-07, 8.7727165e-07, 1.1541025e-06, ...,\n",
              "               4.8414699e-07, 8.2387373e-07, 8.1699665e-07]],\n",
              "     \n",
              "             [[1.0828595e-06, 7.9502183e-07, 9.2098628e-07, ...,\n",
              "               5.5337767e-07, 9.0477965e-07, 7.8697343e-07],\n",
              "              [1.3556720e-06, 1.2058064e-06, 2.1147234e-06, ...,\n",
              "               1.9250376e-07, 1.3707330e-06, 2.1465457e-06],\n",
              "              [1.4617508e-06, 5.1405863e-07, 1.3678521e-06, ...,\n",
              "               1.6514255e-07, 4.6699839e-07, 1.5445501e-06],\n",
              "              ...,\n",
              "              [1.3125851e-06, 1.7651132e-06, 2.4497654e-06, ...,\n",
              "               3.5389013e-07, 4.8398795e-07, 1.6546759e-06],\n",
              "              [2.6160169e-06, 1.0258395e-06, 4.6089540e-06, ...,\n",
              "               6.3038385e-07, 5.9947831e-07, 2.2613006e-06],\n",
              "              [2.0599391e-06, 2.1868143e-06, 1.4571779e-06, ...,\n",
              "               6.5490985e-07, 1.9359363e-06, 2.3318576e-06]],\n",
              "     \n",
              "             [[1.0920063e-06, 7.1778555e-07, 1.9241859e-06, ...,\n",
              "               5.8455151e-07, 8.8901993e-07, 1.1133978e-06],\n",
              "              [1.8673886e-06, 1.7749981e-06, 5.0852805e-06, ...,\n",
              "               2.2148252e-07, 9.1321346e-07, 2.1703322e-06],\n",
              "              [1.7905636e-06, 8.5485163e-07, 1.1560104e-06, ...,\n",
              "               2.0740310e-07, 4.9166266e-07, 1.5152493e-06],\n",
              "              ...,\n",
              "              [1.2374165e-06, 1.3140750e-06, 2.8454654e-06, ...,\n",
              "               4.6352974e-07, 3.7028769e-07, 1.9200179e-06],\n",
              "              [2.8496020e-06, 1.2021987e-06, 4.6189580e-06, ...,\n",
              "               6.6776454e-07, 5.8248526e-07, 2.7785090e-06],\n",
              "              [2.6270063e-06, 2.0341131e-06, 1.5849364e-06, ...,\n",
              "               9.2757779e-07, 1.3593715e-06, 3.9422839e-06]],\n",
              "     \n",
              "             [[1.2096393e-06, 7.9805307e-07, 2.5027441e-06, ...,\n",
              "               8.2006250e-07, 9.2542757e-07, 1.1959845e-06],\n",
              "              [1.4422912e-06, 2.1882709e-06, 3.7480645e-06, ...,\n",
              "               1.3477182e-07, 9.4894904e-07, 2.5157663e-06],\n",
              "              [2.1197275e-06, 1.4460343e-06, 1.3686262e-06, ...,\n",
              "               4.5557803e-07, 5.4863290e-07, 2.2221673e-06],\n",
              "              ...,\n",
              "              [1.2655030e-06, 9.2973693e-07, 1.4592422e-06, ...,\n",
              "               3.1623620e-07, 2.3174228e-07, 1.9264394e-06],\n",
              "              [2.7661583e-06, 1.1465690e-06, 4.8263564e-06, ...,\n",
              "               6.0751296e-07, 7.5419274e-07, 3.0701021e-06],\n",
              "              [3.5006085e-06, 3.2899609e-06, 3.8696644e-06, ...,\n",
              "               1.6557871e-06, 1.9603331e-06, 4.9535101e-06]],\n",
              "     \n",
              "             [[1.1614824e-06, 8.6058373e-07, 2.8687557e-06, ...,\n",
              "               7.9032935e-07, 8.1465163e-07, 1.2672872e-06],\n",
              "              [2.2124802e-06, 1.7614659e-06, 3.8906719e-06, ...,\n",
              "               2.6988161e-07, 6.0632181e-07, 2.3856539e-06],\n",
              "              [2.0475354e-06, 1.3986962e-06, 1.7858591e-06, ...,\n",
              "               4.0413616e-07, 7.0603761e-07, 2.3819844e-06],\n",
              "              ...,\n",
              "              [1.2712322e-06, 8.4419707e-07, 6.0483086e-07, ...,\n",
              "               3.2454508e-07, 2.5492673e-07, 1.3805021e-06],\n",
              "              [2.7088615e-06, 1.1141302e-06, 4.5817742e-06, ...,\n",
              "               5.9134908e-07, 9.1994065e-07, 3.3102610e-06],\n",
              "              [4.0151754e-06, 3.7768718e-06, 5.3743393e-06, ...,\n",
              "               1.3022860e-06, 2.2183483e-06, 6.2412687e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[9.5317660e-07, 4.6908590e-07, 1.1733820e-06, ...,\n",
              "               3.8861529e-07, 5.3709397e-07, 5.1945301e-07],\n",
              "              [1.6711972e-06, 6.7192076e-07, 2.0311334e-06, ...,\n",
              "               3.7735262e-07, 1.2389048e-06, 1.8048612e-06],\n",
              "              [8.8632135e-07, 6.2315775e-07, 6.9959322e-07, ...,\n",
              "               2.5830815e-07, 6.8404483e-07, 1.4464422e-07],\n",
              "              ...,\n",
              "              [1.7788698e-06, 9.9099100e-07, 2.7266781e-06, ...,\n",
              "               5.6605126e-07, 5.6120240e-07, 1.6133023e-06],\n",
              "              [1.7027569e-06, 1.4982551e-06, 3.9616912e-06, ...,\n",
              "               5.0095088e-07, 1.1183456e-06, 1.3269281e-06],\n",
              "              [9.8305873e-07, 9.0972537e-07, 7.8825991e-07, ...,\n",
              "               6.4394141e-07, 6.0965141e-07, 2.2485889e-07]],\n",
              "     \n",
              "             [[1.3122327e-06, 7.8999989e-07, 1.4410970e-06, ...,\n",
              "               7.9124203e-07, 1.0817854e-06, 1.0343048e-06],\n",
              "              [1.9779386e-06, 7.8086418e-07, 2.4260189e-06, ...,\n",
              "               2.2133514e-07, 1.3192604e-06, 2.2438367e-06],\n",
              "              [1.0401736e-06, 4.9271176e-07, 9.6504493e-07, ...,\n",
              "               1.6826277e-07, 7.6886920e-07, 3.0598770e-07],\n",
              "              ...,\n",
              "              [1.8718637e-06, 9.8738769e-07, 2.3863922e-06, ...,\n",
              "               5.2217626e-07, 3.4457022e-07, 1.8988379e-06],\n",
              "              [2.7652134e-06, 2.1865435e-06, 4.1660769e-06, ...,\n",
              "               4.9384266e-07, 1.1759438e-06, 1.6754307e-06],\n",
              "              [1.9543334e-06, 2.4428089e-06, 9.4542798e-07, ...,\n",
              "               7.3310434e-07, 1.6652066e-06, 8.7429947e-07]],\n",
              "     \n",
              "             [[1.2600610e-06, 7.4336396e-07, 2.5515833e-06, ...,\n",
              "               9.3107184e-07, 8.9310328e-07, 1.2629215e-06],\n",
              "              [2.5089373e-06, 1.1985112e-06, 2.2941920e-06, ...,\n",
              "               1.7166471e-07, 1.3362987e-06, 3.3033230e-06],\n",
              "              [2.1775941e-06, 7.9431294e-07, 8.8485854e-07, ...,\n",
              "               2.0846539e-07, 3.3695278e-07, 6.8300704e-07],\n",
              "              ...,\n",
              "              [2.0199907e-06, 7.7037032e-07, 2.3108682e-06, ...,\n",
              "               6.1753582e-07, 2.2229919e-07, 2.1012884e-06],\n",
              "              [3.3710119e-06, 2.6263417e-06, 3.6733231e-06, ...,\n",
              "               5.5017364e-07, 1.3835682e-06, 2.1482185e-06],\n",
              "              [2.3259724e-06, 1.4878274e-06, 1.0969989e-06, ...,\n",
              "               9.4970557e-07, 1.4440792e-06, 1.5740131e-06]],\n",
              "     \n",
              "             [[1.5214912e-06, 9.9931822e-07, 3.0915849e-06, ...,\n",
              "               9.2385505e-07, 1.0858987e-06, 1.7567264e-06],\n",
              "              [2.4602382e-06, 1.8997688e-06, 2.7309279e-06, ...,\n",
              "               2.3213180e-07, 1.0122267e-06, 2.3034706e-06],\n",
              "              [2.3542862e-06, 8.8640127e-07, 6.1775847e-07, ...,\n",
              "               4.3257114e-07, 5.0880084e-07, 6.3899819e-07],\n",
              "              ...,\n",
              "              [1.6735905e-06, 7.2199407e-07, 1.1100514e-06, ...,\n",
              "               6.5257871e-07, 2.5711321e-07, 1.8808591e-06],\n",
              "              [3.9192969e-06, 2.9452674e-06, 3.4160598e-06, ...,\n",
              "               5.3753666e-07, 1.6199549e-06, 2.5820070e-06],\n",
              "              [3.6630584e-06, 2.5027687e-06, 2.6161217e-06, ...,\n",
              "               1.4423883e-06, 1.4299484e-06, 1.9180454e-06]],\n",
              "     \n",
              "             [[1.3092503e-06, 8.6081377e-07, 3.2339128e-06, ...,\n",
              "               8.8007664e-07, 9.5553776e-07, 1.6422588e-06],\n",
              "              [2.5436457e-06, 1.5889937e-06, 2.5435395e-06, ...,\n",
              "               2.1203527e-07, 9.1065226e-07, 2.6647142e-06],\n",
              "              [2.7870785e-06, 1.8794015e-06, 1.7177856e-06, ...,\n",
              "               3.2555292e-07, 9.4929158e-07, 9.4277993e-07],\n",
              "              ...,\n",
              "              [1.5364380e-06, 7.5873197e-07, 5.2149932e-07, ...,\n",
              "               5.7019116e-07, 3.3811421e-07, 1.3855259e-06],\n",
              "              [3.9573065e-06, 2.9212665e-06, 2.8902521e-06, ...,\n",
              "               6.5179205e-07, 1.7213232e-06, 3.2832111e-06],\n",
              "              [4.0742598e-06, 3.0488006e-06, 3.5072078e-06, ...,\n",
              "               1.4116080e-06, 2.0879561e-06, 2.9455448e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[1.0887444e-06, 7.0352945e-07, 1.4088648e-06, ...,\n",
              "               5.7171138e-07, 8.2568477e-07, 5.4900221e-07],\n",
              "              [1.5868466e-06, 1.1103142e-06, 1.2484069e-06, ...,\n",
              "               4.7371745e-07, 1.1441113e-06, 1.8810545e-06],\n",
              "              [6.6071357e-07, 5.1758622e-07, 6.0569289e-07, ...,\n",
              "               2.6522650e-07, 2.0238760e-07, 2.2283217e-07],\n",
              "              ...,\n",
              "              [1.6601624e-06, 1.1932483e-06, 2.0568184e-06, ...,\n",
              "               4.1087634e-07, 4.2056240e-07, 2.1389505e-06],\n",
              "              [1.9450354e-06, 1.7910190e-06, 2.4884162e-06, ...,\n",
              "               1.4169269e-06, 1.0855001e-06, 1.4088876e-06],\n",
              "              [6.4786161e-07, 4.8301951e-07, 5.6716016e-07, ...,\n",
              "               4.8175485e-07, 2.3259999e-07, 9.0620944e-08]],\n",
              "     \n",
              "             [[1.4261768e-06, 1.2049984e-06, 1.8555545e-06, ...,\n",
              "               1.1218569e-06, 1.1786655e-06, 1.2123189e-06],\n",
              "              [2.1606988e-06, 8.0068310e-07, 2.0355542e-06, ...,\n",
              "               1.9512419e-07, 1.4007569e-06, 1.6026816e-06],\n",
              "              [9.3008919e-07, 5.0522033e-07, 2.3909075e-07, ...,\n",
              "               3.3752139e-07, 4.1569871e-07, 4.4069699e-07],\n",
              "              ...,\n",
              "              [1.8549761e-06, 1.4761748e-06, 1.8808345e-06, ...,\n",
              "               4.1484847e-07, 4.6906615e-07, 2.4679318e-06],\n",
              "              [2.4136125e-06, 1.9496206e-06, 2.4992621e-06, ...,\n",
              "               1.5058549e-06, 1.1256567e-06, 1.6122959e-06],\n",
              "              [1.7026300e-06, 1.6159245e-06, 3.7963588e-07, ...,\n",
              "               4.4193129e-07, 1.1433001e-06, 3.8772978e-07]],\n",
              "     \n",
              "             [[1.4741040e-06, 1.3645000e-06, 2.4494382e-06, ...,\n",
              "               1.0200044e-06, 1.0456989e-06, 1.4673851e-06],\n",
              "              [3.1453660e-06, 1.4255650e-06, 2.2332740e-06, ...,\n",
              "               3.0967504e-07, 2.1090204e-06, 2.5695917e-06],\n",
              "              [1.2818688e-06, 4.0323098e-07, 4.1460430e-07, ...,\n",
              "               3.7365444e-07, 3.5021904e-07, 6.3284409e-07],\n",
              "              ...,\n",
              "              [2.2039367e-06, 1.5233658e-06, 1.6197937e-06, ...,\n",
              "               4.6430944e-07, 5.6288951e-07, 2.8702229e-06],\n",
              "              [2.6332191e-06, 2.2118847e-06, 2.2947195e-06, ...,\n",
              "               1.6866836e-06, 1.3215080e-06, 1.6916717e-06],\n",
              "              [2.3417122e-06, 1.1871875e-06, 1.1038960e-06, ...,\n",
              "               7.4952533e-07, 1.3830155e-06, 1.2283662e-06]],\n",
              "     \n",
              "             [[1.6229010e-06, 1.4590383e-06, 3.0648644e-06, ...,\n",
              "               9.6912902e-07, 1.4113848e-06, 1.6348438e-06],\n",
              "              [3.1157088e-06, 2.0234315e-06, 2.7851743e-06, ...,\n",
              "               4.0649758e-07, 1.7704978e-06, 2.2789620e-06],\n",
              "              [1.7527353e-06, 7.5541146e-07, 5.4864586e-07, ...,\n",
              "               6.4026892e-07, 7.8012266e-07, 5.1818552e-07],\n",
              "              ...,\n",
              "              [1.9418953e-06, 1.6966886e-06, 7.2473620e-07, ...,\n",
              "               4.4207678e-07, 5.4781441e-07, 3.0182434e-06],\n",
              "              [2.6253686e-06, 2.4900819e-06, 1.7836689e-06, ...,\n",
              "               1.7687142e-06, 1.6054089e-06, 2.0770763e-06],\n",
              "              [3.6053384e-06, 1.6664249e-06, 2.1963647e-06, ...,\n",
              "               1.3330061e-06, 1.1394272e-06, 2.0644984e-06]],\n",
              "     \n",
              "             [[1.6533847e-06, 1.3984164e-06, 3.0303690e-06, ...,\n",
              "               7.5740263e-07, 1.3831723e-06, 1.6861298e-06],\n",
              "              [3.0992460e-06, 2.4850858e-06, 2.9177381e-06, ...,\n",
              "               3.4487161e-07, 1.9360432e-06, 2.5018785e-06],\n",
              "              [1.9286242e-06, 9.8296528e-07, 6.7721686e-07, ...,\n",
              "               4.9695041e-07, 1.0193901e-06, 4.4273895e-07],\n",
              "              ...,\n",
              "              [1.5247722e-06, 1.4983364e-06, 5.3657067e-07, ...,\n",
              "               4.7371128e-07, 6.5640234e-07, 2.5245656e-06],\n",
              "              [2.5197901e-06, 2.0117623e-06, 1.3525359e-06, ...,\n",
              "               2.1637395e-06, 1.5602028e-06, 2.1525732e-06],\n",
              "              [4.1143085e-06, 2.1064184e-06, 2.7307196e-06, ...,\n",
              "               1.5442446e-06, 1.8076232e-06, 2.8034685e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[9.2088464e-07, 1.0142483e-06, 1.4360962e-06, ...,\n",
              "               7.6287381e-07, 1.0217892e-06, 7.6112377e-07],\n",
              "              [1.6389089e-06, 2.4988753e-06, 1.0691240e-06, ...,\n",
              "               7.8451853e-07, 1.2656139e-06, 1.5039285e-06],\n",
              "              [3.5611174e-07, 4.8269641e-07, 2.3482563e-07, ...,\n",
              "               2.4209422e-07, 1.2264260e-07, 1.3016535e-07],\n",
              "              ...,\n",
              "              [1.6114435e-06, 1.7505308e-06, 1.6990350e-06, ...,\n",
              "               1.1815754e-06, 6.8707317e-07, 2.7043707e-06],\n",
              "              [1.7269228e-06, 2.9431135e-06, 1.9517415e-06, ...,\n",
              "               2.1578580e-06, 1.2691658e-06, 2.6632408e-06],\n",
              "              [6.4168466e-07, 3.6762765e-07, 4.8847960e-07, ...,\n",
              "               2.6733053e-07, 1.5493444e-07, 1.0393355e-07]],\n",
              "     \n",
              "             [[1.1785006e-06, 1.4627232e-06, 1.7317009e-06, ...,\n",
              "               1.1187739e-06, 1.2259020e-06, 9.3051801e-07],\n",
              "              [1.7701623e-06, 1.3061374e-06, 1.3873630e-06, ...,\n",
              "               4.7427341e-07, 2.0220662e-06, 1.3336470e-06],\n",
              "              [6.1244094e-07, 5.6776980e-07, 2.3574094e-07, ...,\n",
              "               1.6725014e-07, 2.2519878e-07, 2.6885945e-07],\n",
              "              ...,\n",
              "              [1.7384576e-06, 2.1616893e-06, 2.0822083e-06, ...,\n",
              "               9.7183818e-07, 1.2186558e-06, 2.9329565e-06],\n",
              "              [1.7723097e-06, 3.0896990e-06, 1.8586973e-06, ...,\n",
              "               2.4530086e-06, 1.4680249e-06, 3.0283993e-06],\n",
              "              [1.4203300e-06, 1.2134898e-06, 4.8925517e-07, ...,\n",
              "               2.5046000e-07, 8.9242957e-07, 3.1379659e-07]],\n",
              "     \n",
              "             [[1.2846797e-06, 1.4385150e-06, 2.4139913e-06, ...,\n",
              "               1.0312423e-06, 1.3466412e-06, 8.9642606e-07],\n",
              "              [2.6383934e-06, 2.3878551e-06, 2.3232872e-06, ...,\n",
              "               4.0711470e-07, 2.0626210e-06, 1.4575223e-06],\n",
              "              [9.8052124e-07, 7.4493164e-07, 4.0837000e-07, ...,\n",
              "               6.0464572e-07, 4.8376143e-07, 1.1547004e-06],\n",
              "              ...,\n",
              "              [1.5280410e-06, 2.3370478e-06, 2.5775446e-06, ...,\n",
              "               1.1184372e-06, 1.9212703e-06, 3.2244225e-06],\n",
              "              [1.3467909e-06, 2.8821680e-06, 1.5043643e-06, ...,\n",
              "               2.4311807e-06, 1.6960532e-06, 3.3876581e-06],\n",
              "              [1.6445434e-06, 1.3147502e-06, 1.0308374e-06, ...,\n",
              "               4.0894250e-07, 1.5998761e-06, 1.1082024e-06]],\n",
              "     \n",
              "             [[1.6307229e-06, 1.1593535e-06, 2.6458674e-06, ...,\n",
              "               9.6060739e-07, 1.4817464e-06, 6.6676461e-07],\n",
              "              [2.9446953e-06, 1.8387263e-06, 2.6676803e-06, ...,\n",
              "               6.3219215e-07, 2.0175389e-06, 1.6121301e-06],\n",
              "              [1.1172979e-06, 1.0297930e-06, 3.8532022e-07, ...,\n",
              "               5.7154665e-07, 5.1689716e-07, 1.5585648e-06],\n",
              "              ...,\n",
              "              [1.1312909e-06, 1.9932645e-06, 1.8265182e-06, ...,\n",
              "               1.1517620e-06, 1.8701572e-06, 3.2555236e-06],\n",
              "              [9.6179065e-07, 2.6114255e-06, 1.1500326e-06, ...,\n",
              "               2.3717216e-06, 2.0889534e-06, 2.9499054e-06],\n",
              "              [2.9578796e-06, 1.5856201e-06, 2.3759037e-06, ...,\n",
              "               9.6972758e-07, 1.3635073e-06, 2.6796083e-06]],\n",
              "     \n",
              "             [[1.8317776e-06, 1.1069191e-06, 2.1092480e-06, ...,\n",
              "               8.3918059e-07, 1.2877915e-06, 8.9515248e-07],\n",
              "              [2.2269005e-06, 2.5243198e-06, 3.0151045e-06, ...,\n",
              "               5.5526232e-07, 2.3544180e-06, 1.9004688e-06],\n",
              "              [1.1723901e-06, 1.1316562e-06, 8.1541629e-07, ...,\n",
              "               7.0479570e-07, 7.3923394e-07, 2.0333873e-06],\n",
              "              ...,\n",
              "              [1.1290817e-06, 1.2413692e-06, 1.0945729e-06, ...,\n",
              "               1.1123396e-06, 1.5635316e-06, 2.9636092e-06],\n",
              "              [7.3657185e-07, 2.5421978e-06, 1.1441223e-06, ...,\n",
              "               2.1792544e-06, 2.3274170e-06, 2.2141753e-06],\n",
              "              [2.9890016e-06, 1.6440113e-06, 2.4493243e-06, ...,\n",
              "               1.2525001e-06, 1.5195212e-06, 3.3414385e-06]]]], dtype=float32)},\n",
              "    'Dense_0': {'bias': array([6.1030274e-05, 3.7751084e-05, 9.5962831e-05, 6.6623790e-05,\n",
              "            7.3136209e-05, 7.4875577e-05, 5.8184793e-05, 7.4033196e-05,\n",
              "            8.9397552e-05, 1.0160340e-04], dtype=float32),\n",
              "     'kernel': array([[4.17882147e-05, 4.99834005e-06, 4.85574892e-05, 2.41716589e-05,\n",
              "             2.58661894e-05, 4.90197926e-05, 1.53476030e-05, 4.94129563e-05,\n",
              "             2.27519686e-05, 4.61121563e-05],\n",
              "            [4.70982268e-05, 2.90281077e-05, 9.35775388e-05, 7.44693680e-05,\n",
              "             3.27527014e-05, 5.08188605e-05, 4.33246278e-05, 2.40419904e-05,\n",
              "             5.35709223e-05, 2.12815175e-05],\n",
              "            [3.40452352e-05, 3.10351461e-05, 5.77912178e-05, 3.36365629e-05,\n",
              "             9.33866177e-05, 9.36230481e-06, 9.24435335e-06, 8.42995651e-05,\n",
              "             6.52941671e-05, 1.27007006e-04],\n",
              "            [1.26657915e-05, 6.40230564e-06, 4.19441458e-05, 7.09309097e-05,\n",
              "             9.75696312e-05, 8.72189339e-05, 1.19245497e-05, 6.50111906e-05,\n",
              "             8.35989995e-05, 1.33474023e-04],\n",
              "            [9.28192640e-06, 2.71280569e-05, 1.08536573e-04, 8.03348084e-05,\n",
              "             1.21571156e-05, 2.01343973e-05, 3.21022503e-06, 9.82071942e-05,\n",
              "             1.80765273e-05, 1.99407277e-05],\n",
              "            [8.97370701e-05, 2.66364059e-05, 2.12835894e-05, 1.20623272e-05,\n",
              "             4.37241652e-05, 8.50872602e-05, 7.34632558e-05, 4.59596522e-05,\n",
              "             1.40947930e-04, 7.44289355e-05],\n",
              "            [4.49856634e-05, 5.27752536e-06, 1.07894208e-04, 7.77190289e-05,\n",
              "             9.53975086e-06, 5.26187141e-05, 1.10719684e-05, 9.24300984e-05,\n",
              "             5.30466896e-05, 7.27374500e-05],\n",
              "            [2.82689507e-05, 3.11792901e-05, 3.33972275e-05, 3.72788490e-05,\n",
              "             2.68369477e-05, 4.22715493e-05, 2.94156216e-05, 1.37506522e-05,\n",
              "             5.14498752e-05, 1.69767627e-05],\n",
              "            [1.02754457e-04, 1.90031210e-06, 8.54928221e-05, 6.45646069e-05,\n",
              "             1.35887412e-05, 1.36919320e-04, 1.04240135e-04, 1.56915521e-05,\n",
              "             1.55193848e-04, 9.65717845e-05],\n",
              "            [4.81193274e-05, 2.13264066e-05, 8.05567688e-05, 8.70542717e-05,\n",
              "             2.30666265e-05, 8.26199466e-05, 3.32424752e-05, 5.49161377e-05,\n",
              "             1.53067231e-05, 4.49254694e-05],\n",
              "            [5.82794019e-05, 9.30374244e-06, 1.01901926e-04, 6.91137611e-05,\n",
              "             1.52986886e-05, 2.96297403e-05, 1.12447024e-05, 7.52289052e-05,\n",
              "             7.53691056e-05, 8.82305321e-05],\n",
              "            [1.58209787e-05, 3.66358654e-06, 2.86406230e-05, 9.65308573e-06,\n",
              "             2.99952444e-05, 3.81598838e-05, 2.64830142e-05, 1.52163248e-05,\n",
              "             3.43277316e-05, 1.95458233e-05],\n",
              "            [3.96797186e-05, 1.71660704e-05, 3.14367862e-05, 1.94619734e-05,\n",
              "             3.32769778e-05, 2.34518720e-05, 1.38417699e-05, 4.39183641e-05,\n",
              "             2.39571145e-05, 5.69248950e-05],\n",
              "            [5.14166313e-05, 3.06166621e-05, 5.87615286e-05, 1.31974639e-05,\n",
              "             4.39576907e-05, 1.81367832e-05, 5.25164396e-05, 2.64721912e-05,\n",
              "             5.00093920e-05, 1.65190686e-05],\n",
              "            [1.46492266e-05, 5.95343226e-06, 1.14027252e-05, 3.55149787e-05,\n",
              "             2.69241536e-05, 4.55533263e-05, 1.58342336e-05, 9.83063455e-06,\n",
              "             5.63308349e-05, 3.36902231e-05],\n",
              "            [1.78090286e-05, 2.44663825e-05, 3.67167559e-05, 1.60453274e-05,\n",
              "             2.82316360e-05, 4.21856967e-05, 3.84461928e-05, 4.07252337e-05,\n",
              "             2.20367692e-05, 2.41690632e-05],\n",
              "            [6.84430561e-05, 4.62560056e-05, 4.45329933e-05, 1.29941782e-05,\n",
              "             1.20823657e-04, 4.41728007e-05, 9.69360699e-05, 3.67556750e-05,\n",
              "             3.89749257e-05, 6.01607826e-05],\n",
              "            [1.46551329e-05, 2.99456206e-05, 1.39920216e-04, 1.36657647e-04,\n",
              "             1.18888420e-04, 8.91832751e-05, 7.92739593e-06, 1.55696413e-04,\n",
              "             1.10747991e-04, 2.00220951e-04],\n",
              "            [2.03724248e-05, 3.51223789e-05, 4.81264688e-05, 1.89023176e-05,\n",
              "             7.45591460e-05, 1.37089291e-05, 2.15529144e-05, 7.63209755e-05,\n",
              "             1.26342156e-05, 8.52367375e-05],\n",
              "            [5.50370787e-05, 2.61993523e-06, 5.74891055e-05, 1.77226702e-05,\n",
              "             4.23196616e-05, 2.28312601e-05, 5.31683836e-05, 5.60513990e-06,\n",
              "             8.88572758e-05, 4.57403803e-05],\n",
              "            [9.85498627e-06, 6.44634565e-05, 5.63081339e-05, 4.41586672e-05,\n",
              "             3.72937648e-05, 3.55715856e-05, 2.68507774e-05, 5.68950309e-05,\n",
              "             3.42182211e-05, 5.68303512e-05],\n",
              "            [3.81137252e-05, 1.63878690e-06, 1.32455207e-05, 1.02165923e-05,\n",
              "             2.18696168e-05, 7.41478289e-05, 5.51916382e-05, 1.33514104e-05,\n",
              "             7.40573960e-05, 5.98217812e-05],\n",
              "            [6.09774688e-05, 1.86048455e-05, 7.07526269e-05, 4.70668892e-05,\n",
              "             2.91446031e-05, 3.73409421e-05, 6.06398899e-05, 1.03214124e-05,\n",
              "             2.34932759e-05, 2.77885683e-05],\n",
              "            [7.48438033e-05, 4.01285797e-06, 3.60820668e-05, 9.29842099e-06,\n",
              "             8.97391510e-05, 2.39260644e-05, 8.27734475e-05, 5.47559148e-06,\n",
              "             7.34093846e-05, 8.76038248e-05],\n",
              "            [4.82781543e-05, 1.59557703e-06, 3.44766340e-05, 4.45283331e-05,\n",
              "             2.10252256e-05, 1.11705333e-04, 7.50385225e-05, 1.02855347e-05,\n",
              "             1.20556455e-04, 7.45978177e-05],\n",
              "            [1.20205195e-05, 2.57157481e-05, 7.02497637e-05, 8.97614227e-05,\n",
              "             1.01299287e-04, 7.58269598e-05, 7.23204948e-06, 1.02256927e-04,\n",
              "             3.46365050e-05, 1.47589701e-04],\n",
              "            [2.71843292e-05, 1.26486138e-05, 5.29714562e-05, 2.35885364e-05,\n",
              "             1.25288716e-05, 4.83395052e-05, 2.08838246e-05, 2.12075829e-05,\n",
              "             2.86102095e-05, 2.44549065e-05],\n",
              "            [1.26291743e-05, 3.10303608e-06, 4.58442046e-05, 1.43573088e-05,\n",
              "             2.60837714e-05, 9.61409842e-06, 2.60672550e-05, 1.59532010e-05,\n",
              "             4.83061376e-05, 1.58452185e-05],\n",
              "            [4.88216865e-05, 1.36441458e-05, 4.42305573e-05, 1.39480599e-05,\n",
              "             8.73788085e-05, 2.04487878e-05, 5.52479760e-05, 9.81733683e-06,\n",
              "             2.96912585e-05, 7.35653448e-05],\n",
              "            [8.18698609e-05, 2.94430829e-05, 5.76798993e-05, 9.52397568e-06,\n",
              "             9.28213340e-05, 3.32132331e-05, 9.34858617e-05, 5.04765521e-05,\n",
              "             3.85715648e-05, 6.61443337e-05],\n",
              "            [4.59968796e-05, 1.41538248e-05, 4.28688327e-05, 9.82664187e-06,\n",
              "             5.24086936e-05, 2.14342890e-05, 6.12306467e-05, 9.03348973e-06,\n",
              "             7.28484301e-05, 3.96403884e-05],\n",
              "            [1.32844962e-05, 2.03430282e-05, 9.35512217e-05, 6.93022448e-05,\n",
              "             2.14354950e-05, 4.62459429e-05, 2.44268449e-05, 6.24411041e-05,\n",
              "             1.35718419e-05, 1.79809776e-05]], dtype=float32)}}},\n",
              "  {'count': array(9370, dtype=int32)}],\n",
              " 'params': {'BatchNorm_0': {'bias': array([-0.9560376 , -1.0071353 , -0.5818822 , -0.36917838, -0.46915853,\n",
              "          -0.88612205, -0.7405656 , -0.70763654, -0.5655106 , -0.6128401 ,\n",
              "          -0.11745953, -0.67799485, -0.41351303, -0.6595287 , -0.04844438,\n",
              "          -0.96831995], dtype=float32),\n",
              "   'scale': array([0.68497014, 0.6631405 , 0.97644866, 1.370787  , 0.95040226,\n",
              "          1.5948869 , 1.4642572 , 1.7942637 , 0.6577792 , 1.3193667 ,\n",
              "          0.4464469 , 1.6518598 , 0.50821584, 1.4551423 , 0.38622802,\n",
              "          1.7144771 ], dtype=float32)},\n",
              "  'BatchNorm_1': {'bias': array([-1.0211192 , -0.60230476, -1.5056435 , -1.0937318 , -1.322513  ,\n",
              "          -0.73252356, -1.2277774 , -1.0907402 , -1.5822235 , -0.69986266,\n",
              "          -1.3024088 , -1.7028631 , -0.950533  , -0.8634483 , -1.5353473 ,\n",
              "          -0.8497833 , -0.8179694 , -0.634372  , -1.2587211 , -1.4026659 ,\n",
              "          -0.8564892 , -1.6817492 , -0.9177957 , -1.5516243 , -1.1748406 ,\n",
              "          -0.7706709 , -0.99771905, -1.9419168 , -1.485772  , -1.056794  ,\n",
              "          -0.8236432 , -1.2707542 ], dtype=float32),\n",
              "   'scale': array([2.430145 , 2.3404508, 3.2803457, 3.0762024, 3.0346034, 2.7043588,\n",
              "          2.9072576, 2.7754447, 3.4146767, 2.6467443, 3.0049117, 2.6088834,\n",
              "          2.631925 , 2.6358528, 2.9629335, 2.4700449, 2.9825718, 3.27846  ,\n",
              "          3.2108822, 2.7371256, 2.535984 , 3.0012107, 2.3586755, 3.0869825,\n",
              "          2.8108566, 2.9940963, 2.3774724, 2.8719318, 2.7653446, 3.018106 ,\n",
              "          2.470961 , 3.0917578], dtype=float32)},\n",
              "  'Conv_0': {'bias': array([-0.12124658,  0.21526994,  0.1908961 , -0.5423655 ,  0.26979086,\n",
              "           0.07708597, -0.11589558, -0.05794707, -0.14736499,  0.35204503,\n",
              "          -0.31011868,  0.02378077, -0.03783318, -0.54379666,  0.03324856,\n",
              "          -0.05634847], dtype=float32),\n",
              "   'kernel': array([[[[ 0.64782727, -0.17432922, -0.32592636,  0.3030778 ,\n",
              "              0.5027317 ,  0.36331734, -2.1424067 , -0.8756017 ,\n",
              "             -0.30372965,  0.37555608, -0.26650122, -0.4898407 ,\n",
              "              1.0499591 , -0.7504754 , -0.3856439 ,  0.01464792]],\n",
              "   \n",
              "           [[ 1.183019  ,  0.6119799 , -0.8591108 ,  0.13028064,\n",
              "              0.82164305,  0.00515884, -1.0350893 , -1.1092191 ,\n",
              "             -0.43393165,  0.31454465, -0.01155446, -0.5536822 ,\n",
              "              0.03462585,  0.05208077, -0.60849684, -0.6167568 ]],\n",
              "   \n",
              "           [[ 0.46887583,  0.5998866 , -0.5845997 ,  0.3373274 ,\n",
              "              0.17818029, -0.02189137, -0.73549926, -0.45316777,\n",
              "             -0.14059718, -0.21141341,  0.36026305, -1.2226975 ,\n",
              "             -0.02522681,  0.07589944, -0.48635525, -0.9094742 ]],\n",
              "   \n",
              "           [[ 0.45528427, -0.08278326, -0.2828693 ,  0.32486942,\n",
              "              0.32810628,  0.14958397, -0.7572427 , -0.08082341,\n",
              "             -0.09799496, -0.45568955,  0.5268058 , -0.9658188 ,\n",
              "             -0.1417978 ,  0.18414195, -0.47526273, -0.98870176]],\n",
              "   \n",
              "           [[ 0.44873825, -0.95132065,  0.17325735, -0.7484398 ,\n",
              "              0.5469649 , -0.11064401, -0.8692778 ,  0.18113689,\n",
              "              0.3813415 , -0.9855947 , -0.04184837, -0.936838  ,\n",
              "             -0.4041919 ,  0.44523397, -0.8982627 , -0.06242581]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.3741862 , -0.46547988,  0.6610525 ,  0.10502229,\n",
              "             -0.5632499 ,  0.2341168 ,  0.12268656, -1.1315424 ,\n",
              "             -0.4898315 ,  0.22385569,  0.29224738, -0.3522718 ,\n",
              "              0.8968037 , -0.5224669 ,  0.34029964,  0.17268294]],\n",
              "   \n",
              "           [[ 0.4182973 , -0.10219502, -0.03723501,  0.09170327,\n",
              "              0.00682544,  0.1264632 ,  0.05736255,  0.29092553,\n",
              "             -0.24981558,  0.2875749 ,  0.5841611 ,  0.19526215,\n",
              "             -0.20628901,  0.3245325 ,  0.28387132,  0.29951736]],\n",
              "   \n",
              "           [[-0.19569203,  0.64426136, -0.27980408,  0.3392184 ,\n",
              "              0.06060128,  0.3421444 , -0.5142252 ,  0.44709212,\n",
              "              0.15430954, -0.11411954,  0.2270908 ,  0.2533129 ,\n",
              "              0.12388808,  0.34018183,  0.15545177,  0.15517098]],\n",
              "   \n",
              "           [[-0.10587048,  0.959171  , -0.7590151 , -0.04560016,\n",
              "              0.07526003,  0.34264216, -0.43717295,  0.12401806,\n",
              "              0.6397484 , -0.05948701, -0.19528113,  0.35094705,\n",
              "             -0.25324437,  0.23163539,  0.21105687, -0.67388994]],\n",
              "   \n",
              "           [[-0.00313488, -0.26790944, -0.4992853 , -1.1800517 ,\n",
              "              0.3891673 , -0.26983485, -0.9770972 ,  0.28400323,\n",
              "              0.27568057, -0.95242786, -0.80285907, -1.2967141 ,\n",
              "             -0.3049377 ,  0.17232482,  0.17894973, -0.7871524 ]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.3622138 , -0.6195789 ,  0.28534827,  0.09232275,\n",
              "             -1.140187  , -0.02770001,  0.09881473, -1.2537775 ,\n",
              "             -0.64884156,  0.34204698,  0.59211916,  0.03229462,\n",
              "              0.36805335, -0.8303921 ,  0.6564378 , -0.36718452]],\n",
              "   \n",
              "           [[ 0.29326567, -0.3948941 ,  0.12969029,  0.31957862,\n",
              "             -0.3681527 ,  0.6208092 , -0.07086673,  0.06941307,\n",
              "              0.08393645,  0.38734853,  0.00215044,  0.10847469,\n",
              "              0.05025116,  0.02959838,  0.3564692 ,  0.24292332]],\n",
              "   \n",
              "           [[ 0.36730048, -0.09569304,  0.20913905, -0.02879266,\n",
              "             -0.23552403,  0.35337836,  0.04841204,  0.25249878,\n",
              "              0.3316112 ,  0.13759144, -0.66884375,  0.41324675,\n",
              "              0.21593249,  0.5701588 ,  0.3639    ,  0.18056788]],\n",
              "   \n",
              "           [[ 0.12749128,  0.6681947 ,  0.03136163, -1.5320091 ,\n",
              "             -0.09465533,  0.08848479, -0.13141288,  0.31774235,\n",
              "              0.8745519 , -1.0054078 , -0.75958616,  0.3911229 ,\n",
              "              0.16566731,  0.29488692,  0.4319292 , -0.22251338]],\n",
              "   \n",
              "           [[ 0.16294616,  0.2254109 , -0.75953156, -0.32326213,\n",
              "              0.21207787, -0.88187164, -0.8312757 ,  0.35001817,\n",
              "             -0.09286804, -0.15107964,  0.06492356, -1.2146746 ,\n",
              "             -0.14433125, -0.18002185,  0.7040861 , -1.3227665 ]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.14642547, -0.15623635, -0.19250678, -0.26548976,\n",
              "             -0.3109286 , -0.56687576,  0.26907375, -0.84600353,\n",
              "             -1.1520443 ,  0.54624534, -0.40708056, -0.01210441,\n",
              "              0.83606476, -0.83859223,  0.10819665, -0.04827357]],\n",
              "   \n",
              "           [[ 0.640795  , -0.7162036 ,  0.05487156, -0.4891346 ,\n",
              "             -0.43510848,  0.6498185 ,  0.16482961, -0.2608201 ,\n",
              "             -0.31787497,  0.24423534, -0.8147323 ,  0.01593917,\n",
              "              0.672853  , -1.5127211 ,  0.21020289,  0.39767766]],\n",
              "   \n",
              "           [[ 0.6132377 , -0.3795535 ,  0.54358864, -1.6840773 ,\n",
              "             -0.11883807,  0.30954224,  0.24701884,  0.21263517,\n",
              "              0.37575063, -0.05941815, -0.3314252 ,  0.34354508,\n",
              "              0.28724185, -1.7988123 ,  0.49855226,  0.2384835 ]],\n",
              "   \n",
              "           [[ 0.02211616,  0.4224189 ,  0.45817968, -0.12779358,\n",
              "              0.19232306, -0.23025335,  0.5845966 ,  0.19912316,\n",
              "              0.78784364, -1.1737162 ,  0.17383021,  0.40885365,\n",
              "             -0.286033  , -0.9435978 ,  0.34940186,  0.10727476]],\n",
              "   \n",
              "           [[-0.87691176,  0.64875495, -0.6517925 ,  0.09606696,\n",
              "             -0.11612179, -0.84104043, -0.09136637, -0.03662461,\n",
              "             -0.31110504,  0.19353914,  0.9552088 , -1.2929926 ,\n",
              "              0.20630983, -0.9480863 ,  0.28023595, -1.1748906 ]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.01952863,  0.47624204, -0.69386846, -1.4002706 ,\n",
              "             -0.40652215, -1.4287274 ,  0.67923427, -0.4199842 ,\n",
              "             -0.88274467,  0.33637762, -0.47203776, -0.05720491,\n",
              "              0.06159073,  0.21791393, -0.67689073,  0.54870194]],\n",
              "   \n",
              "           [[ 0.34692904,  0.32679164, -0.24702366, -0.49098933,\n",
              "             -0.01134646, -1.5313402 ,  0.17282984, -1.0425565 ,\n",
              "             -0.53817034, -0.7353689 ,  0.5224685 , -0.09396771,\n",
              "             -0.06279413,  0.3096483 , -0.628696  ,  0.5066678 ]],\n",
              "   \n",
              "           [[ 0.1429163 , -0.1959934 ,  0.34064713, -0.08202139,\n",
              "             -0.6971636 , -1.3674848 ,  0.11166536, -1.2122682 ,\n",
              "              0.08585708, -1.1984037 ,  0.3784086 , -0.2768002 ,\n",
              "              0.51255816,  0.23424909, -0.55089676,  0.38809606]],\n",
              "   \n",
              "           [[-0.43227205, -0.37278682,  0.37969545,  0.2752031 ,\n",
              "             -1.060796  , -0.7388941 ,  0.30958918, -0.9514509 ,\n",
              "              0.73317814, -0.68725777,  0.40185162, -0.96248984,\n",
              "              0.5688042 ,  0.2788079 , -0.70461935, -0.04007906]],\n",
              "   \n",
              "           [[-1.457102  ,  0.13370638, -0.06872683,  0.3552011 ,\n",
              "             -1.542418  , -1.0843675 ,  0.42679232, -0.73780274,\n",
              "              0.17243275,  0.08256423, -0.19578129, -0.6366778 ,\n",
              "              1.6170475 , -0.0634924 , -0.30125785, -1.2607358 ]]]],\n",
              "         dtype=float32)},\n",
              "  'Conv_1': {'bias': array([-0.02760017,  0.03291213, -0.01298245, -0.05640659,  0.13788472,\n",
              "          -0.0685158 ,  0.08101743,  0.04094861,  0.12840098,  0.00504573,\n",
              "          -0.00127399,  0.01530092,  0.01746071,  0.00490984, -0.08981805,\n",
              "          -0.00567709,  0.15777013,  0.03950912, -0.02732002, -0.17827073,\n",
              "           0.02474667,  0.22368672,  0.07854657,  0.08093248, -0.20384686,\n",
              "          -0.10459942,  0.06327984, -0.03061697,  0.02760119, -0.00180869,\n",
              "           0.0445908 , -0.19292884], dtype=float32),\n",
              "   'kernel': array([[[[ 1.93409380e-02,  4.37839657e-01, -6.66991711e-01, ...,\n",
              "             -7.23776102e-01, -7.39609301e-02, -1.27079532e-01],\n",
              "            [-4.34368640e-01, -4.28436428e-01,  6.91177607e-01, ...,\n",
              "             -3.70928973e-01,  8.47150981e-01, -3.09342027e-01],\n",
              "            [ 3.67305011e-01,  5.43924868e-01,  5.40041983e-01, ...,\n",
              "             -1.48139060e-01,  3.59097868e-01,  5.50767720e-01],\n",
              "            ...,\n",
              "            [-2.74424195e-01,  5.84714890e-01, -3.13240886e-01, ...,\n",
              "             -5.90839460e-02, -3.95846814e-02,  5.05587831e-02],\n",
              "            [ 2.30384350e-01,  1.48383558e-01,  5.67139924e-01, ...,\n",
              "             -5.81228316e-01, -3.44901383e-01,  3.63096297e-02],\n",
              "            [-7.53403455e-02,  8.57335985e-01,  3.11225295e-01, ...,\n",
              "              3.10860854e-02,  6.00185215e-01,  1.12143481e+00]],\n",
              "   \n",
              "           [[ 4.18420494e-01,  3.37611209e-03, -6.42247319e-01, ...,\n",
              "             -2.71654546e-01,  1.60484999e-01, -3.77947718e-01],\n",
              "            [ 1.83762074e-01, -7.67965138e-01,  9.68019009e-01, ...,\n",
              "             -6.18985176e-01,  6.90331042e-01, -7.48557985e-01],\n",
              "            [ 4.27974761e-01,  3.29997718e-01,  2.21355885e-01, ...,\n",
              "              2.28926390e-01,  3.59276950e-01,  5.42974710e-01],\n",
              "            ...,\n",
              "            [-1.78929463e-01,  5.07973790e-01,  7.06370967e-03, ...,\n",
              "             -2.50161951e-03, -2.02916324e-01,  6.57196492e-02],\n",
              "            [ 2.05996856e-01, -7.58901983e-03,  3.20194811e-01, ...,\n",
              "             -4.74830091e-01, -3.97116452e-01,  9.96367708e-02],\n",
              "            [-4.83255863e-01,  3.68727922e-01,  3.85358254e-03, ...,\n",
              "             -8.59798044e-02,  6.68098867e-01,  8.82670701e-01]],\n",
              "   \n",
              "           [[ 1.14338852e-01,  1.03844710e-01, -1.17405839e-01, ...,\n",
              "             -5.21285951e-01, -5.42017519e-01, -3.20152819e-01],\n",
              "            [-3.63802135e-01, -7.52815232e-02,  6.13839924e-01, ...,\n",
              "             -1.02392721e+00,  6.67807579e-01, -7.16309965e-01],\n",
              "            [ 2.75118828e-01,  8.20544437e-02, -2.93038934e-02, ...,\n",
              "              2.04075336e-01,  2.66809255e-01,  5.06089628e-01],\n",
              "            ...,\n",
              "            [-4.52428520e-01,  6.35264933e-01, -4.29104082e-02, ...,\n",
              "             -3.79726112e-01, -3.94368768e-01,  2.30973125e-01],\n",
              "            [ 9.04610679e-02, -2.10014358e-02,  6.54927492e-02, ...,\n",
              "             -6.18047655e-01, -6.28775239e-01,  1.92247406e-01],\n",
              "            [-1.38358280e-01,  4.10118103e-01, -2.51401305e-01, ...,\n",
              "             -5.99184372e-02, -1.89065993e-01,  7.02512681e-01]],\n",
              "   \n",
              "           [[ 2.34839041e-02, -4.65713084e-01, -2.09279120e-01, ...,\n",
              "             -4.02883887e-01, -9.18178320e-01, -4.51240331e-01],\n",
              "            [ 6.46275207e-02, -1.51290614e-02,  9.04622972e-01, ...,\n",
              "             -1.53694177e+00,  5.79644084e-01, -4.93099570e-01],\n",
              "            [ 5.53627133e-01,  3.77390444e-01, -9.67953652e-02, ...,\n",
              "             -6.41859174e-01, -2.49813363e-01,  3.36562008e-01],\n",
              "            ...,\n",
              "            [-7.54974484e-01,  6.17233753e-01, -1.74103856e-01, ...,\n",
              "             -3.25598419e-02, -4.94956583e-01,  3.15790921e-01],\n",
              "            [-3.39362631e-03, -3.07963967e-01, -1.93736658e-01, ...,\n",
              "             -3.43166262e-01, -2.39683792e-01,  4.31233913e-01],\n",
              "            [-1.57944083e-01,  6.90853655e-01, -5.27632758e-02, ...,\n",
              "             -2.21426979e-01, -1.89001799e-01,  7.99863338e-01]],\n",
              "   \n",
              "           [[-1.83288038e-01, -2.87166178e-01, -1.22892387e-01, ...,\n",
              "              1.04839988e-01, -4.47755933e-01, -5.55149555e-01],\n",
              "            [ 3.26161712e-01,  1.06454313e-01,  1.16801381e+00, ...,\n",
              "             -8.55114043e-01, -6.50674477e-03, -3.75847250e-01],\n",
              "            [-4.03704047e-02,  1.75555557e-01, -1.06764473e-01, ...,\n",
              "             -4.08542715e-02, -5.91866300e-02,  1.17597997e-01],\n",
              "            ...,\n",
              "            [ 4.97081503e-02, -5.27068265e-02, -9.21087742e-01, ...,\n",
              "              3.87819633e-02, -1.96879283e-01,  1.82981178e-01],\n",
              "            [ 4.57194507e-01, -1.00358367e+00, -7.23270655e-01, ...,\n",
              "             -8.18398535e-01, -3.57577473e-01,  8.85840654e-01],\n",
              "            [ 2.19396710e-01,  4.82242316e-01,  2.88781911e-01, ...,\n",
              "             -1.49027511e-01, -9.88098513e-03,  4.60391372e-01]]],\n",
              "   \n",
              "   \n",
              "          [[[ 2.95673639e-01,  4.76280719e-01, -3.84605467e-01, ...,\n",
              "             -4.20189649e-01, -1.27291337e-01,  4.38194752e-01],\n",
              "            [-4.96813923e-01,  9.02814493e-02,  4.01683688e-01, ...,\n",
              "             -6.22051477e-01,  1.15568316e+00, -1.30609170e-01],\n",
              "            [ 2.63060927e-01,  1.49966609e-02, -3.16068023e-01, ...,\n",
              "              1.19753042e-03,  2.25427181e-01,  7.41740882e-01],\n",
              "            ...,\n",
              "            [ 1.28917560e-01,  5.60272515e-01,  1.83866709e-01, ...,\n",
              "             -2.12920055e-01, -1.34316787e-01,  4.26955670e-01],\n",
              "            [-3.48477997e-02, -4.98075876e-03,  8.86525273e-01, ...,\n",
              "             -1.44057617e-01, -3.84804547e-01,  1.38254821e-01],\n",
              "            [-8.66834283e-01,  6.46388471e-01,  1.89818870e-02, ...,\n",
              "             -3.12964618e-01,  7.28416264e-01,  4.32728320e-01]],\n",
              "   \n",
              "           [[ 8.07334661e-01, -1.41607240e-01, -1.62304446e-01, ...,\n",
              "             -2.83012986e-01,  1.53815607e-02, -2.30119959e-01],\n",
              "            [-5.50592124e-01, -1.41082034e-01,  5.56094587e-01, ...,\n",
              "             -8.45838368e-01,  8.15022349e-01, -3.48706067e-01],\n",
              "            [ 7.19359815e-01, -2.89453477e-01, -4.68004614e-01, ...,\n",
              "              5.26906103e-02, -1.37358993e-01,  5.29533327e-01],\n",
              "            ...,\n",
              "            [-1.31362960e-01,  2.40817606e-01,  5.67120798e-02, ...,\n",
              "             -1.09539293e-01, -4.13109988e-01, -2.26823881e-01],\n",
              "            [-1.34351820e-01, -8.50936994e-02,  3.36807013e-01, ...,\n",
              "             -6.04969561e-01, -1.27915010e-01, -1.00876026e-01],\n",
              "            [-5.88893056e-01,  6.07379854e-01, -2.18716025e-01, ...,\n",
              "             -4.13172022e-02, -2.33688802e-01,  3.13475996e-01]],\n",
              "   \n",
              "           [[-1.45533442e-01,  6.46763891e-02,  1.17146097e-01, ...,\n",
              "             -6.15812242e-01, -4.80819613e-01,  3.28393579e-01],\n",
              "            [-3.35242242e-01, -1.36123911e-01,  3.29814285e-01, ...,\n",
              "             -8.10597658e-01,  4.38578278e-01, -2.11788282e-01],\n",
              "            [ 8.38776678e-03,  7.88218081e-02, -5.47903121e-01, ...,\n",
              "             -5.79102673e-02,  1.55635566e-01,  2.15576887e-01],\n",
              "            ...,\n",
              "            [-3.96276176e-01, -1.55959865e-02,  1.79747000e-01, ...,\n",
              "             -1.61173820e-01, -4.78789747e-01,  1.69808511e-02],\n",
              "            [ 5.65232374e-02,  9.54218730e-02,  7.10986480e-02, ...,\n",
              "             -2.87606061e-01,  1.57171525e-02, -7.33006075e-02],\n",
              "            [-4.26266342e-01, -6.17199019e-02, -1.12541187e+00, ...,\n",
              "             -2.55165607e-01, -5.01993239e-01,  2.58967012e-01]],\n",
              "   \n",
              "           [[ 2.95217574e-01, -2.02662975e-01,  1.29130945e-01, ...,\n",
              "             -4.69617248e-01, -4.58077937e-01,  2.09451869e-01],\n",
              "            [-1.22569621e-01,  2.91074932e-01,  1.81581303e-01, ...,\n",
              "             -1.64588666e+00,  4.87268209e-01, -2.89495200e-01],\n",
              "            [ 3.86361219e-02,  1.73146695e-01, -9.60025847e-01, ...,\n",
              "              3.10017139e-01,  3.39673348e-02,  3.67379993e-01],\n",
              "            ...,\n",
              "            [-4.73732471e-01, -7.03745723e-01, -1.79683000e-01, ...,\n",
              "             -3.83916616e-01, -1.14790881e+00,  3.26873899e-01],\n",
              "            [ 4.77913395e-02, -4.26383168e-02,  8.12185407e-02, ...,\n",
              "             -6.07563734e-01,  3.79176810e-02, -1.38891384e-01],\n",
              "            [-4.30928290e-01,  1.93584263e-01, -2.96655059e-01, ...,\n",
              "             -2.21259370e-01, -2.19555035e-01,  1.57217503e-01]],\n",
              "   \n",
              "           [[ 1.11104973e-01, -3.91808376e-02,  2.23053828e-01, ...,\n",
              "             -1.97398867e-02, -8.02847147e-02,  2.35897198e-01],\n",
              "            [ 3.10187578e-01,  4.35037076e-01,  1.22367427e-01, ...,\n",
              "             -7.08229721e-01,  1.70117561e-02, -2.39615798e-01],\n",
              "            [-1.90065682e-01,  2.89076447e-01, -8.93187582e-01, ...,\n",
              "              3.50249618e-01,  4.46459860e-01,  2.14253560e-01],\n",
              "            ...,\n",
              "            [-6.18785471e-02, -5.01178741e-01, -7.43322015e-01, ...,\n",
              "             -1.08384156e+00, -9.77658868e-01,  2.61147946e-01],\n",
              "            [ 6.16663098e-01, -2.43775472e-01, -4.44017977e-01, ...,\n",
              "             -1.04191804e+00, -7.97489099e-03,  3.74946415e-01],\n",
              "            [-1.10652469e-01, -3.99445370e-02,  1.43511266e-01, ...,\n",
              "             -2.75350630e-01, -3.67354125e-01,  3.94483835e-01]]],\n",
              "   \n",
              "   \n",
              "          [[[ 1.67170893e-02,  9.53174829e-02, -1.07688136e-01, ...,\n",
              "             -3.86543661e-01,  6.91875741e-02,  1.08372048e-01],\n",
              "            [-1.27954647e-01, -2.88562596e-01,  1.59105763e-01, ...,\n",
              "             -5.95474899e-01,  5.63416004e-01,  4.71698642e-01],\n",
              "            [ 6.85901582e-01, -6.02608323e-01, -2.71147937e-01, ...,\n",
              "             -7.94915631e-02, -2.49939442e-01,  3.08073729e-01],\n",
              "            ...,\n",
              "            [ 2.70253867e-01,  3.05164069e-01,  2.09265083e-01, ...,\n",
              "              1.62355527e-02, -2.05540136e-01,  8.78459990e-01],\n",
              "            [-4.48021054e-01,  1.35036007e-01,  2.64051527e-01, ...,\n",
              "             -4.88250136e-01, -1.40987068e-01, -1.37653843e-01],\n",
              "            [ 4.22791809e-01,  3.31090003e-01,  5.62108994e-01, ...,\n",
              "              1.55920923e-01, -2.43686318e-01, -2.65344858e-01]],\n",
              "   \n",
              "           [[ 2.76998878e-01,  3.67330909e-02, -1.23030216e-01, ...,\n",
              "              3.02734673e-01,  2.27175906e-01, -5.18199265e-01],\n",
              "            [-2.51546830e-01, -6.44984394e-02,  3.08656335e-01, ...,\n",
              "             -6.78127766e-01,  6.84381962e-01,  3.83782685e-02],\n",
              "            [ 3.26370239e-01, -6.05127990e-01, -4.75691378e-01, ...,\n",
              "             -1.74719468e-01,  2.04801857e-02, -3.21633279e-01],\n",
              "            ...,\n",
              "            [ 1.64103001e-01,  1.11707009e-01, -1.16132507e-02, ...,\n",
              "             -6.29961342e-02, -8.91497791e-01,  2.73356855e-01],\n",
              "            [-1.90150291e-01,  1.29223699e-02,  1.67527333e-01, ...,\n",
              "             -3.00577939e-01, -3.28402281e-01, -2.41968080e-01],\n",
              "            [ 3.26371759e-01,  2.60777354e-01, -2.51248538e-01, ...,\n",
              "              1.85524762e-01,  1.14957064e-01,  2.09143683e-01]],\n",
              "   \n",
              "           [[ 1.93426952e-01, -4.06778641e-02,  6.10352755e-01, ...,\n",
              "              4.70559485e-02,  7.14075044e-02,  2.21819952e-01],\n",
              "            [-2.44016245e-01, -1.55626267e-01, -1.71755165e-01, ...,\n",
              "             -1.00538266e+00,  3.56686473e-01,  1.71289966e-01],\n",
              "            [ 1.84422970e-01, -3.39431942e-01, -6.47951543e-01, ...,\n",
              "              6.42830059e-02, -4.94276375e-01,  1.53645948e-01],\n",
              "            ...,\n",
              "            [ 1.58788376e-02, -3.96503896e-01,  1.57040626e-01, ...,\n",
              "             -1.88980043e-01, -1.06870818e+00,  4.56516892e-02],\n",
              "            [-1.47628441e-01, -2.35158503e-01,  2.36288950e-01, ...,\n",
              "             -5.80847740e-01, -2.47601241e-01,  8.36751796e-03],\n",
              "            [ 6.62571341e-02, -1.52890369e-01, -1.07878637e+00, ...,\n",
              "              2.13919785e-02, -3.89065832e-01,  4.70739603e-02]],\n",
              "   \n",
              "           [[-1.44107416e-01,  1.65748999e-01,  4.89296049e-01, ...,\n",
              "             -2.35444799e-01,  1.72903046e-01,  1.81271195e-01],\n",
              "            [-1.78125069e-01,  2.62436837e-01, -4.75191355e-01, ...,\n",
              "             -8.51388097e-01, -3.44089344e-02, -3.57372105e-01],\n",
              "            [-8.19619894e-02, -5.51383533e-02, -7.84157097e-01, ...,\n",
              "              7.33141527e-02, -3.97008061e-01, -2.13852420e-01],\n",
              "            ...,\n",
              "            [-3.05012446e-02, -5.95055521e-01, -4.56760898e-02, ...,\n",
              "             -7.13809207e-02, -7.92183518e-01,  1.31211743e-01],\n",
              "            [-2.79496443e-02, -2.68506408e-01,  1.82104677e-01, ...,\n",
              "             -5.25642455e-01, -9.56046134e-02, -1.90744773e-01],\n",
              "            [ 1.10483758e-01, -2.15391606e-01, -7.40097165e-01, ...,\n",
              "             -5.38899973e-02, -2.00448975e-01,  6.40579611e-02]],\n",
              "   \n",
              "           [[-2.91459531e-01, -4.54567671e-01,  7.27930605e-01, ...,\n",
              "              3.69140714e-01,  2.89630115e-01,  9.12751704e-02],\n",
              "            [-1.13908269e-01, -1.06843956e-01, -3.85068566e-01, ...,\n",
              "             -3.79876107e-01,  3.04321982e-02, -2.68946618e-01],\n",
              "            [ 1.30098835e-01,  1.50853708e-01, -3.76450688e-01, ...,\n",
              "              2.34347656e-01, -4.56552543e-02, -4.14679348e-01],\n",
              "            ...,\n",
              "            [ 3.12749743e-01, -3.88258040e-01, -8.31063330e-01, ...,\n",
              "             -2.86966562e-01, -2.87693262e-01, -3.34305525e-01],\n",
              "            [ 3.82123500e-01, -1.43026233e-01, -2.53730297e-01, ...,\n",
              "             -1.15262151e+00, -3.88139188e-02, -2.07122475e-01],\n",
              "            [ 6.74387142e-02,  2.12765317e-02, -5.63384414e-01, ...,\n",
              "              2.27069601e-01,  7.03759342e-02,  5.36947884e-02]]],\n",
              "   \n",
              "   \n",
              "          [[[-2.79833525e-01,  2.52159357e-01, -4.98240978e-01, ...,\n",
              "             -4.77380782e-01,  6.35032415e-01, -1.51022539e-01],\n",
              "            [-6.49457946e-02, -9.57076490e-01, -3.31931119e-03, ...,\n",
              "             -3.01123559e-01,  1.93327382e-01, -2.42650747e-01],\n",
              "            [ 4.45620239e-01, -3.30278248e-01, -1.07769109e-01, ...,\n",
              "             -6.00389540e-02, -4.48277146e-01,  1.58047974e-01],\n",
              "            ...,\n",
              "            [-2.82116123e-02,  2.63626546e-01, -2.22636223e-01, ...,\n",
              "             -4.79445457e-01, -1.05214167e+00,  8.79570127e-01],\n",
              "            [-5.16280293e-01,  1.56490039e-02, -5.22186160e-02, ...,\n",
              "             -3.67123276e-01, -2.76900083e-01, -4.01629433e-02],\n",
              "            [ 4.34616357e-01,  4.80685890e-01, -6.10900283e-01, ...,\n",
              "             -2.15198606e-01, -2.45255485e-01, -2.44225726e-01]],\n",
              "   \n",
              "           [[ 1.93502635e-01,  1.91306025e-01,  1.74866006e-01, ...,\n",
              "              5.47661304e-01,  3.36911261e-01, -1.99648887e-01],\n",
              "            [-4.83622774e-02, -9.94523048e-01, -3.53868194e-02, ...,\n",
              "             -8.01596761e-01,  5.34495056e-01, -7.66399086e-01],\n",
              "            [ 2.07049459e-01, -3.62225711e-01, -4.49925244e-01, ...,\n",
              "             -1.50531158e-01, -2.10322380e-01,  1.12480344e-02],\n",
              "            ...,\n",
              "            [ 1.90320641e-01,  3.99274707e-01, -5.21654263e-02, ...,\n",
              "             -3.60407919e-01, -9.56774414e-01,  7.03654706e-01],\n",
              "            [-5.15239239e-01, -3.11197750e-02,  1.04926810e-01, ...,\n",
              "             -3.26606542e-01, -3.19021344e-01, -2.52619982e-01],\n",
              "            [ 6.58732235e-01,  4.30997461e-01, -7.75587261e-01, ...,\n",
              "             -2.25855187e-01, -3.75657141e-01, -7.44265676e-01]],\n",
              "   \n",
              "           [[-7.33833015e-03,  3.64060290e-02,  2.34130815e-01, ...,\n",
              "              9.60319862e-02, -5.16445823e-02,  3.76931012e-01],\n",
              "            [-8.50950256e-02, -5.49605191e-01, -2.32021108e-01, ...,\n",
              "             -7.57622659e-01,  2.65451521e-01, -1.10150866e-01],\n",
              "            [ 2.25077748e-01, -1.02580972e-02, -5.40379584e-01, ...,\n",
              "             -4.68587905e-01, -4.11452234e-01, -1.70570239e-01],\n",
              "            ...,\n",
              "            [ 6.09730482e-02, -2.86603514e-02, -2.03581169e-01, ...,\n",
              "             -2.29631543e-01, -6.63585424e-01,  3.12582403e-01],\n",
              "            [-3.45134079e-01, -2.81464726e-01,  4.65398550e-01, ...,\n",
              "             -2.25250453e-01, -4.61396426e-01,  2.87725329e-01],\n",
              "            [ 8.72827768e-01,  2.06782818e-02, -8.18163753e-01, ...,\n",
              "             -6.71546236e-02, -4.57488209e-01, -8.80533159e-02]],\n",
              "   \n",
              "           [[-5.82750142e-03, -2.41864145e-01,  3.18555713e-01, ...,\n",
              "             -2.28457808e-01, -4.56747860e-02,  1.31476581e-01],\n",
              "            [-1.29174083e-01, -5.49212575e-01, -7.85232410e-02, ...,\n",
              "             -5.91924489e-01,  4.60150570e-01, -5.01730144e-01],\n",
              "            [-1.41991675e-01,  4.75257158e-01, -7.89966941e-01, ...,\n",
              "              2.84605205e-01, -9.02540535e-02, -9.78253409e-02],\n",
              "            ...,\n",
              "            [ 1.60378069e-01, -6.96575791e-02, -4.98365074e-01, ...,\n",
              "              2.66323145e-03, -6.47430599e-01,  2.84479886e-01],\n",
              "            [-2.34845638e-01, -2.62286305e-01,  1.66502222e-01, ...,\n",
              "             -2.63371438e-01, -2.63272285e-01,  3.09270956e-02],\n",
              "            [ 6.57959640e-01,  4.86605130e-02, -4.48151708e-01, ...,\n",
              "             -2.42799358e-03, -7.18423307e-01, -4.82602119e-01]],\n",
              "   \n",
              "           [[-3.63265485e-01, -2.88323760e-01,  4.95429933e-01, ...,\n",
              "             -1.99220907e-02, -1.70863062e-01,  4.59049314e-01],\n",
              "            [ 2.81812437e-02,  1.18758589e-01, -1.17806651e-01, ...,\n",
              "             -2.36826658e-01,  5.49137712e-01, -3.19427639e-01],\n",
              "            [ 2.76980072e-01,  4.95717287e-01, -3.48231018e-01, ...,\n",
              "              1.16651520e-01, -1.79123387e-01, -9.25059095e-02],\n",
              "            ...,\n",
              "            [ 2.42278174e-01,  8.62821862e-02, -6.85412467e-01, ...,\n",
              "              4.26259302e-02, -1.78116828e-01,  4.34627205e-01],\n",
              "            [ 2.00019613e-01, -9.43134502e-02, -6.34221077e-01, ...,\n",
              "              5.21364361e-02,  4.16678824e-02, -1.87922865e-01],\n",
              "            [ 1.19303513e+00, -5.10634109e-02, -2.41158865e-02, ...,\n",
              "              4.03079301e-01, -2.16853041e-02, -1.36434436e-01]]],\n",
              "   \n",
              "   \n",
              "          [[[ 2.54789472e-01,  5.15871584e-01,  4.75436568e-01, ...,\n",
              "             -2.16773987e-01,  5.65230131e-01, -1.65312678e-01],\n",
              "            [-2.32921522e-02, -2.82801300e-01, -5.23564041e-01, ...,\n",
              "             -3.57360065e-01, -1.48165256e-01, -1.10222769e+00],\n",
              "            [ 1.86956987e-01,  1.14827631e-02, -4.12610471e-01, ...,\n",
              "              3.59022081e-01, -8.80226433e-01, -9.59637821e-01],\n",
              "            ...,\n",
              "            [ 3.49205226e-01,  5.18787690e-02, -6.48970842e-01, ...,\n",
              "             -6.80925697e-02, -9.71033871e-01,  3.71053517e-01],\n",
              "            [ 1.06672093e-01,  4.56987411e-01, -5.50073385e-01, ...,\n",
              "             -3.83239277e-02, -6.82369620e-02, -1.29782349e-01],\n",
              "            [ 6.06342852e-01,  1.09749007e+00, -1.51097894e+00, ...,\n",
              "             -3.93215939e-02, -6.40115380e-01, -1.23985827e+00]],\n",
              "   \n",
              "           [[-5.34689426e-02, -7.45778382e-02, -6.25535697e-02, ...,\n",
              "              8.12848151e-01, -1.10639811e-01, -9.78362784e-02],\n",
              "            [-2.13911220e-01, -4.01134133e-01, -3.86692643e-01, ...,\n",
              "             -5.14686704e-01,  5.78713775e-01, -1.34710467e+00],\n",
              "            [ 4.46226358e-01, -1.72637329e-01, -5.44561565e-01, ...,\n",
              "             -1.14198053e+00, -6.33693755e-01, -2.46056929e-01],\n",
              "            ...,\n",
              "            [-1.77195206e-01,  5.78568220e-01, -3.06802273e-01, ...,\n",
              "             -3.48290890e-01, -4.47919846e-01,  9.87048090e-01],\n",
              "            [ 2.63735559e-02,  2.80437112e-01, -2.08486617e-01, ...,\n",
              "              3.53254117e-02, -1.59646019e-01,  5.54617643e-01],\n",
              "            [ 9.85325873e-01,  5.87071717e-01, -4.55872148e-01, ...,\n",
              "             -7.57255793e-01, -3.70901585e-01, -3.08952779e-01]],\n",
              "   \n",
              "           [[ 1.98557571e-01, -5.79725616e-02,  1.40944839e-01, ...,\n",
              "              2.16377288e-01, -9.22867805e-02,  3.97195071e-01],\n",
              "            [ 2.19786149e-02, -3.89066264e-02, -1.36507124e-01, ...,\n",
              "             -1.15258408e+00,  3.30686122e-01, -1.26942182e+00],\n",
              "            [ 3.29772830e-01,  3.78849596e-01,  5.41092753e-02, ...,\n",
              "             -1.36918277e-01, -4.15365279e-01,  2.38226742e-01],\n",
              "            ...,\n",
              "            [-3.73573452e-02,  1.94232255e-01,  1.62928045e-01, ...,\n",
              "             -6.87490255e-02, -2.26325721e-01,  5.54815590e-01],\n",
              "            [-5.53735197e-01,  3.62216741e-01, -1.61763415e-01, ...,\n",
              "              2.44010687e-02, -2.17370406e-01,  8.68129015e-01],\n",
              "            [ 8.08752179e-01,  2.56485850e-01, -3.90894443e-01, ...,\n",
              "             -8.84532571e-01, -5.27577214e-02,  1.45421743e-01]],\n",
              "   \n",
              "           [[-1.98046014e-01,  2.05180086e-02,  1.28753893e-02, ...,\n",
              "              4.01559144e-01, -2.74763227e-01, -1.01894677e-01],\n",
              "            [-4.07268740e-02,  1.44918665e-01, -2.72733212e-01, ...,\n",
              "             -8.11241329e-01,  3.62263083e-01, -6.21156335e-01],\n",
              "            [ 3.88089716e-01,  6.06497109e-01, -3.34994555e-01, ...,\n",
              "             -3.16238344e-01, -3.33681583e-01,  6.36197805e-01],\n",
              "            ...,\n",
              "            [-5.13920009e-01,  2.75131643e-01, -3.48353207e-01, ...,\n",
              "              1.63679525e-01, -1.73923224e-01,  1.10774326e+00],\n",
              "            [-7.56895006e-01,  2.76660860e-01, -5.82089186e-01, ...,\n",
              "              2.47609466e-01,  1.20833799e-01,  5.00535548e-01],\n",
              "            [ 6.73231602e-01, -1.86233252e-01,  2.73636281e-01, ...,\n",
              "             -1.90438673e-01, -4.74688172e-01,  5.33869505e-01]],\n",
              "   \n",
              "           [[-3.15676890e-02, -7.73656368e-01,  1.59730062e-01, ...,\n",
              "             -2.49515429e-01, -4.04987305e-01, -2.36988917e-01],\n",
              "            [ 2.68811006e-02,  5.72983146e-01, -2.77341038e-01, ...,\n",
              "             -3.64129245e-01,  7.14195549e-01,  1.77737931e-03],\n",
              "            [ 5.64896405e-01,  6.09612167e-01,  1.37356773e-01, ...,\n",
              "             -9.77230594e-02, -4.81889337e-01,  6.58134162e-01],\n",
              "            ...,\n",
              "            [-6.03392780e-01, -9.57781002e-02, -6.00157738e-01, ...,\n",
              "              4.11736399e-01,  3.52460653e-01,  1.18755233e+00],\n",
              "            [-1.07696688e+00,  7.77849495e-01, -4.79456961e-01, ...,\n",
              "              3.18215102e-01,  8.16803336e-01, -6.30724579e-02],\n",
              "            [ 6.07903898e-01, -8.68975103e-01,  5.83210945e-01, ...,\n",
              "              2.82276943e-02, -4.41761643e-01,  5.42042911e-01]]]],\n",
              "         dtype=float32)},\n",
              "  'Dense_0': {'bias': array([ 0.09042518,  0.9883207 , -0.5461252 ,  0.5591024 , -1.014932  ,\n",
              "           0.03685181, -0.5483948 ,  0.17893407,  0.09069423,  0.42663318],\n",
              "         dtype=float32),\n",
              "   'kernel': array([[ 1.0585502 , -1.3956808 ,  0.34636924, -0.9243253 ,  0.6699518 ,\n",
              "            0.3478436 , -1.3132898 ,  1.0497571 , -1.5761775 , -0.09053539],\n",
              "          [ 0.32554245,  0.65131235,  0.49186245,  0.80666155, -0.4104941 ,\n",
              "           -0.11780734, -0.0439073 , -1.242046  , -0.13078369, -2.0075583 ],\n",
              "          [-0.03930703,  0.42648882, -0.15188728, -0.28224364,  0.8943288 ,\n",
              "           -3.0429964 , -2.3474329 ,  1.0505046 ,  0.55783266,  0.8439393 ],\n",
              "          [-1.8106714 , -1.7483554 , -1.0383319 ,  0.70590925,  0.8624324 ,\n",
              "            0.6231594 , -1.6662395 ,  0.4543369 ,  0.55865437,  0.56678826],\n",
              "          [-0.8799365 ,  0.5406505 ,  1.0845698 ,  1.3094565 , -0.76142496,\n",
              "           -1.0557452 , -1.8899986 ,  1.1710124 , -0.07026887, -1.4067062 ],\n",
              "          [ 0.6540665 ,  0.6292495 , -2.1739986 , -1.8979679 , -0.40589637,\n",
              "            0.22870934,  0.25991988,  0.50809854,  0.76795137, -0.14191902],\n",
              "          [ 0.6679607 , -1.9748625 ,  0.98862076,  0.6099043 , -2.3751013 ,\n",
              "           -0.23683429, -1.6925834 ,  0.9011968 , -0.02695877,  0.33320114],\n",
              "          [ 0.2799459 ,  0.9547268 , -0.74576557,  1.2307259 , -0.01783373,\n",
              "            0.49792215, -0.47266635, -1.6590598 ,  0.7549098 , -2.2042775 ],\n",
              "          [ 0.5997754 , -2.9535203 ,  0.30706394,  0.2468266 , -2.768543  ,\n",
              "            0.67560655,  0.9843934 , -1.5188024 ,  0.6699628 ,  0.4690546 ],\n",
              "          [ 0.6613665 ,  0.16088574, -0.3168095 ,  0.8122926 , -0.8657408 ,\n",
              "            1.0023273 , -0.35324186,  0.24720134, -2.308821  , -0.20775384],\n",
              "          [ 0.7620061 , -1.4445834 ,  0.67909336,  0.9022798 , -1.9370795 ,\n",
              "           -1.1674767 , -1.6605536 ,  0.8177995 ,  0.22709715,  0.7079008 ],\n",
              "          [-1.2467344 , -1.6212808 ,  0.5991567 , -1.7618549 ,  1.0668769 ,\n",
              "            1.0039266 ,  0.38810116,  0.22538279,  0.25290293, -0.8203572 ],\n",
              "          [ 1.2433797 ,  0.29689768, -0.76886374, -0.8980173 ,  0.03523094,\n",
              "           -0.52442664, -1.575716  ,  0.27575853, -0.99056035,  0.6911181 ],\n",
              "          [ 0.7539889 ,  0.91173303,  0.77747923, -1.0377945 ,  0.29228503,\n",
              "           -1.0293409 ,  0.2873672 ,  0.11681654,  0.3716995 , -2.1477218 ],\n",
              "          [-0.5412877 , -0.958979  , -2.1284919 ,  1.5140238 ,  0.648908  ,\n",
              "            1.0241888 , -1.1419033 , -1.5348314 ,  0.8750265 , -0.26669705],\n",
              "          [-1.1223868 ,  1.1346172 , -0.3250557 , -1.2305187 , -0.07616247,\n",
              "            0.7045931 ,  1.035662  ,  0.7788254 , -0.02310352, -1.1854041 ],\n",
              "          [ 0.23997088,  0.8244198 , -1.1998576 , -2.548607  ,  0.74460965,\n",
              "           -0.10683431,  0.6283817 , -0.22918731, -0.7731416 , -0.38630196],\n",
              "          [-2.2417324 , -0.3725433 ,  0.23350902,  0.47917002,  0.64841974,\n",
              "            0.13434617, -2.549328  ,  0.6658044 ,  0.3580458 ,  0.59050244],\n",
              "          [-0.7369954 ,  0.5043032 ,  0.21876381, -1.0767151 ,  0.6187672 ,\n",
              "           -1.5783373 , -0.19088574,  0.8125462 , -2.3482482 ,  0.89920163],\n",
              "          [ 0.34502566, -2.5694413 ,  1.138908  , -0.60091656,  0.4023234 ,\n",
              "           -1.1248877 ,  0.6033688 , -2.1898944 ,  0.9333728 ,  0.0617527 ],\n",
              "          [-2.3115706 ,  0.87495077,  0.21941987,  0.49158254, -1.1033119 ,\n",
              "           -0.00610985,  0.15623653, -0.12987779, -0.0570875 ,  0.77911097],\n",
              "          [-0.01991823, -2.3508358 , -1.8154157 , -1.3452514 , -0.6187231 ,\n",
              "            1.1783906 ,  1.0292022 , -0.54034513,  0.7649426 ,  0.8226675 ],\n",
              "          [ 0.7452354 ,  0.07045756,  0.6198479 ,  1.0426222 , -0.6314761 ,\n",
              "            0.12317616,  0.7753757 , -1.6244898 , -1.0560359 , -0.5074863 ],\n",
              "          [ 0.63647234, -2.0846255 , -0.21029124, -1.2813647 ,  0.9048618 ,\n",
              "           -1.1205572 ,  1.1520034 , -2.8156273 ,  0.30465096,  0.8744729 ],\n",
              "          [-0.32400385, -2.8790267 , -0.82452756,  0.42916846, -1.2816621 ,\n",
              "            0.97144747,  1.049633  , -2.085277  ,  0.74591506,  0.1364304 ],\n",
              "          [-1.6981544 , -0.27329352, -0.48453346,  0.62416697,  0.9788648 ,\n",
              "            0.42749774, -2.0492291 ,  0.44086713, -0.8616808 ,  0.80357397],\n",
              "          [ 0.55568004,  0.07409288,  1.2333273 , -0.81862557, -1.4973769 ,\n",
              "            1.0674679 , -0.5029427 , -0.9597362 , -0.08249622, -0.7234108 ],\n",
              "          [-1.6893413 , -1.5363307 ,  1.4890915 , -0.11975233,  0.883288  ,\n",
              "           -1.7834784 ,  0.85656303,  0.5533208 ,  1.0144161 , -1.2230774 ],\n",
              "          [ 0.44524628, -0.32470143,  0.29902253, -1.467759  ,  1.2573394 ,\n",
              "           -0.97124434,  0.8314855 , -1.8486851 , -1.0785341 ,  0.94371533],\n",
              "          [ 0.6512383 ,  0.35397965, -0.41253412, -3.0363603 ,  0.6052733 ,\n",
              "           -0.6348416 ,  0.5386722 ,  0.40046528, -0.94667107, -0.15558401],\n",
              "          [ 0.07591704,  0.19094907,  0.63252354, -2.176439  ,  0.3965976 ,\n",
              "           -0.7874573 ,  0.6791022 , -2.0750954 ,  0.74295837, -0.20663945],\n",
              "          [-1.9893469 , -0.42655385,  0.7106501 ,  0.76565945, -0.3628365 ,\n",
              "            0.6943133 ,  0.85113996,  0.20940991, -1.2786822 , -1.9053285 ]],\n",
              "         dtype=float32)}},\n",
              " 'step': array(9370, dtype=int32)}"
            ]
          },
          "metadata": {},
          "execution_count": 18
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "step = ckpt_mgr.latest_step()  # step = 4\n",
        "ckpt_mgr.restore(step)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "jxsZNf4z2x3n",
        "outputId": "fcac19d3-2502-4710-c550-22d11ac9d8ac"
      },
      "execution_count": 19,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "{'batch_stats': {'BatchNorm_0': {'mean': array([ 0.59678304,  0.34670404, -0.32364425, -1.5403948 , -0.41289315,\n",
              "          -0.9029092 , -1.0194184 , -1.4013318 , -0.2755111 , -0.40960672,\n",
              "          -0.2929568 , -1.3420625 ,  0.9907112 , -1.3641517 , -0.05670249,\n",
              "          -0.9666436 ], dtype=float32),\n",
              "   'var': array([2.754574  , 0.4765729 , 0.79068047, 3.139733  , 1.9464167 ,\n",
              "          3.3058364 , 3.4047148 , 3.8393342 , 1.8070192 , 2.9429402 ,\n",
              "          0.3669262 , 3.5045764 , 2.1112766 , 2.0194216 , 0.6417175 ,\n",
              "          2.8159597 ], dtype=float32)},\n",
              "  'BatchNorm_1': {'mean': array([-3.7710893 ,  1.0635827 , -5.265705  , -0.9515563 ,  5.7551737 ,\n",
              "          -7.405269  , -0.0529795 , -0.88988537, -2.9906244 ,  5.9144897 ,\n",
              "          -1.8900927 , -3.976402  , -2.58967   , -6.522824  , -2.9571164 ,\n",
              "           1.7212024 ,  0.36768922,  0.04797452, -0.2436241 , -7.0267615 ,\n",
              "           5.0057926 , -4.0950956 ,  1.439795  , -5.6158085 , -4.036963  ,\n",
              "           4.252638  , -0.54707086, -4.9110804 , -3.3844934 , -1.3179133 ,\n",
              "          -4.579265  ,  5.8077955 ], dtype=float32),\n",
              "   'var': array([ 7.0015135,  5.698587 , 10.192861 ,  8.713224 ,  5.925345 ,\n",
              "           7.763242 , 12.205926 ,  8.603589 , 12.3140745,  7.4565024,\n",
              "           8.95226  ,  7.0730014, 10.870935 ,  9.424113 ,  8.020228 ,\n",
              "           4.798607 ,  9.679846 ,  9.174238 , 12.396536 ,  8.061559 ,\n",
              "          10.566014 ,  8.434447 ,  7.7500076, 10.797337 ,  9.642636 ,\n",
              "           8.652558 ,  7.0093684,  7.1579623, 10.251053 , 12.818575 ,\n",
              "           9.238743 ,  8.465704 ], dtype=float32)}},\n",
              " 'key': array([2562233961, 1946702221], dtype=uint32),\n",
              " 'opt_state': [{'count': array(9370, dtype=int32),\n",
              "   'mu': {'BatchNorm_0': {'bias': array([ 0.00011218,  0.00393522,  0.00268865,  0.00872598,  0.00256785,\n",
              "             0.00846969, -0.00938905,  0.03185908, -0.0007597 ,  0.00070556,\n",
              "            -0.00117019, -0.02785125, -0.00064779,  0.01464934, -0.00451334,\n",
              "            -0.00747119], dtype=float32),\n",
              "     'scale': array([-0.0027057 ,  0.00090396,  0.00317749,  0.00622377,  0.01451865,\n",
              "             0.00984382, -0.01228256,  0.01718505, -0.00625001,  0.00326924,\n",
              "             0.00148809, -0.01958428,  0.00911643,  0.00902811, -0.01890563,\n",
              "            -0.00911766], dtype=float32)},\n",
              "    'BatchNorm_1': {'bias': array([ 1.0234136e-03,  7.6868746e-05, -3.3956405e-03, -1.5561990e-03,\n",
              "            -1.2846587e-03, -9.4913441e-04,  3.7964791e-04, -6.5858127e-04,\n",
              "             1.8359930e-03,  2.1006891e-03, -9.8427339e-04,  3.3746858e-03,\n",
              "            -2.2691891e-03, -1.4523538e-03, -1.0055257e-03,  7.6265819e-04,\n",
              "             6.2770257e-04,  7.9935300e-05,  5.7409785e-04, -5.3198484e-04,\n",
              "             3.5397045e-03, -2.3542109e-04,  2.4526514e-04, -2.0634686e-03,\n",
              "            -4.1270757e-04, -1.2505902e-03,  4.1771224e-03, -4.5893551e-04,\n",
              "            -1.9916310e-04, -4.3075311e-04,  2.1294078e-03,  1.9148360e-03],\n",
              "           dtype=float32),\n",
              "     'scale': array([ 1.9946832e-03,  6.6299527e-04, -4.1994359e-03, -2.0279009e-03,\n",
              "            -2.0682735e-03, -8.0172432e-04, -3.6106270e-05, -2.0919568e-03,\n",
              "             6.7025336e-04,  1.4917095e-03, -1.1616402e-03,  4.7715516e-03,\n",
              "            -2.8496461e-03, -1.0655548e-03, -2.0052085e-03,  1.3321586e-03,\n",
              "            -2.4441164e-04, -4.8973365e-04,  5.2599801e-04, -3.1203878e-04,\n",
              "             3.1512436e-03,  5.5528362e-06,  2.5895861e-04, -2.6188961e-03,\n",
              "            -6.3351390e-04, -1.1359939e-03,  4.0924526e-03, -1.7035348e-04,\n",
              "            -5.7142670e-04, -1.4482257e-03,  2.2642897e-03,  1.9485857e-03],\n",
              "           dtype=float32)},\n",
              "    'Conv_0': {'bias': array([ 3.8337111e-10,  1.1119576e-09, -5.2016891e-10,  1.5100583e-09,\n",
              "             3.0273506e-10,  9.0963148e-10,  1.6262542e-09,  1.1559359e-09,\n",
              "            -2.3942959e-10,  2.5832497e-10, -7.5719075e-10,  3.8713934e-09,\n",
              "            -6.8728306e-10,  3.9595274e-09, -4.6106055e-10, -8.8214352e-10],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[-2.96848884e-04, -1.82235567e-03, -1.01171620e-03,\n",
              "               -3.50947259e-04,  6.89605763e-03,  9.26060043e-03,\n",
              "               -1.25300395e-03, -6.71855058e-04,  1.05820538e-03,\n",
              "                1.95391802e-03, -4.82726702e-03,  1.57979364e-03,\n",
              "                9.78429685e-04,  2.70817336e-03, -2.14182219e-04,\n",
              "               -2.38754367e-03]],\n",
              "     \n",
              "             [[ 3.36391124e-04, -1.20795565e-04,  1.23632024e-03,\n",
              "               -7.53012719e-04,  7.13743037e-03,  9.10865609e-03,\n",
              "               -1.43528532e-03, -1.04250107e-03,  1.47956726e-03,\n",
              "                3.37508973e-04, -3.01246904e-03,  1.44770613e-03,\n",
              "                1.81476411e-03,  2.39824108e-03, -9.19770624e-04,\n",
              "                3.16749210e-04]],\n",
              "     \n",
              "             [[-8.99199746e-04, -7.01596437e-04,  1.03088410e-03,\n",
              "               -1.33303553e-03,  6.97366288e-03,  8.24263040e-03,\n",
              "               -1.56531099e-03, -1.53717119e-03,  1.11995486e-03,\n",
              "               -7.93748477e-04, -8.88962299e-04,  1.20532548e-03,\n",
              "                2.23594811e-03,  3.28022637e-03, -1.06628682e-03,\n",
              "                3.11028445e-04]],\n",
              "     \n",
              "             [[-2.28290679e-03,  2.29788318e-04, -3.05839611e-04,\n",
              "               -2.39662593e-03,  6.51510479e-03,  7.76547100e-03,\n",
              "               -1.36503647e-03, -1.89448555e-03, -5.46038034e-04,\n",
              "               -7.17488991e-04,  2.45883432e-03,  6.70706140e-05,\n",
              "                3.30166565e-03,  2.36966577e-03, -8.04435986e-04,\n",
              "               -2.90883647e-04]],\n",
              "     \n",
              "             [[-2.31370004e-03,  4.40599257e-03,  2.12413725e-05,\n",
              "               -2.10534153e-03,  4.48246859e-03,  6.62992243e-03,\n",
              "               -1.11698965e-03, -2.46614311e-03, -1.60908396e-03,\n",
              "               -1.76038255e-03,  4.64955391e-03, -1.35204216e-04,\n",
              "                3.21591063e-03,  1.41347956e-03, -1.08023451e-05,\n",
              "               -5.53606427e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[ 2.10628728e-04,  1.86706893e-06,  9.42490587e-04,\n",
              "                1.81696098e-03,  4.53603733e-03,  4.22004936e-03,\n",
              "               -3.46384104e-03, -1.29151228e-03,  4.66487574e-04,\n",
              "                7.21966673e-04, -3.04496102e-03,  3.60009912e-03,\n",
              "                2.07994366e-03,  1.21391262e-04, -4.88317863e-04,\n",
              "               -1.31198857e-03]],\n",
              "     \n",
              "             [[ 6.86390325e-04,  2.85311649e-03,  1.76083448e-03,\n",
              "                1.00955949e-03,  5.07141557e-03,  5.34067117e-03,\n",
              "               -4.45502065e-03, -2.57594837e-03,  1.05525716e-03,\n",
              "               -1.29814795e-03, -2.52578803e-03,  3.58945015e-03,\n",
              "                2.33875471e-03,  4.56012785e-05, -6.15677680e-04,\n",
              "               -1.34221488e-03]],\n",
              "     \n",
              "             [[-4.06072591e-04,  2.35722051e-04,  3.92474059e-04,\n",
              "                9.88892396e-04,  4.40715067e-03,  4.82882233e-03,\n",
              "               -4.36806120e-03, -1.78444060e-03,  1.02243177e-03,\n",
              "               -9.63906641e-05, -2.92450981e-03,  4.14713658e-03,\n",
              "                2.46251561e-03, -2.48568947e-04, -1.02097297e-03,\n",
              "               -1.47316419e-03]],\n",
              "     \n",
              "             [[-1.56264647e-03,  1.02772028e-03, -1.13148312e-03,\n",
              "                4.48494626e-04,  3.36440955e-03,  6.05505379e-03,\n",
              "               -4.46543749e-03, -8.66374467e-04, -4.94409469e-04,\n",
              "               -4.56432317e-04, -3.75981326e-05,  1.83933391e-03,\n",
              "                3.86102009e-03,  2.06313387e-04, -6.88367581e-04,\n",
              "               -1.32422359e-03]],\n",
              "     \n",
              "             [[-1.14505668e-03,  2.93598324e-03,  3.60786536e-04,\n",
              "                4.43961384e-04,  2.67292815e-03,  4.35424224e-03,\n",
              "               -3.62533424e-03, -1.26196491e-03, -2.09251943e-04,\n",
              "                8.22215341e-04,  3.09079210e-03,  8.53063364e-04,\n",
              "                3.32497479e-03, -1.56928785e-04, -1.42349571e-03,\n",
              "               -1.14572444e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[-4.20978206e-04,  2.02089129e-03,  7.12846173e-04,\n",
              "                2.67367414e-03,  3.55608854e-03,  4.25342098e-03,\n",
              "               -3.69000179e-03, -2.17419118e-03, -5.26792544e-04,\n",
              "               -2.85987568e-04, -1.12841395e-03,  7.23175984e-03,\n",
              "                1.41481229e-03, -1.07917935e-03,  4.36288799e-04,\n",
              "               -1.21425302e-03]],\n",
              "     \n",
              "             [[ 5.36327658e-04,  2.35502189e-03,  2.58084293e-03,\n",
              "                1.81021704e-03,  3.48159065e-03,  3.42876278e-03,\n",
              "               -5.77986520e-03, -2.56331312e-03,  4.00672929e-04,\n",
              "               -1.76932453e-03, -2.24187691e-03,  7.38749234e-03,\n",
              "                1.66437100e-03, -2.30407284e-04,  1.93427259e-04,\n",
              "               -1.44386897e-04]],\n",
              "     \n",
              "             [[-2.42639973e-04,  1.86177215e-03, -7.84953532e-04,\n",
              "                4.01724741e-04,  3.37032671e-03,  3.77496984e-03,\n",
              "               -6.30620169e-03, -1.95908034e-03,  1.21230574e-03,\n",
              "               -1.34899013e-03, -1.25399732e-03,  6.50181482e-03,\n",
              "                2.44445936e-03, -1.40970806e-04, -7.67748919e-04,\n",
              "               -7.73097156e-04]],\n",
              "     \n",
              "             [[-7.28420098e-04,  3.56603065e-03, -5.70201897e-04,\n",
              "                1.65158679e-04,  2.73926603e-03,  4.32698382e-03,\n",
              "               -6.01168210e-03, -7.22099678e-04,  9.51970636e-04,\n",
              "                1.07393949e-04, -4.97556990e-04,  4.15379927e-03,\n",
              "                2.24152789e-03,  1.77513040e-03, -2.68754410e-03,\n",
              "               -8.94938421e-04]],\n",
              "     \n",
              "             [[-5.62098227e-04,  5.92336291e-03,  1.00490893e-03,\n",
              "               -3.00380372e-04,  2.38642888e-03,  4.18860465e-03,\n",
              "               -5.13678323e-03, -1.07836840e-03,  2.46456824e-03,\n",
              "                2.98419618e-04,  3.81559879e-03,  2.09104014e-03,\n",
              "                2.48142495e-03,  3.30497336e-04, -2.58909026e-03,\n",
              "               -1.17367087e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[ 1.45226149e-04,  3.24689946e-03,  1.11087307e-03,\n",
              "                1.49485760e-03,  1.69978209e-03,  3.44044808e-03,\n",
              "               -2.89280107e-03,  5.09201142e-04, -4.29836626e-04,\n",
              "               -1.88653986e-03, -9.74601251e-04,  6.07332634e-03,\n",
              "                4.03685000e-04, -6.61353231e-04,  2.08185811e-05,\n",
              "               -2.27894401e-03]],\n",
              "     \n",
              "             [[ 1.06983818e-03,  3.04943370e-03,  2.79848720e-03,\n",
              "                8.59202584e-04,  2.35487008e-03,  3.86350602e-03,\n",
              "               -5.02936728e-03,  3.19601008e-04,  6.37328310e-04,\n",
              "               -1.77071313e-03, -1.53440307e-03,  6.16503600e-03,\n",
              "                6.82582031e-04, -4.73065884e-04, -2.92282610e-04,\n",
              "               -7.85905868e-05]],\n",
              "     \n",
              "             [[ 4.38261748e-04,  2.99327215e-03,  7.88566365e-04,\n",
              "                2.47201242e-05,  2.32854951e-03,  2.39719916e-03,\n",
              "               -5.89687331e-03, -4.41182259e-04,  1.07513706e-03,\n",
              "               -1.03774783e-03, -3.09451483e-03,  5.32757118e-03,\n",
              "                4.40885255e-04,  4.18478565e-04, -1.16085913e-03,\n",
              "               -7.35967187e-04]],\n",
              "     \n",
              "             [[-2.88543990e-04,  4.60772216e-03,  7.86329270e-04,\n",
              "                5.00949682e-07,  2.24070810e-03,  2.11548805e-03,\n",
              "               -6.00133464e-03,  1.91354728e-03,  1.26067828e-03,\n",
              "                4.94028092e-04, -1.04709099e-04,  1.81203044e-03,\n",
              "                5.47574949e-04,  1.06105162e-03, -2.64088134e-03,\n",
              "               -8.99355276e-04]],\n",
              "     \n",
              "             [[ 1.02134887e-04,  5.85604832e-03,  7.10962107e-04,\n",
              "               -1.86746113e-03,  2.30373163e-03,  2.77752080e-03,\n",
              "               -6.35959813e-03, -7.10555818e-04,  4.07691300e-03,\n",
              "                2.61772890e-04,  1.47207710e-03,  1.92602514e-04,\n",
              "                6.99150318e-04, -1.41346082e-03, -2.80387164e-03,\n",
              "               -2.09122710e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[ 7.17369781e-04,  1.69137132e-03,  1.43994112e-03,\n",
              "               -3.23556451e-04,  1.76859507e-03,  2.51535466e-03,\n",
              "               -4.39248746e-03,  1.01209749e-04,  3.21246102e-04,\n",
              "               -2.17845896e-03,  1.19511597e-03,  3.99510935e-03,\n",
              "               -7.93229148e-04, -1.19540433e-03, -6.95473747e-04,\n",
              "               -6.18221704e-03]],\n",
              "     \n",
              "             [[ 1.09639228e-03,  1.91119651e-03,  1.90337119e-03,\n",
              "               -4.92034364e-04,  1.80853182e-03,  1.19770737e-03,\n",
              "               -7.11822510e-03,  6.35278120e-04,  6.02800807e-04,\n",
              "               -1.10981870e-03, -7.43359618e-04,  3.61470180e-03,\n",
              "               -8.66549322e-04, -1.90420006e-03, -6.86248459e-05,\n",
              "               -3.30044236e-03]],\n",
              "     \n",
              "             [[-2.29325451e-05,  1.12205755e-03,  2.44274526e-03,\n",
              "                3.76092212e-04,  1.80439674e-03,  2.09183912e-04,\n",
              "               -9.18548927e-03,  1.19185681e-03,  1.44636561e-03,\n",
              "               -6.60017715e-04, -2.02961219e-03,  3.54592572e-03,\n",
              "               -6.04091852e-04, -1.16006727e-03, -2.05886690e-03,\n",
              "               -8.15851614e-04]],\n",
              "     \n",
              "             [[-1.67922262e-04,  1.91549747e-03,  2.93884799e-03,\n",
              "               -9.07139096e-04,  2.26230570e-03, -1.75220557e-05,\n",
              "               -1.12855323e-02,  9.15110402e-04,  8.21084541e-04,\n",
              "               -8.69847281e-05,  1.01887842e-03,  9.50310612e-04,\n",
              "               -6.25683169e-04, -1.90275977e-03, -2.47131288e-03,\n",
              "                6.06529880e-04]],\n",
              "     \n",
              "             [[-4.14940936e-04,  4.16685967e-03,  3.17479204e-03,\n",
              "               -7.07376399e-04,  1.46284839e-03,  6.33068150e-04,\n",
              "               -1.04531404e-02, -2.42429087e-04,  2.71723932e-03,\n",
              "                6.24050008e-05,  7.09235319e-04, -4.33081470e-04,\n",
              "               -2.21650727e-04, -3.59851122e-03, -4.02120082e-03,\n",
              "               -2.95640435e-04]]]], dtype=float32)},\n",
              "    'Conv_1': {'bias': array([ 1.6833404e-10, -6.2878071e-11, -8.9434604e-10, -4.2032230e-10,\n",
              "            -3.3347308e-10,  2.1256097e-10,  1.7294936e-10, -9.6940650e-11,\n",
              "             8.9777248e-11,  3.0720310e-10, -4.1880666e-12, -3.8123962e-10,\n",
              "             2.6087233e-11, -8.9846908e-11,  3.2923600e-10, -2.6533684e-10,\n",
              "             7.7515730e-11, -2.8084637e-11, -9.8188346e-11,  2.0060734e-10,\n",
              "            -6.0148671e-11,  1.2979215e-10, -1.9278119e-10, -8.1351662e-12,\n",
              "            -1.1877860e-12, -9.7488413e-11, -2.5172619e-11, -1.0348088e-09,\n",
              "            -2.6827940e-10,  1.5170598e-10, -4.2758330e-10,  3.8968906e-10],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[ 2.65180046e-04, -1.58748808e-04, -7.65517398e-05, ...,\n",
              "               -1.38355404e-04,  2.18768313e-04, -3.47222194e-05],\n",
              "              [-1.63499237e-04, -2.88425828e-04, -3.79343517e-04, ...,\n",
              "                7.84159056e-05,  7.23809862e-05, -2.29914629e-04],\n",
              "              [ 2.04533790e-04,  7.95665197e-04, -3.68506699e-05, ...,\n",
              "                1.12390017e-05,  6.98538497e-04,  4.05922852e-04],\n",
              "              ...,\n",
              "              [ 1.21325269e-04,  9.98987467e-04, -3.22674983e-04, ...,\n",
              "                7.10227541e-05, -3.66870430e-04,  2.42481983e-04],\n",
              "              [-1.50810316e-04,  1.36578936e-04, -1.13024536e-04, ...,\n",
              "               -3.68598521e-05,  2.71208235e-04, -6.99009615e-05],\n",
              "              [-2.46579493e-05,  4.70872212e-04, -5.49526558e-05, ...,\n",
              "               -2.57760803e-05,  2.27394994e-04,  5.60447574e-04]],\n",
              "     \n",
              "             [[ 9.99519980e-05,  8.78508072e-05,  5.40248448e-06, ...,\n",
              "               -1.44907375e-04,  6.02625252e-04,  1.46625433e-04],\n",
              "              [-2.29845755e-04,  7.47657541e-05, -3.70538008e-04, ...,\n",
              "                1.92358522e-04,  2.45147123e-04,  2.53945211e-04],\n",
              "              [ 1.26309547e-04,  1.34414295e-04, -9.21049214e-05, ...,\n",
              "                1.69229079e-05,  5.49524484e-05, -1.61320233e-04],\n",
              "              ...,\n",
              "              [ 1.16136791e-04,  8.56697443e-04, -8.97426726e-05, ...,\n",
              "               -3.16342703e-05, -3.04014917e-04, -3.34955577e-04],\n",
              "              [-1.53030705e-04, -1.80924762e-04, -5.32274717e-04, ...,\n",
              "               -3.81387290e-05,  2.47882854e-04,  5.65848677e-05],\n",
              "              [ 5.15010033e-04,  9.31412287e-05,  7.06920400e-06, ...,\n",
              "               -2.27483994e-04,  4.38830059e-04, -1.11040936e-05]],\n",
              "     \n",
              "             [[ 6.43997737e-06,  2.02209150e-04,  1.01977261e-04, ...,\n",
              "               -1.50092601e-04,  4.17725038e-04,  1.58478826e-04],\n",
              "              [-7.60889452e-05, -6.69687870e-05,  2.48451455e-04, ...,\n",
              "                7.24714846e-05, -1.99989125e-04, -1.24066617e-04],\n",
              "              [ 2.05952805e-04, -3.20764142e-04, -3.10781325e-04, ...,\n",
              "               -1.18719821e-04, -1.88086269e-04,  3.81325692e-04],\n",
              "              ...,\n",
              "              [ 2.03306612e-04,  9.75052943e-04, -6.67968008e-04, ...,\n",
              "                1.32132584e-04, -2.33074068e-04, -8.88053910e-06],\n",
              "              [-2.58253654e-04, -3.76267999e-04, -5.44821844e-04, ...,\n",
              "               -9.16923018e-05,  2.12021376e-04, -1.09969929e-04],\n",
              "              [ 7.08086664e-05, -2.96231701e-05, -6.47709676e-05, ...,\n",
              "               -6.03940716e-05, -1.11382426e-04,  3.76316137e-04]],\n",
              "     \n",
              "             [[-1.80052266e-05,  3.11816584e-05, -3.84119252e-04, ...,\n",
              "                1.01265206e-04,  3.89911846e-04,  6.43840758e-06],\n",
              "              [ 2.41836402e-04,  3.07005306e-04, -4.58410359e-04, ...,\n",
              "                2.57401058e-04, -5.91273245e-04,  1.38837131e-04],\n",
              "              [-2.16405620e-04, -5.81283704e-04, -5.52660669e-04, ...,\n",
              "               -1.54683599e-04,  1.96801135e-04,  4.85336321e-04],\n",
              "              ...,\n",
              "              [ 1.14720962e-04,  2.83418631e-04, -2.81256624e-04, ...,\n",
              "               -3.45654225e-05, -1.48446998e-04, -1.31723718e-04],\n",
              "              [-2.24870135e-04, -6.80025405e-05,  1.52861219e-04, ...,\n",
              "               -1.13608541e-04,  1.63753575e-04, -2.36240405e-04],\n",
              "              [ 3.35107150e-04, -4.93427797e-04, -4.00470250e-04, ...,\n",
              "               -1.50999986e-05,  5.76562597e-05, -6.27599948e-05]],\n",
              "     \n",
              "             [[ 7.13637637e-05,  1.17138508e-04, -4.58287715e-04, ...,\n",
              "               -2.99455187e-05,  1.39277385e-04,  1.15127536e-04],\n",
              "              [-1.10418856e-04,  2.72324280e-04, -9.03263572e-05, ...,\n",
              "                2.28339952e-04, -2.68911215e-04, -2.44501163e-04],\n",
              "              [ 1.54584501e-04,  5.04062104e-04,  1.91984989e-04, ...,\n",
              "               -9.36427750e-05, -7.90263875e-05,  1.19567121e-05],\n",
              "              ...,\n",
              "              [-8.65850889e-05, -9.08394650e-05,  3.50212358e-05, ...,\n",
              "               -4.03087688e-06, -3.43029169e-05,  2.64024740e-04],\n",
              "              [-1.08902903e-04,  1.05102343e-04,  1.47360843e-04, ...,\n",
              "               -8.58535350e-05, -8.51812074e-05, -1.37923838e-04],\n",
              "              [ 3.38473066e-04,  2.10120925e-06, -3.25624918e-04, ...,\n",
              "               -3.76124663e-05, -2.47129472e-04,  4.52382315e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[-5.18782326e-05,  5.04504715e-06, -1.63804114e-04, ...,\n",
              "               -7.93366198e-05,  1.54454210e-05,  4.58485796e-04],\n",
              "              [ 6.35700198e-05,  3.69854213e-04, -1.03871935e-04, ...,\n",
              "                1.12186593e-04,  1.02979050e-03, -1.11079797e-04],\n",
              "              [-1.89305531e-04,  4.30047403e-05, -3.19992942e-05, ...,\n",
              "                1.29746215e-04, -4.24646896e-05,  3.26832727e-04],\n",
              "              ...,\n",
              "              [ 1.53593923e-04,  2.40229390e-04,  1.03397742e-05, ...,\n",
              "               -6.00188541e-05, -3.06362199e-04, -2.90820492e-04],\n",
              "              [ 3.30803712e-04,  1.40129152e-04, -6.39469712e-04, ...,\n",
              "               -1.36608389e-04,  3.15662124e-04,  2.54645041e-04],\n",
              "              [-9.71967529e-05,  1.55744303e-04,  9.81413614e-05, ...,\n",
              "               -2.02167157e-05, -5.56196574e-06,  2.55753519e-04]],\n",
              "     \n",
              "             [[-8.48815398e-05, -1.60873649e-04, -2.89056712e-04, ...,\n",
              "               -9.00360537e-05,  4.11963352e-04,  3.28942202e-04],\n",
              "              [ 1.15508570e-04,  6.95192284e-05,  2.42685255e-05, ...,\n",
              "               -3.71903552e-05, -2.10544487e-04, -5.25360010e-05],\n",
              "              [ 4.34820686e-04,  4.15774120e-05,  1.16700343e-04, ...,\n",
              "               -1.62397337e-04,  4.68295366e-05, -5.23099734e-05],\n",
              "              ...,\n",
              "              [-7.11073953e-05,  3.23338842e-04,  7.51097687e-05, ...,\n",
              "               -6.06883223e-05, -1.35862836e-04, -1.80344796e-04],\n",
              "              [ 6.24761567e-04,  2.96585669e-04, -4.51960310e-04, ...,\n",
              "                2.56789383e-04,  3.43549822e-04,  4.38903604e-04],\n",
              "              [-1.86446181e-04,  1.94605265e-04,  2.65515642e-04, ...,\n",
              "                4.15844639e-04,  4.96856053e-04,  2.78200459e-05]],\n",
              "     \n",
              "             [[-1.13287475e-04, -2.03540621e-05, -1.21618767e-04, ...,\n",
              "                9.87497333e-05, -9.73453643e-06, -9.10045856e-07],\n",
              "              [ 3.69070505e-04, -1.84672797e-04, -4.48773702e-04, ...,\n",
              "                6.30201976e-05,  7.05638449e-05,  6.65805070e-04],\n",
              "              [ 3.61572427e-04, -1.12230213e-04, -2.82090972e-04, ...,\n",
              "                2.40201189e-05, -1.20400087e-04, -9.66462685e-05],\n",
              "              ...,\n",
              "              [-2.95854989e-04,  4.82371222e-04, -1.52617664e-04, ...,\n",
              "               -1.71662068e-05, -9.57882439e-05, -3.19776911e-04],\n",
              "              [ 7.33862515e-04,  3.36096506e-04, -2.73012440e-04, ...,\n",
              "                4.34905465e-04,  1.52350738e-04,  5.43636677e-04],\n",
              "              [ 5.32674836e-04, -3.05522466e-04,  2.28793710e-04, ...,\n",
              "               -1.70645973e-04, -4.74395019e-05, -2.11634528e-04]],\n",
              "     \n",
              "             [[-3.39200429e-04, -1.81415220e-04, -2.87208997e-04, ...,\n",
              "                1.01514088e-04,  4.52427448e-05,  4.44584148e-05],\n",
              "              [-2.10817394e-04, -2.11603488e-04, -7.15095084e-04, ...,\n",
              "               -9.57893571e-05,  5.56620507e-05, -6.32646406e-05],\n",
              "              [ 3.16935562e-04,  3.86969245e-04,  2.97371647e-04, ...,\n",
              "               -1.70360639e-04, -1.43950107e-04,  2.30268764e-04],\n",
              "              ...,\n",
              "              [-1.44799589e-04,  3.18534760e-04,  1.67194317e-04, ...,\n",
              "               -7.94512162e-05, -8.83119901e-06, -1.95113127e-04],\n",
              "              [ 5.78768493e-04,  9.79002361e-05, -1.98232941e-04, ...,\n",
              "                3.02420463e-04,  1.56482929e-05,  4.15915449e-04],\n",
              "              [ 2.14205618e-04, -4.89884987e-04, -1.27113599e-04, ...,\n",
              "                1.00990495e-04,  7.13038462e-05, -2.43296992e-04]],\n",
              "     \n",
              "             [[-8.83791072e-05,  2.01433868e-04, -6.83841179e-04, ...,\n",
              "               -8.72450910e-05, -8.60171858e-05,  3.22297565e-04],\n",
              "              [ 4.28768108e-04,  2.04850614e-04, -2.26252232e-04, ...,\n",
              "                2.29568905e-05, -3.76320895e-05,  8.48763302e-05],\n",
              "              [ 4.90282080e-04, -7.49875384e-04, -1.59536023e-04, ...,\n",
              "                9.98784672e-05,  2.28838311e-04,  4.55915637e-04],\n",
              "              ...,\n",
              "              [ 2.85972856e-05, -6.99395532e-05,  1.58955008e-04, ...,\n",
              "               -1.06599677e-04,  7.70006300e-05, -1.25886174e-04],\n",
              "              [-4.51036613e-06,  2.52237951e-04, -2.66127754e-04, ...,\n",
              "                8.19615670e-05, -2.15669366e-04,  2.82778667e-04],\n",
              "              [-6.13977172e-05,  1.48849271e-04, -5.10077807e-04, ...,\n",
              "               -4.36788483e-04,  4.50217689e-04,  1.01831730e-03]]],\n",
              "     \n",
              "     \n",
              "            [[[ 4.08642576e-04, -7.71950727e-05,  1.22419835e-04, ...,\n",
              "                2.43673232e-04,  5.13271603e-04,  2.25442956e-04],\n",
              "              [ 9.45465436e-05,  1.24460566e-04, -1.22099853e-04, ...,\n",
              "               -2.31189870e-06,  3.96248011e-04,  3.84611340e-04],\n",
              "              [-7.20905882e-05,  3.90149740e-04, -1.21951434e-05, ...,\n",
              "               -2.07648583e-04,  3.22536012e-04,  8.07479410e-07],\n",
              "              ...,\n",
              "              [ 4.70696395e-05,  3.95116396e-04, -4.46754799e-04, ...,\n",
              "               -4.26487415e-04, -1.37604366e-06, -1.12302012e-04],\n",
              "              [ 1.48897394e-04,  1.52900902e-05, -3.29992326e-04, ...,\n",
              "                1.61859876e-04,  6.63097366e-04, -5.91918724e-05],\n",
              "              [ 2.12001069e-05,  4.92469349e-04,  6.03380395e-05, ...,\n",
              "               -4.87709010e-04,  7.36611837e-05, -5.19848472e-05]],\n",
              "     \n",
              "             [[ 2.85595830e-04, -1.54454159e-04, -1.94159846e-04, ...,\n",
              "                2.42987910e-04,  2.96054553e-04,  4.11949877e-04],\n",
              "              [ 7.15494854e-04, -2.61184934e-04, -1.00422272e-04, ...,\n",
              "               -1.43988043e-04,  4.88353427e-04, -4.44041158e-04],\n",
              "              [-2.32302351e-04, -4.36013434e-05,  2.48924607e-05, ...,\n",
              "                1.35396214e-04, -6.66283595e-05,  4.00961988e-04],\n",
              "              ...,\n",
              "              [ 2.00607959e-04,  5.87661285e-04, -2.29466314e-04, ...,\n",
              "               -7.03109545e-05, -1.02218350e-04, -1.10729365e-04],\n",
              "              [ 5.72392455e-05, -1.44624846e-05, -2.94497178e-04, ...,\n",
              "                9.87954118e-05,  5.27000986e-04, -7.69374237e-05],\n",
              "              [-1.17887830e-05, -1.14024060e-05,  6.01678767e-05, ...,\n",
              "                1.23160600e-04, -5.95156162e-05,  1.49196509e-04]],\n",
              "     \n",
              "             [[ 2.26597811e-04, -2.31745391e-04, -1.72392509e-04, ...,\n",
              "                3.45644017e-04,  2.80312292e-04, -3.48110072e-04],\n",
              "              [-2.80001696e-04,  1.53586021e-04, -2.63112481e-04, ...,\n",
              "               -1.02565144e-04,  4.95593049e-05,  6.88726723e-04],\n",
              "              [ 2.42191425e-04,  1.93237051e-04, -3.99693963e-05, ...,\n",
              "               -1.39817603e-05,  1.43253346e-04, -4.64644909e-05],\n",
              "              ...,\n",
              "              [ 1.59307965e-04,  4.05334140e-04, -9.23877014e-05, ...,\n",
              "               -3.69149726e-04, -9.81054691e-05,  7.90680861e-05],\n",
              "              [-1.58165873e-04,  6.72968163e-05, -2.20969465e-04, ...,\n",
              "               -3.25302717e-05,  7.06752879e-04, -5.48198150e-05],\n",
              "              [ 3.57915706e-04, -2.22995805e-04,  3.10060947e-04, ...,\n",
              "               -2.85272894e-04, -5.60154876e-05, -2.93513469e-04]],\n",
              "     \n",
              "             [[ 6.20724750e-05,  2.46390591e-05, -3.76819407e-05, ...,\n",
              "               -1.25866107e-04, -2.17374050e-04,  1.71410109e-04],\n",
              "              [ 7.95459258e-04, -1.99404763e-04,  1.18164258e-04, ...,\n",
              "               -2.30268255e-04,  1.04949824e-04,  4.62523487e-04],\n",
              "              [ 5.06938610e-04, -2.33641680e-04,  5.14042622e-05, ...,\n",
              "                2.01389121e-05,  1.97199712e-04,  6.01934553e-05],\n",
              "              ...,\n",
              "              [-6.32087322e-05,  1.65054749e-04, -8.54613827e-05, ...,\n",
              "               -1.98512193e-04,  1.10675108e-04, -2.66220450e-05],\n",
              "              [-3.38908198e-04,  2.96076149e-04, -3.63144733e-04, ...,\n",
              "               -1.46827515e-04,  6.01085019e-04,  2.10582366e-05],\n",
              "              [ 5.17729670e-04, -1.95933098e-04,  2.76757055e-06, ...,\n",
              "               -2.91475269e-04, -1.29073829e-04, -2.92928657e-04]],\n",
              "     \n",
              "             [[ 1.18547876e-04, -4.82274882e-05, -3.75596399e-04, ...,\n",
              "               -2.05693621e-04, -1.74615940e-04,  3.96647200e-04],\n",
              "              [ 1.12499278e-04, -4.24244790e-05, -5.57090680e-05, ...,\n",
              "               -8.15243911e-05, -8.45086615e-05,  2.36208914e-04],\n",
              "              [-4.33543813e-04,  6.59635814e-04,  2.54847895e-04, ...,\n",
              "               -6.74877592e-05,  1.45080674e-04,  7.33515044e-05],\n",
              "              ...,\n",
              "              [ 2.71231402e-05,  2.47074029e-04,  5.34318315e-05, ...,\n",
              "               -1.60735726e-04, -6.34449534e-05, -1.09377535e-04],\n",
              "              [ 1.83025084e-04,  1.82753021e-04, -2.27552635e-04, ...,\n",
              "                4.09729982e-05,  6.24609413e-04,  1.89607526e-04],\n",
              "              [-8.99556617e-05,  4.23338817e-04,  4.28921456e-04, ...,\n",
              "               -4.08097869e-04,  1.15954055e-04,  5.30838093e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[ 1.58783572e-04,  1.29858206e-04,  4.61305171e-05, ...,\n",
              "                1.69565828e-04,  1.23154881e-04,  1.11788628e-04],\n",
              "              [ 2.25827825e-04,  3.47984838e-04, -4.96189226e-04, ...,\n",
              "               -2.52993806e-04,  4.13090194e-04,  5.20556467e-04],\n",
              "              [ 4.43298486e-04,  2.68617907e-04,  1.23226346e-04, ...,\n",
              "               -1.08556480e-04,  3.19560786e-05, -9.02212632e-05],\n",
              "              ...,\n",
              "              [ 2.91490054e-04, -5.73075667e-04, -5.30100369e-04, ...,\n",
              "                1.92515290e-04, -1.71936597e-04, -4.95878921e-05],\n",
              "              [ 3.16339720e-04,  1.87182843e-04, -3.36626108e-04, ...,\n",
              "               -4.06117470e-05, -1.94361171e-04, -8.80751613e-05],\n",
              "              [-9.27299043e-05,  9.75651637e-05,  4.70470404e-05, ...,\n",
              "               -5.85299276e-05, -7.14938069e-05,  1.46507346e-05]],\n",
              "     \n",
              "             [[-9.07502545e-06,  9.37773002e-05, -5.51674457e-04, ...,\n",
              "                3.57969606e-04,  6.55593351e-04,  2.28378995e-04],\n",
              "              [-4.04851926e-05, -2.28632576e-04, -2.52940372e-05, ...,\n",
              "               -1.09322595e-04,  4.59987641e-05,  2.21229238e-05],\n",
              "              [ 1.69889914e-04,  1.56360366e-05, -7.82044226e-05, ...,\n",
              "               -1.17098760e-04, -5.60033717e-04,  4.05705578e-05],\n",
              "              ...,\n",
              "              [ 1.73520806e-04, -2.15005173e-04,  6.02458749e-06, ...,\n",
              "               -4.45166952e-05, -1.32497116e-05, -2.98469269e-04],\n",
              "              [ 3.20475374e-04,  2.30372796e-04, -2.58548534e-04, ...,\n",
              "                2.08676211e-05, -1.63055622e-04,  5.50163059e-05],\n",
              "              [-5.33191196e-05, -1.52751367e-04, -2.48567244e-06, ...,\n",
              "                9.54969801e-05,  8.56761108e-05, -4.30031432e-05]],\n",
              "     \n",
              "             [[ 1.28648448e-04, -7.79822512e-05, -3.08168674e-04, ...,\n",
              "               -1.11130503e-05,  2.08851212e-04, -1.12577975e-04],\n",
              "              [ 8.82086693e-04, -5.57523570e-04, -1.34431015e-04, ...,\n",
              "               -1.55372501e-04,  6.05049543e-04, -1.02429374e-04],\n",
              "              [ 8.66940478e-04,  1.36165909e-04,  2.27040109e-05, ...,\n",
              "                2.63500260e-04, -5.85156959e-05,  1.60158961e-05],\n",
              "              ...,\n",
              "              [-4.61916497e-04, -8.43941380e-05, -3.01542386e-05, ...,\n",
              "               -2.65487852e-05,  2.69298180e-05,  2.53203267e-04],\n",
              "              [ 1.39061565e-04,  2.38228095e-05, -6.43177889e-04, ...,\n",
              "                1.43609956e-04,  1.44081248e-04,  9.49710957e-05],\n",
              "              [ 3.75706557e-04, -4.06305335e-04,  6.74381081e-05, ...,\n",
              "               -2.88049894e-04,  3.67025059e-05, -1.02114805e-04]],\n",
              "     \n",
              "             [[-6.37027406e-05, -3.74368043e-04, -6.16456207e-04, ...,\n",
              "               -2.11878767e-04, -2.93404246e-05,  3.68092587e-04],\n",
              "              [-5.27194527e-04, -4.45285725e-04,  6.31441799e-05, ...,\n",
              "               -1.62296463e-04, -6.72743918e-05,  3.03120032e-04],\n",
              "              [ 1.89721453e-04,  1.09776083e-04,  7.70942934e-05, ...,\n",
              "               -3.96040909e-04, -3.11696385e-05,  4.23147540e-05],\n",
              "              ...,\n",
              "              [-2.71365978e-04,  3.49016103e-04, -4.25016515e-05, ...,\n",
              "                7.10157037e-05,  5.23951196e-04,  3.59463331e-04],\n",
              "              [ 1.09962399e-04, -2.13607433e-04, -3.84937273e-04, ...,\n",
              "                2.23276555e-04,  1.57104834e-04,  1.39997501e-05],\n",
              "              [ 5.54146318e-05,  1.39950440e-04,  1.70257554e-04, ...,\n",
              "               -5.24368079e-04, -2.09902777e-04, -3.42912361e-04]],\n",
              "     \n",
              "             [[-1.04663719e-04, -3.21642554e-04, -5.49714896e-04, ...,\n",
              "               -8.35694373e-05,  2.53943494e-04,  3.97505093e-04],\n",
              "              [ 1.86109508e-04,  5.03748714e-04,  1.19068813e-04, ...,\n",
              "               -6.34352255e-05,  5.47025120e-04,  3.51667753e-04],\n",
              "              [ 5.76688675e-04,  3.08276532e-04, -1.81762603e-04, ...,\n",
              "                1.27267442e-04,  1.90171282e-04,  1.34892798e-06],\n",
              "              ...,\n",
              "              [ 2.93688936e-04,  1.27402498e-04, -9.49770110e-05, ...,\n",
              "                1.24656493e-04,  2.56061088e-04,  1.78371876e-04],\n",
              "              [-3.26998328e-04, -2.97222665e-04, -2.80270644e-04, ...,\n",
              "                2.76813400e-04,  6.92257017e-05, -7.04839040e-05],\n",
              "              [ 3.23777815e-04,  3.48883332e-04,  1.64421552e-04, ...,\n",
              "               -5.34887542e-04,  5.56760991e-04,  2.69383658e-04]]],\n",
              "     \n",
              "     \n",
              "            [[[ 3.35284349e-05,  5.83102974e-07, -4.86237637e-04, ...,\n",
              "               -8.79198633e-05,  3.07771435e-04,  2.16757166e-04],\n",
              "              [ 5.43992093e-04,  6.24357563e-06, -2.37118220e-04, ...,\n",
              "               -2.54089944e-04,  3.80044687e-04,  4.37676063e-04],\n",
              "              [-2.15629960e-04,  4.30949614e-04, -3.21664229e-05, ...,\n",
              "               -5.46277224e-05,  1.67727903e-05,  9.13465847e-05],\n",
              "              ...,\n",
              "              [ 4.81250026e-06,  5.95011457e-04, -3.63009676e-05, ...,\n",
              "                3.94741539e-04, -3.38662328e-04,  2.62201356e-04],\n",
              "              [ 6.86822968e-05,  4.06465879e-05, -3.05528054e-04, ...,\n",
              "               -3.12692064e-05,  6.95644412e-04,  3.03944762e-05],\n",
              "              [-7.83935757e-05,  2.14583459e-04,  3.00274114e-05, ...,\n",
              "               -2.57378651e-05, -3.63482250e-05,  3.94498711e-05]],\n",
              "     \n",
              "             [[ 1.42611476e-04, -3.32841330e-04, -3.83944425e-04, ...,\n",
              "               -3.70847643e-04,  4.33624286e-04,  3.12048855e-04],\n",
              "              [ 2.85010727e-04,  2.19734320e-05, -3.47685622e-04, ...,\n",
              "                1.53126224e-04, -1.37208510e-04, -1.20888544e-04],\n",
              "              [ 4.08394408e-04,  2.94701858e-05, -1.50876986e-06, ...,\n",
              "                1.46594641e-04,  1.34236452e-05, -5.25834512e-05],\n",
              "              ...,\n",
              "              [-8.99713268e-05,  4.80330782e-04,  6.49777940e-05, ...,\n",
              "               -6.88147993e-05, -4.21753066e-04,  6.30933209e-06],\n",
              "              [ 2.10485625e-04,  2.21073729e-04, -1.95356421e-04, ...,\n",
              "                1.67904655e-05,  5.53763879e-04,  1.37377472e-04],\n",
              "              [ 7.12296169e-05, -3.36095487e-04,  4.05699247e-05, ...,\n",
              "               -5.68484247e-05,  7.22257537e-05, -5.19562018e-05]],\n",
              "     \n",
              "             [[-1.81407668e-05, -2.06030993e-04, -6.93643116e-04, ...,\n",
              "               -2.55800900e-04, -1.69752253e-04,  2.05778837e-04],\n",
              "              [ 4.93684493e-04,  2.91508914e-04, -1.89470142e-04, ...,\n",
              "                1.65192789e-04, -1.84516059e-04,  6.65977539e-04],\n",
              "              [-9.87883686e-06, -7.31149385e-06, -5.24177267e-05, ...,\n",
              "               -2.09369973e-04,  8.04626761e-05,  1.07269778e-04],\n",
              "              ...,\n",
              "              [-2.17980836e-04,  2.81202549e-04, -2.53918377e-04, ...,\n",
              "               -4.59800722e-05, -3.12798074e-04,  3.04890767e-04],\n",
              "              [ 2.82170979e-04,  7.10579858e-04, -1.24606056e-04, ...,\n",
              "               -2.18124769e-04,  5.73777128e-04,  2.30041333e-05],\n",
              "              [ 1.06333711e-04, -8.28604316e-05, -2.83478090e-04, ...,\n",
              "               -2.34608873e-04, -1.83090800e-04, -9.19354352e-06]],\n",
              "     \n",
              "             [[-1.84245204e-04, -1.70761981e-04, -4.49031359e-04, ...,\n",
              "               -2.42324080e-04, -2.37617405e-05,  1.61793840e-04],\n",
              "              [ 1.01771287e-03,  1.18201016e-04, -1.52463093e-04, ...,\n",
              "               -1.27067935e-04, -3.04890593e-04, -8.27634358e-05],\n",
              "              [ 4.19860240e-04,  1.00601741e-04, -6.46690660e-06, ...,\n",
              "               -1.01076497e-04, -8.68151692e-05, -1.85349432e-04],\n",
              "              ...,\n",
              "              [-1.86259116e-04,  2.73823971e-05, -1.74067784e-04, ...,\n",
              "                1.89333041e-05, -1.14873757e-04, -2.26058037e-04],\n",
              "              [ 3.10033094e-04,  6.03427354e-04, -3.01974098e-04, ...,\n",
              "               -7.84345088e-04,  3.63066385e-04,  1.14594208e-04],\n",
              "              [ 6.21634186e-04,  5.11347898e-05,  3.02486675e-04, ...,\n",
              "                2.48995202e-04, -1.39704352e-04, -3.19267623e-04]],\n",
              "     \n",
              "             [[-2.28739591e-05,  3.32264288e-04, -2.95899023e-04, ...,\n",
              "               -1.74266781e-04,  2.60023575e-04,  2.43017625e-04],\n",
              "              [-1.73280569e-04,  4.96249122e-04, -2.37474422e-04, ...,\n",
              "               -6.47584748e-05, -2.19667563e-04,  3.77228662e-06],\n",
              "              [-1.57551578e-04,  8.22974325e-05, -6.49462600e-05, ...,\n",
              "               -1.39177413e-04,  2.35752101e-04, -4.14299575e-05],\n",
              "              ...,\n",
              "              [-3.32510128e-04, -4.06030995e-05, -1.42801538e-04, ...,\n",
              "                8.27382828e-05,  6.21649888e-05, -1.90370745e-04],\n",
              "              [ 3.13787168e-04,  5.26859425e-04, -3.58685327e-04, ...,\n",
              "               -1.00001786e-03,  4.89010883e-04,  9.49606037e-05],\n",
              "              [ 5.14475454e-04, -1.49429892e-04,  3.94167437e-04, ...,\n",
              "               -1.84563178e-05, -2.59016961e-04, -1.80607356e-04]]]],\n",
              "           dtype=float32)},\n",
              "    'Dense_0': {'bias': array([-2.4973224e-03,  6.7123416e-04,  2.6150723e-03, -2.0085187e-03,\n",
              "            -1.1565093e-03,  3.6970475e-03, -7.1615788e-05, -6.9848995e-04,\n",
              "            -1.2473250e-03,  6.9642608e-04], dtype=float32),\n",
              "     'kernel': array([[-1.48731610e-03, -2.77985018e-05,  4.16548783e-03,\n",
              "             -7.68440659e-04, -3.98487377e-04,  1.91142107e-03,\n",
              "             -2.69848853e-04, -1.00215699e-03, -2.21613655e-03,\n",
              "              9.32750991e-05],\n",
              "            [-1.62777491e-03,  7.54272332e-04,  2.03767628e-03,\n",
              "             -1.47845177e-03, -1.16145436e-03,  2.64588953e-03,\n",
              "              3.71136237e-04, -2.29287194e-03,  1.31406006e-04,\n",
              "              6.20172359e-04],\n",
              "            [-6.86337124e-04, -4.13556438e-04,  4.40467149e-03,\n",
              "              1.33405416e-03, -1.33484066e-03,  9.57296463e-04,\n",
              "              1.08275701e-04, -1.15025532e-03, -2.85982527e-03,\n",
              "             -3.59483762e-04],\n",
              "            [-2.47209158e-04,  3.67065251e-04,  3.26095941e-03,\n",
              "             -1.22702273e-03, -6.86115294e-04,  3.68906581e-03,\n",
              "             -3.25533852e-04, -3.42037971e-03, -1.74347265e-03,\n",
              "              3.32639436e-04],\n",
              "            [ 2.99240619e-05, -5.00901253e-04,  3.36248707e-03,\n",
              "             -2.52732029e-03, -1.06302358e-03,  2.20118277e-03,\n",
              "              9.35666249e-05, -2.04662560e-03,  7.80069167e-05,\n",
              "              3.72703042e-04],\n",
              "            [-1.79705117e-03,  3.66690540e-04,  2.73697195e-04,\n",
              "             -2.62281028e-05, -4.90776147e-04,  2.02716421e-03,\n",
              "             -8.56634811e-04,  1.42198650e-03, -1.40249985e-03,\n",
              "              4.83649143e-04],\n",
              "            [-1.40690128e-03, -5.80933229e-05,  5.26450993e-03,\n",
              "             -1.15257339e-03,  8.56866973e-05,  3.01267323e-03,\n",
              "              8.86828857e-05, -3.65723460e-03, -2.33395281e-03,\n",
              "              1.57201255e-04],\n",
              "            [-9.58415098e-04,  4.29068838e-04,  2.22653197e-03,\n",
              "             -2.71549099e-03, -7.23558187e-04,  3.13881878e-03,\n",
              "             -1.12293975e-03, -3.23026150e-04, -9.33050935e-04,\n",
              "              9.82059049e-04],\n",
              "            [-1.78874715e-03,  5.88588591e-04,  3.83691909e-03,\n",
              "             -1.22039369e-03, -6.99158059e-04,  2.51330435e-03,\n",
              "             -8.44190770e-04, -9.85682200e-05, -2.79339217e-03,\n",
              "              5.05635981e-04],\n",
              "            [-1.49224629e-03,  5.69712429e-04,  1.50130573e-03,\n",
              "             -2.16791173e-03, -5.66096103e-04,  3.83969722e-03,\n",
              "             -5.15992113e-04, -1.47135041e-04, -4.34558810e-04,\n",
              "             -5.86775481e-04],\n",
              "            [-2.22089561e-03,  2.61654292e-04,  1.71519152e-03,\n",
              "              5.99833787e-04, -5.46647585e-04,  1.44127384e-03,\n",
              "              1.41463635e-04, -1.72063359e-03, -3.92447662e-04,\n",
              "              7.21208286e-04],\n",
              "            [-9.85085964e-04,  3.85559892e-04,  1.97384786e-03,\n",
              "             -2.12855008e-03, -5.65390976e-04,  2.01515248e-03,\n",
              "             -1.46155362e-04, -1.85025303e-04, -4.23405669e-04,\n",
              "              5.90516429e-05],\n",
              "            [-2.44088448e-03, -3.09987023e-04,  1.23459799e-03,\n",
              "              3.75277188e-04, -6.46382105e-04,  1.32017594e-03,\n",
              "              3.24573950e-04,  5.80878987e-06, -1.19220116e-04,\n",
              "              2.56039231e-04],\n",
              "            [-1.54703110e-03,  2.56817701e-04,  9.11945361e-04,\n",
              "             -2.13196807e-04, -2.58358719e-04,  1.11243920e-03,\n",
              "             -1.40286924e-04,  5.09927340e-04, -6.78396143e-04,\n",
              "              4.61389136e-05],\n",
              "            [-7.68614002e-04, -8.61590088e-05,  1.26642885e-03,\n",
              "             -1.79284706e-03, -6.56416814e-04,  2.69785197e-03,\n",
              "              1.72224740e-04,  3.46675050e-04, -1.23411464e-03,\n",
              "              5.49707911e-05],\n",
              "            [-1.28161197e-03,  9.90264234e-05,  1.64454326e-03,\n",
              "             -1.14600116e-03, -1.31412398e-03,  2.61259777e-03,\n",
              "             -4.90985462e-04, -2.98799889e-04, -6.67527318e-04,\n",
              "              8.42879526e-04],\n",
              "            [-2.99899676e-03,  1.01122621e-03,  2.22213939e-03,\n",
              "             -1.47662079e-03, -1.95053010e-03,  2.89933011e-03,\n",
              "             -4.19334712e-04, -2.34769177e-04,  2.98903586e-04,\n",
              "              6.48648827e-04],\n",
              "            [-5.68731106e-04, -3.62690407e-05,  3.00736679e-03,\n",
              "             -2.81353178e-03, -3.12945363e-03,  4.42836061e-03,\n",
              "              2.28328499e-04, -1.09879533e-04, -1.65555021e-03,\n",
              "              6.49358379e-04],\n",
              "            [-1.04495906e-03,  1.14828697e-04,  1.39649527e-03,\n",
              "             -1.70309376e-03, -1.93435140e-03,  2.71155848e-03,\n",
              "             -2.80452718e-04,  4.18123673e-04, -8.60053638e-04,\n",
              "              1.18190143e-03],\n",
              "            [-3.13145312e-04,  4.14907903e-04,  2.02818122e-03,\n",
              "              2.69656884e-05, -9.76303883e-04,  4.10453998e-04,\n",
              "             -3.91295209e-04, -4.00419958e-04, -1.10643567e-03,\n",
              "              3.07090522e-04],\n",
              "            [-8.07978970e-04,  1.63070648e-03, -8.90036579e-04,\n",
              "             -1.10395800e-03, -1.84089283e-03,  2.39393534e-03,\n",
              "             -9.55584634e-04,  9.71515779e-04, -2.30862759e-04,\n",
              "              8.33152328e-04],\n",
              "            [-1.26279378e-03,  4.41962620e-05,  1.78748858e-03,\n",
              "             -9.22488922e-04, -1.10592763e-03,  2.25606025e-03,\n",
              "             -5.23395895e-04,  1.15729083e-04, -1.27722847e-03,\n",
              "              8.88357754e-04],\n",
              "            [-1.96867529e-03,  9.73261369e-04,  1.55687798e-03,\n",
              "             -1.45418709e-03, -2.84072303e-04,  1.85476313e-03,\n",
              "              6.29440940e-04, -6.11223688e-04, -1.70006533e-04,\n",
              "             -5.26177522e-04],\n",
              "            [-3.12658888e-03,  6.96551229e-04,  2.88152427e-04,\n",
              "             -9.32161347e-04, -8.88012437e-06,  2.06604856e-03,\n",
              "              7.70174665e-04,  4.17361473e-04, -3.25931993e-04,\n",
              "              1.55270594e-04],\n",
              "            [-1.05083047e-03,  2.33270708e-04,  2.19589635e-03,\n",
              "             -3.13130347e-03,  7.35043068e-05,  3.26320436e-03,\n",
              "              2.20186965e-04, -1.39880603e-05, -1.48027600e-03,\n",
              "             -3.09665658e-04],\n",
              "            [-6.93936192e-04,  4.10693057e-04,  1.81289716e-03,\n",
              "             -3.20841908e-03, -1.84890896e-03,  4.95979004e-03,\n",
              "             -6.88030195e-05, -8.79705651e-04,  1.78752758e-04,\n",
              "             -6.62361737e-04],\n",
              "            [-1.12702255e-03, -2.28004705e-04,  1.28644984e-03,\n",
              "             -1.32495805e-03, -1.38475763e-04,  2.49337917e-03,\n",
              "              5.99985942e-04, -7.44629360e-04, -5.28421660e-04,\n",
              "             -2.88301526e-04],\n",
              "            [-7.67885824e-04,  5.17686596e-04,  3.42002977e-03,\n",
              "             -1.10946456e-03, -3.48728237e-04,  1.20564899e-03,\n",
              "             -1.53049652e-04, -2.01125583e-03, -1.13708538e-03,\n",
              "              3.84105253e-04],\n",
              "            [-2.59864540e-03,  4.06491890e-04,  1.30082283e-03,\n",
              "             -9.43018822e-04, -1.14628361e-04,  1.56978914e-03,\n",
              "              4.95719432e-04,  1.98360969e-04, -7.52196065e-05,\n",
              "             -2.39672925e-04],\n",
              "            [-2.60437909e-03, -3.76146054e-04,  2.17644777e-03,\n",
              "              3.27274611e-04, -6.00389147e-04,  1.51273923e-03,\n",
              "              4.57231654e-05,  9.48567758e-05, -1.23452232e-03,\n",
              "              6.58391567e-04],\n",
              "            [-1.67279947e-03,  9.29660455e-05,  1.58472569e-03,\n",
              "             -8.06469470e-04, -2.09968319e-04,  1.16490154e-03,\n",
              "             -1.26735496e-04, -9.15755576e-04,  5.92704804e-04,\n",
              "              2.96429003e-04],\n",
              "            [-8.97705497e-04, -3.16556252e-05, -4.28597385e-04,\n",
              "             -4.83497570e-04,  1.11538247e-05,  2.69695814e-03,\n",
              "              2.43210976e-04, -5.13377716e-04, -1.01780356e-03,\n",
              "              4.21313627e-04]], dtype=float32)}},\n",
              "   'nu': {'BatchNorm_0': {'bias': array([0.00101139, 0.00073047, 0.0005574 , 0.00392027, 0.00126213,\n",
              "            0.00135049, 0.00178212, 0.00649406, 0.00089966, 0.00162631,\n",
              "            0.00140204, 0.00443221, 0.00142618, 0.00380787, 0.00134232,\n",
              "            0.001241  ], dtype=float32),\n",
              "     'scale': array([0.005206  , 0.00545367, 0.00071883, 0.00151108, 0.00142804,\n",
              "            0.00117757, 0.0017718 , 0.0031161 , 0.00336226, 0.00132863,\n",
              "            0.00442754, 0.00228052, 0.00558872, 0.00159105, 0.00482422,\n",
              "            0.00110281], dtype=float32)},\n",
              "    'BatchNorm_1': {'bias': array([1.0130052e-04, 8.6864165e-05, 9.1456022e-05, 7.1520648e-05,\n",
              "            9.8731631e-05, 1.0031465e-04, 9.1163311e-05, 1.2768224e-04,\n",
              "            8.5281543e-05, 8.5903295e-05, 1.0165682e-04, 6.2871150e-05,\n",
              "            8.8750174e-05, 8.9115434e-05, 9.5840369e-05, 9.8070464e-05,\n",
              "            1.0910749e-04, 6.8171772e-05, 9.8025012e-05, 7.6197102e-05,\n",
              "            7.8218480e-05, 8.6962289e-05, 6.4860164e-05, 7.6427299e-05,\n",
              "            8.9037843e-05, 7.7504548e-05, 1.0445524e-04, 9.7961376e-05,\n",
              "            9.4355295e-05, 9.1443973e-05, 8.7844826e-05, 9.1353802e-05],\n",
              "           dtype=float32),\n",
              "     'scale': array([1.14510993e-04, 7.08603256e-05, 8.45457835e-05, 6.42467567e-05,\n",
              "            9.40625177e-05, 7.79249531e-05, 8.37739863e-05, 1.08421671e-04,\n",
              "            7.79901093e-05, 7.19922464e-05, 1.00503355e-04, 1.02730759e-04,\n",
              "            8.13975130e-05, 6.73692266e-05, 1.06988438e-04, 8.94192126e-05,\n",
              "            8.18244007e-05, 3.52624738e-05, 7.63886128e-05, 9.31179529e-05,\n",
              "            7.60773401e-05, 9.98369942e-05, 7.54975263e-05, 8.60409127e-05,\n",
              "            9.75254952e-05, 5.64062138e-05, 1.22228608e-04, 1.46436272e-04,\n",
              "            1.26084167e-04, 7.42758784e-05, 7.36813090e-05, 7.99041227e-05],\n",
              "           dtype=float32)},\n",
              "    'Conv_0': {'bias': array([7.5595590e-18, 3.3558989e-17, 3.1486167e-17, 2.8415655e-16,\n",
              "            2.6370811e-17, 4.8783208e-17, 8.4870921e-17, 6.3300164e-16,\n",
              "            1.6688483e-17, 5.1249732e-17, 1.3506875e-16, 3.4762026e-16,\n",
              "            7.4673581e-18, 4.5009954e-16, 2.4117832e-17, 4.6492990e-17],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[2.17553807e-05, 1.56926995e-04, 1.10942980e-04,\n",
              "               3.80651996e-04, 1.38571791e-04, 2.84411974e-04,\n",
              "               2.73982350e-05, 1.52390840e-05, 2.84822127e-05,\n",
              "               2.20491827e-04, 2.48716504e-04, 5.20604262e-05,\n",
              "               2.69853335e-05, 2.48752476e-04, 6.92595640e-05,\n",
              "               2.47153366e-04]],\n",
              "     \n",
              "             [[1.02198355e-05, 1.47596671e-04, 6.82614482e-05,\n",
              "               3.42318963e-04, 1.31515713e-04, 2.90435477e-04,\n",
              "               2.58711698e-05, 4.46222875e-05, 4.24700083e-05,\n",
              "               2.05234130e-04, 2.33479863e-04, 1.86857487e-05,\n",
              "               3.16315636e-05, 3.27240530e-04, 6.59364159e-05,\n",
              "               1.98383204e-04]],\n",
              "     \n",
              "             [[1.35699920e-05, 1.03929640e-04, 4.07984662e-05,\n",
              "               2.86029506e-04, 1.26875224e-04, 2.95335922e-04,\n",
              "               2.79171163e-05, 1.50565262e-04, 6.02963300e-05,\n",
              "               1.73092529e-04, 2.37789223e-04, 9.12566338e-06,\n",
              "               3.53315954e-05, 4.51573811e-04, 6.93487091e-05,\n",
              "               1.15867588e-04]],\n",
              "     \n",
              "             [[2.60287052e-05, 1.41609606e-04, 3.58211000e-05,\n",
              "               2.23875031e-04, 1.23854828e-04, 2.87491333e-04,\n",
              "               2.49183249e-05, 3.09297524e-04, 5.66879426e-05,\n",
              "               9.69232860e-05, 2.92727666e-04, 8.01480473e-06,\n",
              "               4.22518242e-05, 5.75260026e-04, 7.30354295e-05,\n",
              "               2.40450663e-05]],\n",
              "     \n",
              "             [[3.60032136e-05, 1.84338787e-04, 4.09309941e-05,\n",
              "               1.65861537e-04, 1.25244565e-04, 2.65414361e-04,\n",
              "               2.10914222e-05, 4.59069939e-04, 5.82339053e-05,\n",
              "               3.89077322e-05, 2.83294154e-04, 5.55882207e-06,\n",
              "               4.12305817e-05, 6.57361932e-04, 7.43802884e-05,\n",
              "               2.00288741e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[1.62210417e-05, 1.33281865e-04, 1.76450849e-04,\n",
              "               3.08491552e-04, 9.30126407e-05, 2.61369423e-04,\n",
              "               7.52506166e-05, 2.84777416e-05, 2.76757783e-05,\n",
              "               1.95900066e-04, 2.19873429e-04, 8.67206109e-05,\n",
              "               2.45947431e-05, 2.32000442e-04, 8.53684178e-05,\n",
              "               2.70517456e-04]],\n",
              "     \n",
              "             [[8.08313689e-06, 1.38300224e-04, 1.36472110e-04,\n",
              "               2.25841388e-04, 8.52569065e-05, 2.57973268e-04,\n",
              "               5.98847582e-05, 7.95262094e-05, 4.10433968e-05,\n",
              "               1.62938348e-04, 2.10889048e-04, 6.03375011e-05,\n",
              "               2.59072312e-05, 3.11306940e-04, 8.48857235e-05,\n",
              "               2.23202369e-04]],\n",
              "     \n",
              "             [[1.45022677e-05, 1.03196202e-04, 7.56759327e-05,\n",
              "               1.46486185e-04, 8.34713792e-05, 2.48304976e-04,\n",
              "               5.03013944e-05, 1.93152795e-04, 5.26780532e-05,\n",
              "               1.12629546e-04, 2.40546244e-04, 4.53662251e-05,\n",
              "               3.10658688e-05, 3.93179245e-04, 8.92893877e-05,\n",
              "               1.31073059e-04]],\n",
              "     \n",
              "             [[2.54541992e-05, 9.52034170e-05, 4.10120556e-05,\n",
              "               7.54180219e-05, 8.83360990e-05, 2.45777046e-04,\n",
              "               4.71865169e-05, 3.64197796e-04, 4.41422599e-05,\n",
              "               4.42062228e-05, 2.54170242e-04, 2.47170738e-05,\n",
              "               3.96175892e-05, 4.70307510e-04, 9.21074679e-05,\n",
              "               3.40524966e-05]],\n",
              "     \n",
              "             [[3.44854416e-05, 1.66179045e-04, 2.84929592e-05,\n",
              "               4.49585677e-05, 9.72454291e-05, 2.21315218e-04,\n",
              "               5.09184356e-05, 4.97851637e-04, 5.88062321e-05,\n",
              "               1.63402383e-05, 2.10416503e-04, 9.05501292e-06,\n",
              "               3.85157437e-05, 5.26044576e-04, 9.35738062e-05,\n",
              "               2.09356258e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[1.85524423e-05, 1.02177743e-04, 2.03906500e-04,\n",
              "               1.92455132e-04, 3.95503776e-05, 2.17860797e-04,\n",
              "               1.57825314e-04, 5.82208013e-05, 2.37063832e-05,\n",
              "               1.75909561e-04, 2.16102431e-04, 1.48587671e-04,\n",
              "               2.15307591e-05, 1.30910121e-04, 7.41691329e-05,\n",
              "               2.80315144e-04]],\n",
              "     \n",
              "             [[1.22814445e-05, 1.26005820e-04, 2.25981668e-04,\n",
              "               9.64743886e-05, 3.87695254e-05, 1.96721769e-04,\n",
              "               1.36200266e-04, 1.22448997e-04, 3.69333793e-05,\n",
              "               1.30994536e-04, 2.07230754e-04, 1.20801109e-04,\n",
              "               2.02016890e-05, 1.35872528e-04, 7.93515137e-05,\n",
              "               2.39104746e-04]],\n",
              "     \n",
              "             [[1.76956855e-05, 1.27541338e-04, 1.61972799e-04,\n",
              "               3.52585630e-05, 4.42637320e-05, 1.71758613e-04,\n",
              "               1.22675789e-04, 2.24004005e-04, 4.18176896e-05,\n",
              "               7.25754944e-05, 2.24453033e-04, 9.69439061e-05,\n",
              "               2.71129229e-05, 1.55936912e-04, 8.67588096e-05,\n",
              "               1.47129336e-04]],\n",
              "     \n",
              "             [[2.86602808e-05, 8.90282172e-05, 9.43506893e-05,\n",
              "               1.53635847e-05, 5.36801490e-05, 1.54258174e-04,\n",
              "               1.40348420e-04, 3.23231041e-04, 3.67223911e-05,\n",
              "               2.55407158e-05, 2.22465853e-04, 4.97689252e-05,\n",
              "               3.38624668e-05, 2.02094365e-04, 8.35114042e-05,\n",
              "               4.96891880e-05]],\n",
              "     \n",
              "             [[3.49572474e-05, 1.42783058e-04, 5.26404656e-05,\n",
              "               4.86293939e-05, 6.86979474e-05, 1.30166765e-04,\n",
              "               1.67948907e-04, 4.14308219e-04, 6.36083787e-05,\n",
              "               2.05979268e-05, 2.34703795e-04, 2.03355794e-05,\n",
              "               3.44305372e-05, 2.68095435e-04, 8.58290368e-05,\n",
              "               2.18300374e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[2.40587324e-05, 9.39479069e-05, 1.59450414e-04,\n",
              "               8.66098126e-05, 1.70397525e-05, 1.12261274e-04,\n",
              "               2.45697622e-04, 5.17076114e-05, 2.58722412e-05,\n",
              "               1.62839206e-04, 1.89832659e-04, 2.05224977e-04,\n",
              "               1.77304537e-05, 3.89422348e-05, 8.61685330e-05,\n",
              "               2.92778888e-04]],\n",
              "     \n",
              "             [[2.06979003e-05, 1.23962469e-04, 2.30030768e-04,\n",
              "               2.41666512e-05, 1.86241632e-05, 8.50731958e-05,\n",
              "               2.32323655e-04, 7.66332159e-05, 4.55980298e-05,\n",
              "               1.12288450e-04, 1.94477558e-04, 1.51144850e-04,\n",
              "               1.62523156e-05, 2.49209825e-05, 8.84720794e-05,\n",
              "               2.57852516e-04]],\n",
              "     \n",
              "             [[2.71974441e-05, 1.66063706e-04, 2.35202286e-04,\n",
              "               1.27666563e-05, 2.40488662e-05, 6.36129917e-05,\n",
              "               2.29896061e-04, 1.11906855e-04, 4.77706708e-05,\n",
              "               4.64742843e-05, 2.37972868e-04, 1.00449804e-04,\n",
              "               2.11591887e-05, 2.80864879e-05, 8.73843383e-05,\n",
              "               1.67686085e-04]],\n",
              "     \n",
              "             [[3.68203728e-05, 1.38905103e-04, 1.75944369e-04,\n",
              "               4.45920014e-05, 2.86677296e-05, 5.31483529e-05,\n",
              "               2.58356507e-04, 1.74971457e-04, 4.65428311e-05,\n",
              "               1.94825443e-05, 2.42646784e-04, 3.54085605e-05,\n",
              "               2.60928864e-05, 5.40963374e-05, 8.59271167e-05,\n",
              "               7.52206834e-05]],\n",
              "     \n",
              "             [[3.75628988e-05, 1.44293459e-04, 1.24249476e-04,\n",
              "               1.25885912e-04, 4.14483111e-05, 4.09333734e-05,\n",
              "               2.92536482e-04, 2.79560219e-04, 7.14347334e-05,\n",
              "               3.09232528e-05, 2.45297648e-04, 7.94808329e-06,\n",
              "               2.96312664e-05, 1.03705373e-04, 9.64839128e-05,\n",
              "               2.36916439e-05]]],\n",
              "     \n",
              "     \n",
              "            [[[3.22187625e-05, 1.04889434e-04, 9.13152107e-05,\n",
              "               3.49735419e-05, 1.46328612e-05, 2.80992808e-05,\n",
              "               3.00966785e-04, 3.99171768e-05, 3.77300639e-05,\n",
              "               1.63741308e-04, 1.96294961e-04, 2.41357047e-04,\n",
              "               2.15153032e-05, 1.69014398e-04, 7.86294113e-05,\n",
              "               3.20727879e-04]],\n",
              "     \n",
              "             [[3.09621428e-05, 1.41234646e-04, 1.44223028e-04,\n",
              "               2.25875392e-05, 1.72733744e-05, 1.31291663e-05,\n",
              "               3.04876710e-04, 2.26790726e-05, 6.89796507e-05,\n",
              "               1.09058354e-04, 2.51745485e-04, 1.66458180e-04,\n",
              "               1.93368651e-05, 1.53633344e-04, 7.51127664e-05,\n",
              "               2.90485506e-04]],\n",
              "     \n",
              "             [[3.89136767e-05, 1.80785952e-04, 1.90779392e-04,\n",
              "               5.92770084e-05, 1.84036471e-05, 1.80006355e-05,\n",
              "               3.11873126e-04, 3.07017144e-05, 6.79078948e-05,\n",
              "               3.78467266e-05, 2.61860550e-04, 8.90512092e-05,\n",
              "               2.22884191e-05, 1.63271121e-04, 7.34556888e-05,\n",
              "               2.30640057e-04]],\n",
              "     \n",
              "             [[4.32706947e-05, 1.78141825e-04, 2.01409508e-04,\n",
              "               1.45480459e-04, 1.77116763e-05, 1.72270466e-05,\n",
              "               3.37932666e-04, 9.42432889e-05, 5.71111595e-05,\n",
              "               2.06155109e-05, 2.61940731e-04, 2.28259014e-05,\n",
              "               2.50256689e-05, 2.06126584e-04, 7.83806245e-05,\n",
              "               1.49136802e-04]],\n",
              "     \n",
              "             [[3.71751157e-05, 1.68148079e-04, 1.73353503e-04,\n",
              "               2.41057758e-04, 2.46487616e-05, 1.15173616e-05,\n",
              "               3.61212500e-04, 2.18563262e-04, 7.74297660e-05,\n",
              "               5.61804809e-05, 2.87889387e-04, 2.06387303e-05,\n",
              "               2.78351126e-05, 2.76670384e-04, 8.63972018e-05,\n",
              "               7.33213019e-05]]]], dtype=float32)},\n",
              "    'Conv_1': {'bias': array([1.4738033e-18, 8.2453762e-19, 1.9965005e-18, 8.3805132e-19,\n",
              "            3.1968215e-18, 2.5789610e-18, 5.9042777e-19, 1.3029746e-18,\n",
              "            1.0570425e-18, 1.5025055e-18, 1.2287711e-18, 1.5226551e-18,\n",
              "            7.0519189e-19, 1.4929031e-18, 1.7486625e-18, 1.2624759e-18,\n",
              "            8.8870682e-19, 7.1771768e-19, 7.8910399e-19, 2.2977760e-18,\n",
              "            7.2541754e-19, 1.9135370e-18, 4.6440427e-19, 1.5561282e-18,\n",
              "            1.3385546e-18, 1.0623657e-18, 7.8485688e-19, 3.8601703e-18,\n",
              "            1.0199614e-18, 6.3067305e-19, 1.1293457e-18, 1.6840298e-18],\n",
              "           dtype=float32),\n",
              "     'kernel': array([[[[2.8610174e-07, 5.0465201e-07, 8.4211666e-07, ...,\n",
              "               1.6617160e-07, 1.7124954e-07, 3.0630079e-07],\n",
              "              [6.8292098e-07, 8.1452390e-07, 2.0196990e-06, ...,\n",
              "               2.6442964e-07, 1.2363732e-06, 1.2104858e-06],\n",
              "              [9.7704140e-07, 7.0382242e-07, 1.1282133e-06, ...,\n",
              "               2.3505252e-07, 8.4228145e-07, 1.1342238e-06],\n",
              "              ...,\n",
              "              [6.0204246e-07, 1.1971181e-06, 2.2046804e-06, ...,\n",
              "               6.0622062e-07, 9.2031462e-07, 1.0933817e-06],\n",
              "              [1.6879949e-06, 1.7395898e-06, 3.4639388e-06, ...,\n",
              "               4.9283130e-07, 7.1435949e-07, 1.8324724e-06],\n",
              "              [1.0915494e-06, 7.1735258e-07, 1.5369504e-06, ...,\n",
              "               4.2575277e-07, 7.2065603e-07, 1.3305580e-06]],\n",
              "     \n",
              "             [[6.5596504e-07, 7.9022556e-07, 8.6776714e-07, ...,\n",
              "               3.7014129e-07, 6.3142170e-07, 5.1085004e-07],\n",
              "              [8.2255758e-07, 7.7006888e-07, 2.4250558e-06, ...,\n",
              "               3.9156808e-07, 1.1801134e-06, 1.4750742e-06],\n",
              "              [1.1797617e-06, 6.9033945e-07, 1.2354448e-06, ...,\n",
              "               4.3956103e-07, 8.7670725e-07, 1.4867433e-06],\n",
              "              ...,\n",
              "              [7.5781264e-07, 1.6223945e-06, 2.0534358e-06, ...,\n",
              "               5.0984062e-07, 8.1574854e-07, 1.3792339e-06],\n",
              "              [2.1541912e-06, 1.9830379e-06, 4.4462563e-06, ...,\n",
              "               3.7103993e-07, 6.8486963e-07, 2.4484532e-06],\n",
              "              [1.8075516e-06, 1.9003926e-06, 1.5476693e-06, ...,\n",
              "               7.4183413e-07, 1.7448089e-06, 2.4775788e-06]],\n",
              "     \n",
              "             [[6.2444514e-07, 6.8260817e-07, 1.3802569e-06, ...,\n",
              "               4.7526501e-07, 8.4599793e-07, 6.7687375e-07],\n",
              "              [7.7591335e-07, 1.7702297e-06, 3.3301806e-06, ...,\n",
              "               3.7061983e-07, 1.4096659e-06, 1.3331359e-06],\n",
              "              [1.5165334e-06, 1.4070288e-06, 2.1340268e-06, ...,\n",
              "               3.7517441e-07, 8.5994998e-07, 2.5054019e-06],\n",
              "              ...,\n",
              "              [8.9606999e-07, 1.7076989e-06, 1.7257299e-06, ...,\n",
              "               5.6228919e-07, 7.1960659e-07, 1.9341971e-06],\n",
              "              [2.4944254e-06, 1.7915729e-06, 4.3307746e-06, ...,\n",
              "               3.4770346e-07, 7.1961171e-07, 3.1569921e-06],\n",
              "              [2.5046493e-06, 1.9573067e-06, 2.2929614e-06, ...,\n",
              "               1.0908664e-06, 1.9684219e-06, 4.0859445e-06]],\n",
              "     \n",
              "             [[6.6243109e-07, 8.2372071e-07, 1.8620798e-06, ...,\n",
              "               6.1979000e-07, 8.8056805e-07, 6.2675269e-07],\n",
              "              [9.3574999e-07, 1.6962111e-06, 4.2634110e-06, ...,\n",
              "               3.0420028e-07, 1.1035296e-06, 1.8931788e-06],\n",
              "              [1.4163213e-06, 1.6837876e-06, 2.7761298e-06, ...,\n",
              "               1.9463343e-07, 8.6559936e-07, 2.5096522e-06],\n",
              "              ...,\n",
              "              [9.3614705e-07, 1.4464878e-06, 1.3112569e-06, ...,\n",
              "               6.2639600e-07, 4.8773899e-07, 1.8568799e-06],\n",
              "              [2.3811808e-06, 1.4288971e-06, 3.7798077e-06, ...,\n",
              "               3.2772215e-07, 7.4276937e-07, 3.6411288e-06],\n",
              "              [2.9767618e-06, 3.0816229e-06, 4.9175064e-06, ...,\n",
              "               1.5211803e-06, 2.3880298e-06, 5.4749553e-06]],\n",
              "     \n",
              "             [[6.9850483e-07, 7.7586134e-07, 2.1051428e-06, ...,\n",
              "               6.1727582e-07, 8.0996165e-07, 6.3026283e-07],\n",
              "              [1.3708191e-06, 1.6180832e-06, 3.5912306e-06, ...,\n",
              "               2.9813441e-07, 5.6772672e-07, 1.4056030e-06],\n",
              "              [1.8928781e-06, 1.2775943e-06, 2.9703695e-06, ...,\n",
              "               2.5070236e-07, 7.5016368e-07, 2.7763733e-06],\n",
              "              ...,\n",
              "              [9.6877727e-07, 9.6420672e-07, 6.3679011e-07, ...,\n",
              "               5.5107029e-07, 4.7317417e-07, 1.6057645e-06],\n",
              "              [2.1766998e-06, 1.1007963e-06, 3.4257805e-06, ...,\n",
              "               3.9364508e-07, 7.7424221e-07, 3.6949978e-06],\n",
              "              [3.3416170e-06, 2.8928073e-06, 5.8674036e-06, ...,\n",
              "               1.3256594e-06, 2.2520007e-06, 4.9887994e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[6.3219727e-07, 5.7446061e-07, 9.7072871e-07, ...,\n",
              "               2.1055119e-07, 2.3398539e-07, 5.8322610e-07],\n",
              "              [1.4576360e-06, 9.0813683e-07, 2.0520138e-06, ...,\n",
              "               2.8865222e-07, 1.5476151e-06, 1.7522293e-06],\n",
              "              [7.0030654e-07, 4.5599396e-07, 1.1546233e-06, ...,\n",
              "               2.1831434e-07, 4.8035912e-07, 6.2678248e-07],\n",
              "              ...,\n",
              "              [1.2439048e-06, 1.5984234e-06, 2.6660659e-06, ...,\n",
              "               3.9721706e-07, 6.9344657e-07, 1.4332178e-06],\n",
              "              [2.0122204e-06, 7.9767443e-07, 3.9064366e-06, ...,\n",
              "               6.8287625e-07, 5.7840617e-07, 1.6695807e-06],\n",
              "              [9.5047892e-07, 8.7727165e-07, 1.1541025e-06, ...,\n",
              "               4.8414699e-07, 8.2387373e-07, 8.1699665e-07]],\n",
              "     \n",
              "             [[1.0828595e-06, 7.9502183e-07, 9.2098628e-07, ...,\n",
              "               5.5337767e-07, 9.0477965e-07, 7.8697343e-07],\n",
              "              [1.3556720e-06, 1.2058064e-06, 2.1147234e-06, ...,\n",
              "               1.9250376e-07, 1.3707330e-06, 2.1465457e-06],\n",
              "              [1.4617508e-06, 5.1405863e-07, 1.3678521e-06, ...,\n",
              "               1.6514255e-07, 4.6699839e-07, 1.5445501e-06],\n",
              "              ...,\n",
              "              [1.3125851e-06, 1.7651132e-06, 2.4497654e-06, ...,\n",
              "               3.5389013e-07, 4.8398795e-07, 1.6546759e-06],\n",
              "              [2.6160169e-06, 1.0258395e-06, 4.6089540e-06, ...,\n",
              "               6.3038385e-07, 5.9947831e-07, 2.2613006e-06],\n",
              "              [2.0599391e-06, 2.1868143e-06, 1.4571779e-06, ...,\n",
              "               6.5490985e-07, 1.9359363e-06, 2.3318576e-06]],\n",
              "     \n",
              "             [[1.0920063e-06, 7.1778555e-07, 1.9241859e-06, ...,\n",
              "               5.8455151e-07, 8.8901993e-07, 1.1133978e-06],\n",
              "              [1.8673886e-06, 1.7749981e-06, 5.0852805e-06, ...,\n",
              "               2.2148252e-07, 9.1321346e-07, 2.1703322e-06],\n",
              "              [1.7905636e-06, 8.5485163e-07, 1.1560104e-06, ...,\n",
              "               2.0740310e-07, 4.9166266e-07, 1.5152493e-06],\n",
              "              ...,\n",
              "              [1.2374165e-06, 1.3140750e-06, 2.8454654e-06, ...,\n",
              "               4.6352974e-07, 3.7028769e-07, 1.9200179e-06],\n",
              "              [2.8496020e-06, 1.2021987e-06, 4.6189580e-06, ...,\n",
              "               6.6776454e-07, 5.8248526e-07, 2.7785090e-06],\n",
              "              [2.6270063e-06, 2.0341131e-06, 1.5849364e-06, ...,\n",
              "               9.2757779e-07, 1.3593715e-06, 3.9422839e-06]],\n",
              "     \n",
              "             [[1.2096393e-06, 7.9805307e-07, 2.5027441e-06, ...,\n",
              "               8.2006250e-07, 9.2542757e-07, 1.1959845e-06],\n",
              "              [1.4422912e-06, 2.1882709e-06, 3.7480645e-06, ...,\n",
              "               1.3477182e-07, 9.4894904e-07, 2.5157663e-06],\n",
              "              [2.1197275e-06, 1.4460343e-06, 1.3686262e-06, ...,\n",
              "               4.5557803e-07, 5.4863290e-07, 2.2221673e-06],\n",
              "              ...,\n",
              "              [1.2655030e-06, 9.2973693e-07, 1.4592422e-06, ...,\n",
              "               3.1623620e-07, 2.3174228e-07, 1.9264394e-06],\n",
              "              [2.7661583e-06, 1.1465690e-06, 4.8263564e-06, ...,\n",
              "               6.0751296e-07, 7.5419274e-07, 3.0701021e-06],\n",
              "              [3.5006085e-06, 3.2899609e-06, 3.8696644e-06, ...,\n",
              "               1.6557871e-06, 1.9603331e-06, 4.9535101e-06]],\n",
              "     \n",
              "             [[1.1614824e-06, 8.6058373e-07, 2.8687557e-06, ...,\n",
              "               7.9032935e-07, 8.1465163e-07, 1.2672872e-06],\n",
              "              [2.2124802e-06, 1.7614659e-06, 3.8906719e-06, ...,\n",
              "               2.6988161e-07, 6.0632181e-07, 2.3856539e-06],\n",
              "              [2.0475354e-06, 1.3986962e-06, 1.7858591e-06, ...,\n",
              "               4.0413616e-07, 7.0603761e-07, 2.3819844e-06],\n",
              "              ...,\n",
              "              [1.2712322e-06, 8.4419707e-07, 6.0483086e-07, ...,\n",
              "               3.2454508e-07, 2.5492673e-07, 1.3805021e-06],\n",
              "              [2.7088615e-06, 1.1141302e-06, 4.5817742e-06, ...,\n",
              "               5.9134908e-07, 9.1994065e-07, 3.3102610e-06],\n",
              "              [4.0151754e-06, 3.7768718e-06, 5.3743393e-06, ...,\n",
              "               1.3022860e-06, 2.2183483e-06, 6.2412687e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[9.5317660e-07, 4.6908590e-07, 1.1733820e-06, ...,\n",
              "               3.8861529e-07, 5.3709397e-07, 5.1945301e-07],\n",
              "              [1.6711972e-06, 6.7192076e-07, 2.0311334e-06, ...,\n",
              "               3.7735262e-07, 1.2389048e-06, 1.8048612e-06],\n",
              "              [8.8632135e-07, 6.2315775e-07, 6.9959322e-07, ...,\n",
              "               2.5830815e-07, 6.8404483e-07, 1.4464422e-07],\n",
              "              ...,\n",
              "              [1.7788698e-06, 9.9099100e-07, 2.7266781e-06, ...,\n",
              "               5.6605126e-07, 5.6120240e-07, 1.6133023e-06],\n",
              "              [1.7027569e-06, 1.4982551e-06, 3.9616912e-06, ...,\n",
              "               5.0095088e-07, 1.1183456e-06, 1.3269281e-06],\n",
              "              [9.8305873e-07, 9.0972537e-07, 7.8825991e-07, ...,\n",
              "               6.4394141e-07, 6.0965141e-07, 2.2485889e-07]],\n",
              "     \n",
              "             [[1.3122327e-06, 7.8999989e-07, 1.4410970e-06, ...,\n",
              "               7.9124203e-07, 1.0817854e-06, 1.0343048e-06],\n",
              "              [1.9779386e-06, 7.8086418e-07, 2.4260189e-06, ...,\n",
              "               2.2133514e-07, 1.3192604e-06, 2.2438367e-06],\n",
              "              [1.0401736e-06, 4.9271176e-07, 9.6504493e-07, ...,\n",
              "               1.6826277e-07, 7.6886920e-07, 3.0598770e-07],\n",
              "              ...,\n",
              "              [1.8718637e-06, 9.8738769e-07, 2.3863922e-06, ...,\n",
              "               5.2217626e-07, 3.4457022e-07, 1.8988379e-06],\n",
              "              [2.7652134e-06, 2.1865435e-06, 4.1660769e-06, ...,\n",
              "               4.9384266e-07, 1.1759438e-06, 1.6754307e-06],\n",
              "              [1.9543334e-06, 2.4428089e-06, 9.4542798e-07, ...,\n",
              "               7.3310434e-07, 1.6652066e-06, 8.7429947e-07]],\n",
              "     \n",
              "             [[1.2600610e-06, 7.4336396e-07, 2.5515833e-06, ...,\n",
              "               9.3107184e-07, 8.9310328e-07, 1.2629215e-06],\n",
              "              [2.5089373e-06, 1.1985112e-06, 2.2941920e-06, ...,\n",
              "               1.7166471e-07, 1.3362987e-06, 3.3033230e-06],\n",
              "              [2.1775941e-06, 7.9431294e-07, 8.8485854e-07, ...,\n",
              "               2.0846539e-07, 3.3695278e-07, 6.8300704e-07],\n",
              "              ...,\n",
              "              [2.0199907e-06, 7.7037032e-07, 2.3108682e-06, ...,\n",
              "               6.1753582e-07, 2.2229919e-07, 2.1012884e-06],\n",
              "              [3.3710119e-06, 2.6263417e-06, 3.6733231e-06, ...,\n",
              "               5.5017364e-07, 1.3835682e-06, 2.1482185e-06],\n",
              "              [2.3259724e-06, 1.4878274e-06, 1.0969989e-06, ...,\n",
              "               9.4970557e-07, 1.4440792e-06, 1.5740131e-06]],\n",
              "     \n",
              "             [[1.5214912e-06, 9.9931822e-07, 3.0915849e-06, ...,\n",
              "               9.2385505e-07, 1.0858987e-06, 1.7567264e-06],\n",
              "              [2.4602382e-06, 1.8997688e-06, 2.7309279e-06, ...,\n",
              "               2.3213180e-07, 1.0122267e-06, 2.3034706e-06],\n",
              "              [2.3542862e-06, 8.8640127e-07, 6.1775847e-07, ...,\n",
              "               4.3257114e-07, 5.0880084e-07, 6.3899819e-07],\n",
              "              ...,\n",
              "              [1.6735905e-06, 7.2199407e-07, 1.1100514e-06, ...,\n",
              "               6.5257871e-07, 2.5711321e-07, 1.8808591e-06],\n",
              "              [3.9192969e-06, 2.9452674e-06, 3.4160598e-06, ...,\n",
              "               5.3753666e-07, 1.6199549e-06, 2.5820070e-06],\n",
              "              [3.6630584e-06, 2.5027687e-06, 2.6161217e-06, ...,\n",
              "               1.4423883e-06, 1.4299484e-06, 1.9180454e-06]],\n",
              "     \n",
              "             [[1.3092503e-06, 8.6081377e-07, 3.2339128e-06, ...,\n",
              "               8.8007664e-07, 9.5553776e-07, 1.6422588e-06],\n",
              "              [2.5436457e-06, 1.5889937e-06, 2.5435395e-06, ...,\n",
              "               2.1203527e-07, 9.1065226e-07, 2.6647142e-06],\n",
              "              [2.7870785e-06, 1.8794015e-06, 1.7177856e-06, ...,\n",
              "               3.2555292e-07, 9.4929158e-07, 9.4277993e-07],\n",
              "              ...,\n",
              "              [1.5364380e-06, 7.5873197e-07, 5.2149932e-07, ...,\n",
              "               5.7019116e-07, 3.3811421e-07, 1.3855259e-06],\n",
              "              [3.9573065e-06, 2.9212665e-06, 2.8902521e-06, ...,\n",
              "               6.5179205e-07, 1.7213232e-06, 3.2832111e-06],\n",
              "              [4.0742598e-06, 3.0488006e-06, 3.5072078e-06, ...,\n",
              "               1.4116080e-06, 2.0879561e-06, 2.9455448e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[1.0887444e-06, 7.0352945e-07, 1.4088648e-06, ...,\n",
              "               5.7171138e-07, 8.2568477e-07, 5.4900221e-07],\n",
              "              [1.5868466e-06, 1.1103142e-06, 1.2484069e-06, ...,\n",
              "               4.7371745e-07, 1.1441113e-06, 1.8810545e-06],\n",
              "              [6.6071357e-07, 5.1758622e-07, 6.0569289e-07, ...,\n",
              "               2.6522650e-07, 2.0238760e-07, 2.2283217e-07],\n",
              "              ...,\n",
              "              [1.6601624e-06, 1.1932483e-06, 2.0568184e-06, ...,\n",
              "               4.1087634e-07, 4.2056240e-07, 2.1389505e-06],\n",
              "              [1.9450354e-06, 1.7910190e-06, 2.4884162e-06, ...,\n",
              "               1.4169269e-06, 1.0855001e-06, 1.4088876e-06],\n",
              "              [6.4786161e-07, 4.8301951e-07, 5.6716016e-07, ...,\n",
              "               4.8175485e-07, 2.3259999e-07, 9.0620944e-08]],\n",
              "     \n",
              "             [[1.4261768e-06, 1.2049984e-06, 1.8555545e-06, ...,\n",
              "               1.1218569e-06, 1.1786655e-06, 1.2123189e-06],\n",
              "              [2.1606988e-06, 8.0068310e-07, 2.0355542e-06, ...,\n",
              "               1.9512419e-07, 1.4007569e-06, 1.6026816e-06],\n",
              "              [9.3008919e-07, 5.0522033e-07, 2.3909075e-07, ...,\n",
              "               3.3752139e-07, 4.1569871e-07, 4.4069699e-07],\n",
              "              ...,\n",
              "              [1.8549761e-06, 1.4761748e-06, 1.8808345e-06, ...,\n",
              "               4.1484847e-07, 4.6906615e-07, 2.4679318e-06],\n",
              "              [2.4136125e-06, 1.9496206e-06, 2.4992621e-06, ...,\n",
              "               1.5058549e-06, 1.1256567e-06, 1.6122959e-06],\n",
              "              [1.7026300e-06, 1.6159245e-06, 3.7963588e-07, ...,\n",
              "               4.4193129e-07, 1.1433001e-06, 3.8772978e-07]],\n",
              "     \n",
              "             [[1.4741040e-06, 1.3645000e-06, 2.4494382e-06, ...,\n",
              "               1.0200044e-06, 1.0456989e-06, 1.4673851e-06],\n",
              "              [3.1453660e-06, 1.4255650e-06, 2.2332740e-06, ...,\n",
              "               3.0967504e-07, 2.1090204e-06, 2.5695917e-06],\n",
              "              [1.2818688e-06, 4.0323098e-07, 4.1460430e-07, ...,\n",
              "               3.7365444e-07, 3.5021904e-07, 6.3284409e-07],\n",
              "              ...,\n",
              "              [2.2039367e-06, 1.5233658e-06, 1.6197937e-06, ...,\n",
              "               4.6430944e-07, 5.6288951e-07, 2.8702229e-06],\n",
              "              [2.6332191e-06, 2.2118847e-06, 2.2947195e-06, ...,\n",
              "               1.6866836e-06, 1.3215080e-06, 1.6916717e-06],\n",
              "              [2.3417122e-06, 1.1871875e-06, 1.1038960e-06, ...,\n",
              "               7.4952533e-07, 1.3830155e-06, 1.2283662e-06]],\n",
              "     \n",
              "             [[1.6229010e-06, 1.4590383e-06, 3.0648644e-06, ...,\n",
              "               9.6912902e-07, 1.4113848e-06, 1.6348438e-06],\n",
              "              [3.1157088e-06, 2.0234315e-06, 2.7851743e-06, ...,\n",
              "               4.0649758e-07, 1.7704978e-06, 2.2789620e-06],\n",
              "              [1.7527353e-06, 7.5541146e-07, 5.4864586e-07, ...,\n",
              "               6.4026892e-07, 7.8012266e-07, 5.1818552e-07],\n",
              "              ...,\n",
              "              [1.9418953e-06, 1.6966886e-06, 7.2473620e-07, ...,\n",
              "               4.4207678e-07, 5.4781441e-07, 3.0182434e-06],\n",
              "              [2.6253686e-06, 2.4900819e-06, 1.7836689e-06, ...,\n",
              "               1.7687142e-06, 1.6054089e-06, 2.0770763e-06],\n",
              "              [3.6053384e-06, 1.6664249e-06, 2.1963647e-06, ...,\n",
              "               1.3330061e-06, 1.1394272e-06, 2.0644984e-06]],\n",
              "     \n",
              "             [[1.6533847e-06, 1.3984164e-06, 3.0303690e-06, ...,\n",
              "               7.5740263e-07, 1.3831723e-06, 1.6861298e-06],\n",
              "              [3.0992460e-06, 2.4850858e-06, 2.9177381e-06, ...,\n",
              "               3.4487161e-07, 1.9360432e-06, 2.5018785e-06],\n",
              "              [1.9286242e-06, 9.8296528e-07, 6.7721686e-07, ...,\n",
              "               4.9695041e-07, 1.0193901e-06, 4.4273895e-07],\n",
              "              ...,\n",
              "              [1.5247722e-06, 1.4983364e-06, 5.3657067e-07, ...,\n",
              "               4.7371128e-07, 6.5640234e-07, 2.5245656e-06],\n",
              "              [2.5197901e-06, 2.0117623e-06, 1.3525359e-06, ...,\n",
              "               2.1637395e-06, 1.5602028e-06, 2.1525732e-06],\n",
              "              [4.1143085e-06, 2.1064184e-06, 2.7307196e-06, ...,\n",
              "               1.5442446e-06, 1.8076232e-06, 2.8034685e-06]]],\n",
              "     \n",
              "     \n",
              "            [[[9.2088464e-07, 1.0142483e-06, 1.4360962e-06, ...,\n",
              "               7.6287381e-07, 1.0217892e-06, 7.6112377e-07],\n",
              "              [1.6389089e-06, 2.4988753e-06, 1.0691240e-06, ...,\n",
              "               7.8451853e-07, 1.2656139e-06, 1.5039285e-06],\n",
              "              [3.5611174e-07, 4.8269641e-07, 2.3482563e-07, ...,\n",
              "               2.4209422e-07, 1.2264260e-07, 1.3016535e-07],\n",
              "              ...,\n",
              "              [1.6114435e-06, 1.7505308e-06, 1.6990350e-06, ...,\n",
              "               1.1815754e-06, 6.8707317e-07, 2.7043707e-06],\n",
              "              [1.7269228e-06, 2.9431135e-06, 1.9517415e-06, ...,\n",
              "               2.1578580e-06, 1.2691658e-06, 2.6632408e-06],\n",
              "              [6.4168466e-07, 3.6762765e-07, 4.8847960e-07, ...,\n",
              "               2.6733053e-07, 1.5493444e-07, 1.0393355e-07]],\n",
              "     \n",
              "             [[1.1785006e-06, 1.4627232e-06, 1.7317009e-06, ...,\n",
              "               1.1187739e-06, 1.2259020e-06, 9.3051801e-07],\n",
              "              [1.7701623e-06, 1.3061374e-06, 1.3873630e-06, ...,\n",
              "               4.7427341e-07, 2.0220662e-06, 1.3336470e-06],\n",
              "              [6.1244094e-07, 5.6776980e-07, 2.3574094e-07, ...,\n",
              "               1.6725014e-07, 2.2519878e-07, 2.6885945e-07],\n",
              "              ...,\n",
              "              [1.7384576e-06, 2.1616893e-06, 2.0822083e-06, ...,\n",
              "               9.7183818e-07, 1.2186558e-06, 2.9329565e-06],\n",
              "              [1.7723097e-06, 3.0896990e-06, 1.8586973e-06, ...,\n",
              "               2.4530086e-06, 1.4680249e-06, 3.0283993e-06],\n",
              "              [1.4203300e-06, 1.2134898e-06, 4.8925517e-07, ...,\n",
              "               2.5046000e-07, 8.9242957e-07, 3.1379659e-07]],\n",
              "     \n",
              "             [[1.2846797e-06, 1.4385150e-06, 2.4139913e-06, ...,\n",
              "               1.0312423e-06, 1.3466412e-06, 8.9642606e-07],\n",
              "              [2.6383934e-06, 2.3878551e-06, 2.3232872e-06, ...,\n",
              "               4.0711470e-07, 2.0626210e-06, 1.4575223e-06],\n",
              "              [9.8052124e-07, 7.4493164e-07, 4.0837000e-07, ...,\n",
              "               6.0464572e-07, 4.8376143e-07, 1.1547004e-06],\n",
              "              ...,\n",
              "              [1.5280410e-06, 2.3370478e-06, 2.5775446e-06, ...,\n",
              "               1.1184372e-06, 1.9212703e-06, 3.2244225e-06],\n",
              "              [1.3467909e-06, 2.8821680e-06, 1.5043643e-06, ...,\n",
              "               2.4311807e-06, 1.6960532e-06, 3.3876581e-06],\n",
              "              [1.6445434e-06, 1.3147502e-06, 1.0308374e-06, ...,\n",
              "               4.0894250e-07, 1.5998761e-06, 1.1082024e-06]],\n",
              "     \n",
              "             [[1.6307229e-06, 1.1593535e-06, 2.6458674e-06, ...,\n",
              "               9.6060739e-07, 1.4817464e-06, 6.6676461e-07],\n",
              "              [2.9446953e-06, 1.8387263e-06, 2.6676803e-06, ...,\n",
              "               6.3219215e-07, 2.0175389e-06, 1.6121301e-06],\n",
              "              [1.1172979e-06, 1.0297930e-06, 3.8532022e-07, ...,\n",
              "               5.7154665e-07, 5.1689716e-07, 1.5585648e-06],\n",
              "              ...,\n",
              "              [1.1312909e-06, 1.9932645e-06, 1.8265182e-06, ...,\n",
              "               1.1517620e-06, 1.8701572e-06, 3.2555236e-06],\n",
              "              [9.6179065e-07, 2.6114255e-06, 1.1500326e-06, ...,\n",
              "               2.3717216e-06, 2.0889534e-06, 2.9499054e-06],\n",
              "              [2.9578796e-06, 1.5856201e-06, 2.3759037e-06, ...,\n",
              "               9.6972758e-07, 1.3635073e-06, 2.6796083e-06]],\n",
              "     \n",
              "             [[1.8317776e-06, 1.1069191e-06, 2.1092480e-06, ...,\n",
              "               8.3918059e-07, 1.2877915e-06, 8.9515248e-07],\n",
              "              [2.2269005e-06, 2.5243198e-06, 3.0151045e-06, ...,\n",
              "               5.5526232e-07, 2.3544180e-06, 1.9004688e-06],\n",
              "              [1.1723901e-06, 1.1316562e-06, 8.1541629e-07, ...,\n",
              "               7.0479570e-07, 7.3923394e-07, 2.0333873e-06],\n",
              "              ...,\n",
              "              [1.1290817e-06, 1.2413692e-06, 1.0945729e-06, ...,\n",
              "               1.1123396e-06, 1.5635316e-06, 2.9636092e-06],\n",
              "              [7.3657185e-07, 2.5421978e-06, 1.1441223e-06, ...,\n",
              "               2.1792544e-06, 2.3274170e-06, 2.2141753e-06],\n",
              "              [2.9890016e-06, 1.6440113e-06, 2.4493243e-06, ...,\n",
              "               1.2525001e-06, 1.5195212e-06, 3.3414385e-06]]]], dtype=float32)},\n",
              "    'Dense_0': {'bias': array([6.1030274e-05, 3.7751084e-05, 9.5962831e-05, 6.6623790e-05,\n",
              "            7.3136209e-05, 7.4875577e-05, 5.8184793e-05, 7.4033196e-05,\n",
              "            8.9397552e-05, 1.0160340e-04], dtype=float32),\n",
              "     'kernel': array([[4.17882147e-05, 4.99834005e-06, 4.85574892e-05, 2.41716589e-05,\n",
              "             2.58661894e-05, 4.90197926e-05, 1.53476030e-05, 4.94129563e-05,\n",
              "             2.27519686e-05, 4.61121563e-05],\n",
              "            [4.70982268e-05, 2.90281077e-05, 9.35775388e-05, 7.44693680e-05,\n",
              "             3.27527014e-05, 5.08188605e-05, 4.33246278e-05, 2.40419904e-05,\n",
              "             5.35709223e-05, 2.12815175e-05],\n",
              "            [3.40452352e-05, 3.10351461e-05, 5.77912178e-05, 3.36365629e-05,\n",
              "             9.33866177e-05, 9.36230481e-06, 9.24435335e-06, 8.42995651e-05,\n",
              "             6.52941671e-05, 1.27007006e-04],\n",
              "            [1.26657915e-05, 6.40230564e-06, 4.19441458e-05, 7.09309097e-05,\n",
              "             9.75696312e-05, 8.72189339e-05, 1.19245497e-05, 6.50111906e-05,\n",
              "             8.35989995e-05, 1.33474023e-04],\n",
              "            [9.28192640e-06, 2.71280569e-05, 1.08536573e-04, 8.03348084e-05,\n",
              "             1.21571156e-05, 2.01343973e-05, 3.21022503e-06, 9.82071942e-05,\n",
              "             1.80765273e-05, 1.99407277e-05],\n",
              "            [8.97370701e-05, 2.66364059e-05, 2.12835894e-05, 1.20623272e-05,\n",
              "             4.37241652e-05, 8.50872602e-05, 7.34632558e-05, 4.59596522e-05,\n",
              "             1.40947930e-04, 7.44289355e-05],\n",
              "            [4.49856634e-05, 5.27752536e-06, 1.07894208e-04, 7.77190289e-05,\n",
              "             9.53975086e-06, 5.26187141e-05, 1.10719684e-05, 9.24300984e-05,\n",
              "             5.30466896e-05, 7.27374500e-05],\n",
              "            [2.82689507e-05, 3.11792901e-05, 3.33972275e-05, 3.72788490e-05,\n",
              "             2.68369477e-05, 4.22715493e-05, 2.94156216e-05, 1.37506522e-05,\n",
              "             5.14498752e-05, 1.69767627e-05],\n",
              "            [1.02754457e-04, 1.90031210e-06, 8.54928221e-05, 6.45646069e-05,\n",
              "             1.35887412e-05, 1.36919320e-04, 1.04240135e-04, 1.56915521e-05,\n",
              "             1.55193848e-04, 9.65717845e-05],\n",
              "            [4.81193274e-05, 2.13264066e-05, 8.05567688e-05, 8.70542717e-05,\n",
              "             2.30666265e-05, 8.26199466e-05, 3.32424752e-05, 5.49161377e-05,\n",
              "             1.53067231e-05, 4.49254694e-05],\n",
              "            [5.82794019e-05, 9.30374244e-06, 1.01901926e-04, 6.91137611e-05,\n",
              "             1.52986886e-05, 2.96297403e-05, 1.12447024e-05, 7.52289052e-05,\n",
              "             7.53691056e-05, 8.82305321e-05],\n",
              "            [1.58209787e-05, 3.66358654e-06, 2.86406230e-05, 9.65308573e-06,\n",
              "             2.99952444e-05, 3.81598838e-05, 2.64830142e-05, 1.52163248e-05,\n",
              "             3.43277316e-05, 1.95458233e-05],\n",
              "            [3.96797186e-05, 1.71660704e-05, 3.14367862e-05, 1.94619734e-05,\n",
              "             3.32769778e-05, 2.34518720e-05, 1.38417699e-05, 4.39183641e-05,\n",
              "             2.39571145e-05, 5.69248950e-05],\n",
              "            [5.14166313e-05, 3.06166621e-05, 5.87615286e-05, 1.31974639e-05,\n",
              "             4.39576907e-05, 1.81367832e-05, 5.25164396e-05, 2.64721912e-05,\n",
              "             5.00093920e-05, 1.65190686e-05],\n",
              "            [1.46492266e-05, 5.95343226e-06, 1.14027252e-05, 3.55149787e-05,\n",
              "             2.69241536e-05, 4.55533263e-05, 1.58342336e-05, 9.83063455e-06,\n",
              "             5.63308349e-05, 3.36902231e-05],\n",
              "            [1.78090286e-05, 2.44663825e-05, 3.67167559e-05, 1.60453274e-05,\n",
              "             2.82316360e-05, 4.21856967e-05, 3.84461928e-05, 4.07252337e-05,\n",
              "             2.20367692e-05, 2.41690632e-05],\n",
              "            [6.84430561e-05, 4.62560056e-05, 4.45329933e-05, 1.29941782e-05,\n",
              "             1.20823657e-04, 4.41728007e-05, 9.69360699e-05, 3.67556750e-05,\n",
              "             3.89749257e-05, 6.01607826e-05],\n",
              "            [1.46551329e-05, 2.99456206e-05, 1.39920216e-04, 1.36657647e-04,\n",
              "             1.18888420e-04, 8.91832751e-05, 7.92739593e-06, 1.55696413e-04,\n",
              "             1.10747991e-04, 2.00220951e-04],\n",
              "            [2.03724248e-05, 3.51223789e-05, 4.81264688e-05, 1.89023176e-05,\n",
              "             7.45591460e-05, 1.37089291e-05, 2.15529144e-05, 7.63209755e-05,\n",
              "             1.26342156e-05, 8.52367375e-05],\n",
              "            [5.50370787e-05, 2.61993523e-06, 5.74891055e-05, 1.77226702e-05,\n",
              "             4.23196616e-05, 2.28312601e-05, 5.31683836e-05, 5.60513990e-06,\n",
              "             8.88572758e-05, 4.57403803e-05],\n",
              "            [9.85498627e-06, 6.44634565e-05, 5.63081339e-05, 4.41586672e-05,\n",
              "             3.72937648e-05, 3.55715856e-05, 2.68507774e-05, 5.68950309e-05,\n",
              "             3.42182211e-05, 5.68303512e-05],\n",
              "            [3.81137252e-05, 1.63878690e-06, 1.32455207e-05, 1.02165923e-05,\n",
              "             2.18696168e-05, 7.41478289e-05, 5.51916382e-05, 1.33514104e-05,\n",
              "             7.40573960e-05, 5.98217812e-05],\n",
              "            [6.09774688e-05, 1.86048455e-05, 7.07526269e-05, 4.70668892e-05,\n",
              "             2.91446031e-05, 3.73409421e-05, 6.06398899e-05, 1.03214124e-05,\n",
              "             2.34932759e-05, 2.77885683e-05],\n",
              "            [7.48438033e-05, 4.01285797e-06, 3.60820668e-05, 9.29842099e-06,\n",
              "             8.97391510e-05, 2.39260644e-05, 8.27734475e-05, 5.47559148e-06,\n",
              "             7.34093846e-05, 8.76038248e-05],\n",
              "            [4.82781543e-05, 1.59557703e-06, 3.44766340e-05, 4.45283331e-05,\n",
              "             2.10252256e-05, 1.11705333e-04, 7.50385225e-05, 1.02855347e-05,\n",
              "             1.20556455e-04, 7.45978177e-05],\n",
              "            [1.20205195e-05, 2.57157481e-05, 7.02497637e-05, 8.97614227e-05,\n",
              "             1.01299287e-04, 7.58269598e-05, 7.23204948e-06, 1.02256927e-04,\n",
              "             3.46365050e-05, 1.47589701e-04],\n",
              "            [2.71843292e-05, 1.26486138e-05, 5.29714562e-05, 2.35885364e-05,\n",
              "             1.25288716e-05, 4.83395052e-05, 2.08838246e-05, 2.12075829e-05,\n",
              "             2.86102095e-05, 2.44549065e-05],\n",
              "            [1.26291743e-05, 3.10303608e-06, 4.58442046e-05, 1.43573088e-05,\n",
              "             2.60837714e-05, 9.61409842e-06, 2.60672550e-05, 1.59532010e-05,\n",
              "             4.83061376e-05, 1.58452185e-05],\n",
              "            [4.88216865e-05, 1.36441458e-05, 4.42305573e-05, 1.39480599e-05,\n",
              "             8.73788085e-05, 2.04487878e-05, 5.52479760e-05, 9.81733683e-06,\n",
              "             2.96912585e-05, 7.35653448e-05],\n",
              "            [8.18698609e-05, 2.94430829e-05, 5.76798993e-05, 9.52397568e-06,\n",
              "             9.28213340e-05, 3.32132331e-05, 9.34858617e-05, 5.04765521e-05,\n",
              "             3.85715648e-05, 6.61443337e-05],\n",
              "            [4.59968796e-05, 1.41538248e-05, 4.28688327e-05, 9.82664187e-06,\n",
              "             5.24086936e-05, 2.14342890e-05, 6.12306467e-05, 9.03348973e-06,\n",
              "             7.28484301e-05, 3.96403884e-05],\n",
              "            [1.32844962e-05, 2.03430282e-05, 9.35512217e-05, 6.93022448e-05,\n",
              "             2.14354950e-05, 4.62459429e-05, 2.44268449e-05, 6.24411041e-05,\n",
              "             1.35718419e-05, 1.79809776e-05]], dtype=float32)}}},\n",
              "  {'count': array(9370, dtype=int32)}],\n",
              " 'params': {'BatchNorm_0': {'bias': array([-0.9560376 , -1.0071353 , -0.5818822 , -0.36917838, -0.46915853,\n",
              "          -0.88612205, -0.7405656 , -0.70763654, -0.5655106 , -0.6128401 ,\n",
              "          -0.11745953, -0.67799485, -0.41351303, -0.6595287 , -0.04844438,\n",
              "          -0.96831995], dtype=float32),\n",
              "   'scale': array([0.68497014, 0.6631405 , 0.97644866, 1.370787  , 0.95040226,\n",
              "          1.5948869 , 1.4642572 , 1.7942637 , 0.6577792 , 1.3193667 ,\n",
              "          0.4464469 , 1.6518598 , 0.50821584, 1.4551423 , 0.38622802,\n",
              "          1.7144771 ], dtype=float32)},\n",
              "  'BatchNorm_1': {'bias': array([-1.0211192 , -0.60230476, -1.5056435 , -1.0937318 , -1.322513  ,\n",
              "          -0.73252356, -1.2277774 , -1.0907402 , -1.5822235 , -0.69986266,\n",
              "          -1.3024088 , -1.7028631 , -0.950533  , -0.8634483 , -1.5353473 ,\n",
              "          -0.8497833 , -0.8179694 , -0.634372  , -1.2587211 , -1.4026659 ,\n",
              "          -0.8564892 , -1.6817492 , -0.9177957 , -1.5516243 , -1.1748406 ,\n",
              "          -0.7706709 , -0.99771905, -1.9419168 , -1.485772  , -1.056794  ,\n",
              "          -0.8236432 , -1.2707542 ], dtype=float32),\n",
              "   'scale': array([2.430145 , 2.3404508, 3.2803457, 3.0762024, 3.0346034, 2.7043588,\n",
              "          2.9072576, 2.7754447, 3.4146767, 2.6467443, 3.0049117, 2.6088834,\n",
              "          2.631925 , 2.6358528, 2.9629335, 2.4700449, 2.9825718, 3.27846  ,\n",
              "          3.2108822, 2.7371256, 2.535984 , 3.0012107, 2.3586755, 3.0869825,\n",
              "          2.8108566, 2.9940963, 2.3774724, 2.8719318, 2.7653446, 3.018106 ,\n",
              "          2.470961 , 3.0917578], dtype=float32)},\n",
              "  'Conv_0': {'bias': array([-0.12124658,  0.21526994,  0.1908961 , -0.5423655 ,  0.26979086,\n",
              "           0.07708597, -0.11589558, -0.05794707, -0.14736499,  0.35204503,\n",
              "          -0.31011868,  0.02378077, -0.03783318, -0.54379666,  0.03324856,\n",
              "          -0.05634847], dtype=float32),\n",
              "   'kernel': array([[[[ 0.64782727, -0.17432922, -0.32592636,  0.3030778 ,\n",
              "              0.5027317 ,  0.36331734, -2.1424067 , -0.8756017 ,\n",
              "             -0.30372965,  0.37555608, -0.26650122, -0.4898407 ,\n",
              "              1.0499591 , -0.7504754 , -0.3856439 ,  0.01464792]],\n",
              "   \n",
              "           [[ 1.183019  ,  0.6119799 , -0.8591108 ,  0.13028064,\n",
              "              0.82164305,  0.00515884, -1.0350893 , -1.1092191 ,\n",
              "             -0.43393165,  0.31454465, -0.01155446, -0.5536822 ,\n",
              "              0.03462585,  0.05208077, -0.60849684, -0.6167568 ]],\n",
              "   \n",
              "           [[ 0.46887583,  0.5998866 , -0.5845997 ,  0.3373274 ,\n",
              "              0.17818029, -0.02189137, -0.73549926, -0.45316777,\n",
              "             -0.14059718, -0.21141341,  0.36026305, -1.2226975 ,\n",
              "             -0.02522681,  0.07589944, -0.48635525, -0.9094742 ]],\n",
              "   \n",
              "           [[ 0.45528427, -0.08278326, -0.2828693 ,  0.32486942,\n",
              "              0.32810628,  0.14958397, -0.7572427 , -0.08082341,\n",
              "             -0.09799496, -0.45568955,  0.5268058 , -0.9658188 ,\n",
              "             -0.1417978 ,  0.18414195, -0.47526273, -0.98870176]],\n",
              "   \n",
              "           [[ 0.44873825, -0.95132065,  0.17325735, -0.7484398 ,\n",
              "              0.5469649 , -0.11064401, -0.8692778 ,  0.18113689,\n",
              "              0.3813415 , -0.9855947 , -0.04184837, -0.936838  ,\n",
              "             -0.4041919 ,  0.44523397, -0.8982627 , -0.06242581]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.3741862 , -0.46547988,  0.6610525 ,  0.10502229,\n",
              "             -0.5632499 ,  0.2341168 ,  0.12268656, -1.1315424 ,\n",
              "             -0.4898315 ,  0.22385569,  0.29224738, -0.3522718 ,\n",
              "              0.8968037 , -0.5224669 ,  0.34029964,  0.17268294]],\n",
              "   \n",
              "           [[ 0.4182973 , -0.10219502, -0.03723501,  0.09170327,\n",
              "              0.00682544,  0.1264632 ,  0.05736255,  0.29092553,\n",
              "             -0.24981558,  0.2875749 ,  0.5841611 ,  0.19526215,\n",
              "             -0.20628901,  0.3245325 ,  0.28387132,  0.29951736]],\n",
              "   \n",
              "           [[-0.19569203,  0.64426136, -0.27980408,  0.3392184 ,\n",
              "              0.06060128,  0.3421444 , -0.5142252 ,  0.44709212,\n",
              "              0.15430954, -0.11411954,  0.2270908 ,  0.2533129 ,\n",
              "              0.12388808,  0.34018183,  0.15545177,  0.15517098]],\n",
              "   \n",
              "           [[-0.10587048,  0.959171  , -0.7590151 , -0.04560016,\n",
              "              0.07526003,  0.34264216, -0.43717295,  0.12401806,\n",
              "              0.6397484 , -0.05948701, -0.19528113,  0.35094705,\n",
              "             -0.25324437,  0.23163539,  0.21105687, -0.67388994]],\n",
              "   \n",
              "           [[-0.00313488, -0.26790944, -0.4992853 , -1.1800517 ,\n",
              "              0.3891673 , -0.26983485, -0.9770972 ,  0.28400323,\n",
              "              0.27568057, -0.95242786, -0.80285907, -1.2967141 ,\n",
              "             -0.3049377 ,  0.17232482,  0.17894973, -0.7871524 ]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.3622138 , -0.6195789 ,  0.28534827,  0.09232275,\n",
              "             -1.140187  , -0.02770001,  0.09881473, -1.2537775 ,\n",
              "             -0.64884156,  0.34204698,  0.59211916,  0.03229462,\n",
              "              0.36805335, -0.8303921 ,  0.6564378 , -0.36718452]],\n",
              "   \n",
              "           [[ 0.29326567, -0.3948941 ,  0.12969029,  0.31957862,\n",
              "             -0.3681527 ,  0.6208092 , -0.07086673,  0.06941307,\n",
              "              0.08393645,  0.38734853,  0.00215044,  0.10847469,\n",
              "              0.05025116,  0.02959838,  0.3564692 ,  0.24292332]],\n",
              "   \n",
              "           [[ 0.36730048, -0.09569304,  0.20913905, -0.02879266,\n",
              "             -0.23552403,  0.35337836,  0.04841204,  0.25249878,\n",
              "              0.3316112 ,  0.13759144, -0.66884375,  0.41324675,\n",
              "              0.21593249,  0.5701588 ,  0.3639    ,  0.18056788]],\n",
              "   \n",
              "           [[ 0.12749128,  0.6681947 ,  0.03136163, -1.5320091 ,\n",
              "             -0.09465533,  0.08848479, -0.13141288,  0.31774235,\n",
              "              0.8745519 , -1.0054078 , -0.75958616,  0.3911229 ,\n",
              "              0.16566731,  0.29488692,  0.4319292 , -0.22251338]],\n",
              "   \n",
              "           [[ 0.16294616,  0.2254109 , -0.75953156, -0.32326213,\n",
              "              0.21207787, -0.88187164, -0.8312757 ,  0.35001817,\n",
              "             -0.09286804, -0.15107964,  0.06492356, -1.2146746 ,\n",
              "             -0.14433125, -0.18002185,  0.7040861 , -1.3227665 ]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.14642547, -0.15623635, -0.19250678, -0.26548976,\n",
              "             -0.3109286 , -0.56687576,  0.26907375, -0.84600353,\n",
              "             -1.1520443 ,  0.54624534, -0.40708056, -0.01210441,\n",
              "              0.83606476, -0.83859223,  0.10819665, -0.04827357]],\n",
              "   \n",
              "           [[ 0.640795  , -0.7162036 ,  0.05487156, -0.4891346 ,\n",
              "             -0.43510848,  0.6498185 ,  0.16482961, -0.2608201 ,\n",
              "             -0.31787497,  0.24423534, -0.8147323 ,  0.01593917,\n",
              "              0.672853  , -1.5127211 ,  0.21020289,  0.39767766]],\n",
              "   \n",
              "           [[ 0.6132377 , -0.3795535 ,  0.54358864, -1.6840773 ,\n",
              "             -0.11883807,  0.30954224,  0.24701884,  0.21263517,\n",
              "              0.37575063, -0.05941815, -0.3314252 ,  0.34354508,\n",
              "              0.28724185, -1.7988123 ,  0.49855226,  0.2384835 ]],\n",
              "   \n",
              "           [[ 0.02211616,  0.4224189 ,  0.45817968, -0.12779358,\n",
              "              0.19232306, -0.23025335,  0.5845966 ,  0.19912316,\n",
              "              0.78784364, -1.1737162 ,  0.17383021,  0.40885365,\n",
              "             -0.286033  , -0.9435978 ,  0.34940186,  0.10727476]],\n",
              "   \n",
              "           [[-0.87691176,  0.64875495, -0.6517925 ,  0.09606696,\n",
              "             -0.11612179, -0.84104043, -0.09136637, -0.03662461,\n",
              "             -0.31110504,  0.19353914,  0.9552088 , -1.2929926 ,\n",
              "              0.20630983, -0.9480863 ,  0.28023595, -1.1748906 ]]],\n",
              "   \n",
              "   \n",
              "          [[[ 0.01952863,  0.47624204, -0.69386846, -1.4002706 ,\n",
              "             -0.40652215, -1.4287274 ,  0.67923427, -0.4199842 ,\n",
              "             -0.88274467,  0.33637762, -0.47203776, -0.05720491,\n",
              "              0.06159073,  0.21791393, -0.67689073,  0.54870194]],\n",
              "   \n",
              "           [[ 0.34692904,  0.32679164, -0.24702366, -0.49098933,\n",
              "             -0.01134646, -1.5313402 ,  0.17282984, -1.0425565 ,\n",
              "             -0.53817034, -0.7353689 ,  0.5224685 , -0.09396771,\n",
              "             -0.06279413,  0.3096483 , -0.628696  ,  0.5066678 ]],\n",
              "   \n",
              "           [[ 0.1429163 , -0.1959934 ,  0.34064713, -0.08202139,\n",
              "             -0.6971636 , -1.3674848 ,  0.11166536, -1.2122682 ,\n",
              "              0.08585708, -1.1984037 ,  0.3784086 , -0.2768002 ,\n",
              "              0.51255816,  0.23424909, -0.55089676,  0.38809606]],\n",
              "   \n",
              "           [[-0.43227205, -0.37278682,  0.37969545,  0.2752031 ,\n",
              "             -1.060796  , -0.7388941 ,  0.30958918, -0.9514509 ,\n",
              "              0.73317814, -0.68725777,  0.40185162, -0.96248984,\n",
              "              0.5688042 ,  0.2788079 , -0.70461935, -0.04007906]],\n",
              "   \n",
              "           [[-1.457102  ,  0.13370638, -0.06872683,  0.3552011 ,\n",
              "             -1.542418  , -1.0843675 ,  0.42679232, -0.73780274,\n",
              "              0.17243275,  0.08256423, -0.19578129, -0.6366778 ,\n",
              "              1.6170475 , -0.0634924 , -0.30125785, -1.2607358 ]]]],\n",
              "         dtype=float32)},\n",
              "  'Conv_1': {'bias': array([-0.02760017,  0.03291213, -0.01298245, -0.05640659,  0.13788472,\n",
              "          -0.0685158 ,  0.08101743,  0.04094861,  0.12840098,  0.00504573,\n",
              "          -0.00127399,  0.01530092,  0.01746071,  0.00490984, -0.08981805,\n",
              "          -0.00567709,  0.15777013,  0.03950912, -0.02732002, -0.17827073,\n",
              "           0.02474667,  0.22368672,  0.07854657,  0.08093248, -0.20384686,\n",
              "          -0.10459942,  0.06327984, -0.03061697,  0.02760119, -0.00180869,\n",
              "           0.0445908 , -0.19292884], dtype=float32),\n",
              "   'kernel': array([[[[ 1.93409380e-02,  4.37839657e-01, -6.66991711e-01, ...,\n",
              "             -7.23776102e-01, -7.39609301e-02, -1.27079532e-01],\n",
              "            [-4.34368640e-01, -4.28436428e-01,  6.91177607e-01, ...,\n",
              "             -3.70928973e-01,  8.47150981e-01, -3.09342027e-01],\n",
              "            [ 3.67305011e-01,  5.43924868e-01,  5.40041983e-01, ...,\n",
              "             -1.48139060e-01,  3.59097868e-01,  5.50767720e-01],\n",
              "            ...,\n",
              "            [-2.74424195e-01,  5.84714890e-01, -3.13240886e-01, ...,\n",
              "             -5.90839460e-02, -3.95846814e-02,  5.05587831e-02],\n",
              "            [ 2.30384350e-01,  1.48383558e-01,  5.67139924e-01, ...,\n",
              "             -5.81228316e-01, -3.44901383e-01,  3.63096297e-02],\n",
              "            [-7.53403455e-02,  8.57335985e-01,  3.11225295e-01, ...,\n",
              "              3.10860854e-02,  6.00185215e-01,  1.12143481e+00]],\n",
              "   \n",
              "           [[ 4.18420494e-01,  3.37611209e-03, -6.42247319e-01, ...,\n",
              "             -2.71654546e-01,  1.60484999e-01, -3.77947718e-01],\n",
              "            [ 1.83762074e-01, -7.67965138e-01,  9.68019009e-01, ...,\n",
              "             -6.18985176e-01,  6.90331042e-01, -7.48557985e-01],\n",
              "            [ 4.27974761e-01,  3.29997718e-01,  2.21355885e-01, ...,\n",
              "              2.28926390e-01,  3.59276950e-01,  5.42974710e-01],\n",
              "            ...,\n",
              "            [-1.78929463e-01,  5.07973790e-01,  7.06370967e-03, ...,\n",
              "             -2.50161951e-03, -2.02916324e-01,  6.57196492e-02],\n",
              "            [ 2.05996856e-01, -7.58901983e-03,  3.20194811e-01, ...,\n",
              "             -4.74830091e-01, -3.97116452e-01,  9.96367708e-02],\n",
              "            [-4.83255863e-01,  3.68727922e-01,  3.85358254e-03, ...,\n",
              "             -8.59798044e-02,  6.68098867e-01,  8.82670701e-01]],\n",
              "   \n",
              "           [[ 1.14338852e-01,  1.03844710e-01, -1.17405839e-01, ...,\n",
              "             -5.21285951e-01, -5.42017519e-01, -3.20152819e-01],\n",
              "            [-3.63802135e-01, -7.52815232e-02,  6.13839924e-01, ...,\n",
              "             -1.02392721e+00,  6.67807579e-01, -7.16309965e-01],\n",
              "            [ 2.75118828e-01,  8.20544437e-02, -2.93038934e-02, ...,\n",
              "              2.04075336e-01,  2.66809255e-01,  5.06089628e-01],\n",
              "            ...,\n",
              "            [-4.52428520e-01,  6.35264933e-01, -4.29104082e-02, ...,\n",
              "             -3.79726112e-01, -3.94368768e-01,  2.30973125e-01],\n",
              "            [ 9.04610679e-02, -2.10014358e-02,  6.54927492e-02, ...,\n",
              "             -6.18047655e-01, -6.28775239e-01,  1.92247406e-01],\n",
              "            [-1.38358280e-01,  4.10118103e-01, -2.51401305e-01, ...,\n",
              "             -5.99184372e-02, -1.89065993e-01,  7.02512681e-01]],\n",
              "   \n",
              "           [[ 2.34839041e-02, -4.65713084e-01, -2.09279120e-01, ...,\n",
              "             -4.02883887e-01, -9.18178320e-01, -4.51240331e-01],\n",
              "            [ 6.46275207e-02, -1.51290614e-02,  9.04622972e-01, ...,\n",
              "             -1.53694177e+00,  5.79644084e-01, -4.93099570e-01],\n",
              "            [ 5.53627133e-01,  3.77390444e-01, -9.67953652e-02, ...,\n",
              "             -6.41859174e-01, -2.49813363e-01,  3.36562008e-01],\n",
              "            ...,\n",
              "            [-7.54974484e-01,  6.17233753e-01, -1.74103856e-01, ...,\n",
              "             -3.25598419e-02, -4.94956583e-01,  3.15790921e-01],\n",
              "            [-3.39362631e-03, -3.07963967e-01, -1.93736658e-01, ...,\n",
              "             -3.43166262e-01, -2.39683792e-01,  4.31233913e-01],\n",
              "            [-1.57944083e-01,  6.90853655e-01, -5.27632758e-02, ...,\n",
              "             -2.21426979e-01, -1.89001799e-01,  7.99863338e-01]],\n",
              "   \n",
              "           [[-1.83288038e-01, -2.87166178e-01, -1.22892387e-01, ...,\n",
              "              1.04839988e-01, -4.47755933e-01, -5.55149555e-01],\n",
              "            [ 3.26161712e-01,  1.06454313e-01,  1.16801381e+00, ...,\n",
              "             -8.55114043e-01, -6.50674477e-03, -3.75847250e-01],\n",
              "            [-4.03704047e-02,  1.75555557e-01, -1.06764473e-01, ...,\n",
              "             -4.08542715e-02, -5.91866300e-02,  1.17597997e-01],\n",
              "            ...,\n",
              "            [ 4.97081503e-02, -5.27068265e-02, -9.21087742e-01, ...,\n",
              "              3.87819633e-02, -1.96879283e-01,  1.82981178e-01],\n",
              "            [ 4.57194507e-01, -1.00358367e+00, -7.23270655e-01, ...,\n",
              "             -8.18398535e-01, -3.57577473e-01,  8.85840654e-01],\n",
              "            [ 2.19396710e-01,  4.82242316e-01,  2.88781911e-01, ...,\n",
              "             -1.49027511e-01, -9.88098513e-03,  4.60391372e-01]]],\n",
              "   \n",
              "   \n",
              "          [[[ 2.95673639e-01,  4.76280719e-01, -3.84605467e-01, ...,\n",
              "             -4.20189649e-01, -1.27291337e-01,  4.38194752e-01],\n",
              "            [-4.96813923e-01,  9.02814493e-02,  4.01683688e-01, ...,\n",
              "             -6.22051477e-01,  1.15568316e+00, -1.30609170e-01],\n",
              "            [ 2.63060927e-01,  1.49966609e-02, -3.16068023e-01, ...,\n",
              "              1.19753042e-03,  2.25427181e-01,  7.41740882e-01],\n",
              "            ...,\n",
              "            [ 1.28917560e-01,  5.60272515e-01,  1.83866709e-01, ...,\n",
              "             -2.12920055e-01, -1.34316787e-01,  4.26955670e-01],\n",
              "            [-3.48477997e-02, -4.98075876e-03,  8.86525273e-01, ...,\n",
              "             -1.44057617e-01, -3.84804547e-01,  1.38254821e-01],\n",
              "            [-8.66834283e-01,  6.46388471e-01,  1.89818870e-02, ...,\n",
              "             -3.12964618e-01,  7.28416264e-01,  4.32728320e-01]],\n",
              "   \n",
              "           [[ 8.07334661e-01, -1.41607240e-01, -1.62304446e-01, ...,\n",
              "             -2.83012986e-01,  1.53815607e-02, -2.30119959e-01],\n",
              "            [-5.50592124e-01, -1.41082034e-01,  5.56094587e-01, ...,\n",
              "             -8.45838368e-01,  8.15022349e-01, -3.48706067e-01],\n",
              "            [ 7.19359815e-01, -2.89453477e-01, -4.68004614e-01, ...,\n",
              "              5.26906103e-02, -1.37358993e-01,  5.29533327e-01],\n",
              "            ...,\n",
              "            [-1.31362960e-01,  2.40817606e-01,  5.67120798e-02, ...,\n",
              "             -1.09539293e-01, -4.13109988e-01, -2.26823881e-01],\n",
              "            [-1.34351820e-01, -8.50936994e-02,  3.36807013e-01, ...,\n",
              "             -6.04969561e-01, -1.27915010e-01, -1.00876026e-01],\n",
              "            [-5.88893056e-01,  6.07379854e-01, -2.18716025e-01, ...,\n",
              "             -4.13172022e-02, -2.33688802e-01,  3.13475996e-01]],\n",
              "   \n",
              "           [[-1.45533442e-01,  6.46763891e-02,  1.17146097e-01, ...,\n",
              "             -6.15812242e-01, -4.80819613e-01,  3.28393579e-01],\n",
              "            [-3.35242242e-01, -1.36123911e-01,  3.29814285e-01, ...,\n",
              "             -8.10597658e-01,  4.38578278e-01, -2.11788282e-01],\n",
              "            [ 8.38776678e-03,  7.88218081e-02, -5.47903121e-01, ...,\n",
              "             -5.79102673e-02,  1.55635566e-01,  2.15576887e-01],\n",
              "            ...,\n",
              "            [-3.96276176e-01, -1.55959865e-02,  1.79747000e-01, ...,\n",
              "             -1.61173820e-01, -4.78789747e-01,  1.69808511e-02],\n",
              "            [ 5.65232374e-02,  9.54218730e-02,  7.10986480e-02, ...,\n",
              "             -2.87606061e-01,  1.57171525e-02, -7.33006075e-02],\n",
              "            [-4.26266342e-01, -6.17199019e-02, -1.12541187e+00, ...,\n",
              "             -2.55165607e-01, -5.01993239e-01,  2.58967012e-01]],\n",
              "   \n",
              "           [[ 2.95217574e-01, -2.02662975e-01,  1.29130945e-01, ...,\n",
              "             -4.69617248e-01, -4.58077937e-01,  2.09451869e-01],\n",
              "            [-1.22569621e-01,  2.91074932e-01,  1.81581303e-01, ...,\n",
              "             -1.64588666e+00,  4.87268209e-01, -2.89495200e-01],\n",
              "            [ 3.86361219e-02,  1.73146695e-01, -9.60025847e-01, ...,\n",
              "              3.10017139e-01,  3.39673348e-02,  3.67379993e-01],\n",
              "            ...,\n",
              "            [-4.73732471e-01, -7.03745723e-01, -1.79683000e-01, ...,\n",
              "             -3.83916616e-01, -1.14790881e+00,  3.26873899e-01],\n",
              "            [ 4.77913395e-02, -4.26383168e-02,  8.12185407e-02, ...,\n",
              "             -6.07563734e-01,  3.79176810e-02, -1.38891384e-01],\n",
              "            [-4.30928290e-01,  1.93584263e-01, -2.96655059e-01, ...,\n",
              "             -2.21259370e-01, -2.19555035e-01,  1.57217503e-01]],\n",
              "   \n",
              "           [[ 1.11104973e-01, -3.91808376e-02,  2.23053828e-01, ...,\n",
              "             -1.97398867e-02, -8.02847147e-02,  2.35897198e-01],\n",
              "            [ 3.10187578e-01,  4.35037076e-01,  1.22367427e-01, ...,\n",
              "             -7.08229721e-01,  1.70117561e-02, -2.39615798e-01],\n",
              "            [-1.90065682e-01,  2.89076447e-01, -8.93187582e-01, ...,\n",
              "              3.50249618e-01,  4.46459860e-01,  2.14253560e-01],\n",
              "            ...,\n",
              "            [-6.18785471e-02, -5.01178741e-01, -7.43322015e-01, ...,\n",
              "             -1.08384156e+00, -9.77658868e-01,  2.61147946e-01],\n",
              "            [ 6.16663098e-01, -2.43775472e-01, -4.44017977e-01, ...,\n",
              "             -1.04191804e+00, -7.97489099e-03,  3.74946415e-01],\n",
              "            [-1.10652469e-01, -3.99445370e-02,  1.43511266e-01, ...,\n",
              "             -2.75350630e-01, -3.67354125e-01,  3.94483835e-01]]],\n",
              "   \n",
              "   \n",
              "          [[[ 1.67170893e-02,  9.53174829e-02, -1.07688136e-01, ...,\n",
              "             -3.86543661e-01,  6.91875741e-02,  1.08372048e-01],\n",
              "            [-1.27954647e-01, -2.88562596e-01,  1.59105763e-01, ...,\n",
              "             -5.95474899e-01,  5.63416004e-01,  4.71698642e-01],\n",
              "            [ 6.85901582e-01, -6.02608323e-01, -2.71147937e-01, ...,\n",
              "             -7.94915631e-02, -2.49939442e-01,  3.08073729e-01],\n",
              "            ...,\n",
              "            [ 2.70253867e-01,  3.05164069e-01,  2.09265083e-01, ...,\n",
              "              1.62355527e-02, -2.05540136e-01,  8.78459990e-01],\n",
              "            [-4.48021054e-01,  1.35036007e-01,  2.64051527e-01, ...,\n",
              "             -4.88250136e-01, -1.40987068e-01, -1.37653843e-01],\n",
              "            [ 4.22791809e-01,  3.31090003e-01,  5.62108994e-01, ...,\n",
              "              1.55920923e-01, -2.43686318e-01, -2.65344858e-01]],\n",
              "   \n",
              "           [[ 2.76998878e-01,  3.67330909e-02, -1.23030216e-01, ...,\n",
              "              3.02734673e-01,  2.27175906e-01, -5.18199265e-01],\n",
              "            [-2.51546830e-01, -6.44984394e-02,  3.08656335e-01, ...,\n",
              "             -6.78127766e-01,  6.84381962e-01,  3.83782685e-02],\n",
              "            [ 3.26370239e-01, -6.05127990e-01, -4.75691378e-01, ...,\n",
              "             -1.74719468e-01,  2.04801857e-02, -3.21633279e-01],\n",
              "            ...,\n",
              "            [ 1.64103001e-01,  1.11707009e-01, -1.16132507e-02, ...,\n",
              "             -6.29961342e-02, -8.91497791e-01,  2.73356855e-01],\n",
              "            [-1.90150291e-01,  1.29223699e-02,  1.67527333e-01, ...,\n",
              "             -3.00577939e-01, -3.28402281e-01, -2.41968080e-01],\n",
              "            [ 3.26371759e-01,  2.60777354e-01, -2.51248538e-01, ...,\n",
              "              1.85524762e-01,  1.14957064e-01,  2.09143683e-01]],\n",
              "   \n",
              "           [[ 1.93426952e-01, -4.06778641e-02,  6.10352755e-01, ...,\n",
              "              4.70559485e-02,  7.14075044e-02,  2.21819952e-01],\n",
              "            [-2.44016245e-01, -1.55626267e-01, -1.71755165e-01, ...,\n",
              "             -1.00538266e+00,  3.56686473e-01,  1.71289966e-01],\n",
              "            [ 1.84422970e-01, -3.39431942e-01, -6.47951543e-01, ...,\n",
              "              6.42830059e-02, -4.94276375e-01,  1.53645948e-01],\n",
              "            ...,\n",
              "            [ 1.58788376e-02, -3.96503896e-01,  1.57040626e-01, ...,\n",
              "             -1.88980043e-01, -1.06870818e+00,  4.56516892e-02],\n",
              "            [-1.47628441e-01, -2.35158503e-01,  2.36288950e-01, ...,\n",
              "             -5.80847740e-01, -2.47601241e-01,  8.36751796e-03],\n",
              "            [ 6.62571341e-02, -1.52890369e-01, -1.07878637e+00, ...,\n",
              "              2.13919785e-02, -3.89065832e-01,  4.70739603e-02]],\n",
              "   \n",
              "           [[-1.44107416e-01,  1.65748999e-01,  4.89296049e-01, ...,\n",
              "             -2.35444799e-01,  1.72903046e-01,  1.81271195e-01],\n",
              "            [-1.78125069e-01,  2.62436837e-01, -4.75191355e-01, ...,\n",
              "             -8.51388097e-01, -3.44089344e-02, -3.57372105e-01],\n",
              "            [-8.19619894e-02, -5.51383533e-02, -7.84157097e-01, ...,\n",
              "              7.33141527e-02, -3.97008061e-01, -2.13852420e-01],\n",
              "            ...,\n",
              "            [-3.05012446e-02, -5.95055521e-01, -4.56760898e-02, ...,\n",
              "             -7.13809207e-02, -7.92183518e-01,  1.31211743e-01],\n",
              "            [-2.79496443e-02, -2.68506408e-01,  1.82104677e-01, ...,\n",
              "             -5.25642455e-01, -9.56046134e-02, -1.90744773e-01],\n",
              "            [ 1.10483758e-01, -2.15391606e-01, -7.40097165e-01, ...,\n",
              "             -5.38899973e-02, -2.00448975e-01,  6.40579611e-02]],\n",
              "   \n",
              "           [[-2.91459531e-01, -4.54567671e-01,  7.27930605e-01, ...,\n",
              "              3.69140714e-01,  2.89630115e-01,  9.12751704e-02],\n",
              "            [-1.13908269e-01, -1.06843956e-01, -3.85068566e-01, ...,\n",
              "             -3.79876107e-01,  3.04321982e-02, -2.68946618e-01],\n",
              "            [ 1.30098835e-01,  1.50853708e-01, -3.76450688e-01, ...,\n",
              "              2.34347656e-01, -4.56552543e-02, -4.14679348e-01],\n",
              "            ...,\n",
              "            [ 3.12749743e-01, -3.88258040e-01, -8.31063330e-01, ...,\n",
              "             -2.86966562e-01, -2.87693262e-01, -3.34305525e-01],\n",
              "            [ 3.82123500e-01, -1.43026233e-01, -2.53730297e-01, ...,\n",
              "             -1.15262151e+00, -3.88139188e-02, -2.07122475e-01],\n",
              "            [ 6.74387142e-02,  2.12765317e-02, -5.63384414e-01, ...,\n",
              "              2.27069601e-01,  7.03759342e-02,  5.36947884e-02]]],\n",
              "   \n",
              "   \n",
              "          [[[-2.79833525e-01,  2.52159357e-01, -4.98240978e-01, ...,\n",
              "             -4.77380782e-01,  6.35032415e-01, -1.51022539e-01],\n",
              "            [-6.49457946e-02, -9.57076490e-01, -3.31931119e-03, ...,\n",
              "             -3.01123559e-01,  1.93327382e-01, -2.42650747e-01],\n",
              "            [ 4.45620239e-01, -3.30278248e-01, -1.07769109e-01, ...,\n",
              "             -6.00389540e-02, -4.48277146e-01,  1.58047974e-01],\n",
              "            ...,\n",
              "            [-2.82116123e-02,  2.63626546e-01, -2.22636223e-01, ...,\n",
              "             -4.79445457e-01, -1.05214167e+00,  8.79570127e-01],\n",
              "            [-5.16280293e-01,  1.56490039e-02, -5.22186160e-02, ...,\n",
              "             -3.67123276e-01, -2.76900083e-01, -4.01629433e-02],\n",
              "            [ 4.34616357e-01,  4.80685890e-01, -6.10900283e-01, ...,\n",
              "             -2.15198606e-01, -2.45255485e-01, -2.44225726e-01]],\n",
              "   \n",
              "           [[ 1.93502635e-01,  1.91306025e-01,  1.74866006e-01, ...,\n",
              "              5.47661304e-01,  3.36911261e-01, -1.99648887e-01],\n",
              "            [-4.83622774e-02, -9.94523048e-01, -3.53868194e-02, ...,\n",
              "             -8.01596761e-01,  5.34495056e-01, -7.66399086e-01],\n",
              "            [ 2.07049459e-01, -3.62225711e-01, -4.49925244e-01, ...,\n",
              "             -1.50531158e-01, -2.10322380e-01,  1.12480344e-02],\n",
              "            ...,\n",
              "            [ 1.90320641e-01,  3.99274707e-01, -5.21654263e-02, ...,\n",
              "             -3.60407919e-01, -9.56774414e-01,  7.03654706e-01],\n",
              "            [-5.15239239e-01, -3.11197750e-02,  1.04926810e-01, ...,\n",
              "             -3.26606542e-01, -3.19021344e-01, -2.52619982e-01],\n",
              "            [ 6.58732235e-01,  4.30997461e-01, -7.75587261e-01, ...,\n",
              "             -2.25855187e-01, -3.75657141e-01, -7.44265676e-01]],\n",
              "   \n",
              "           [[-7.33833015e-03,  3.64060290e-02,  2.34130815e-01, ...,\n",
              "              9.60319862e-02, -5.16445823e-02,  3.76931012e-01],\n",
              "            [-8.50950256e-02, -5.49605191e-01, -2.32021108e-01, ...,\n",
              "             -7.57622659e-01,  2.65451521e-01, -1.10150866e-01],\n",
              "            [ 2.25077748e-01, -1.02580972e-02, -5.40379584e-01, ...,\n",
              "             -4.68587905e-01, -4.11452234e-01, -1.70570239e-01],\n",
              "            ...,\n",
              "            [ 6.09730482e-02, -2.86603514e-02, -2.03581169e-01, ...,\n",
              "             -2.29631543e-01, -6.63585424e-01,  3.12582403e-01],\n",
              "            [-3.45134079e-01, -2.81464726e-01,  4.65398550e-01, ...,\n",
              "             -2.25250453e-01, -4.61396426e-01,  2.87725329e-01],\n",
              "            [ 8.72827768e-01,  2.06782818e-02, -8.18163753e-01, ...,\n",
              "             -6.71546236e-02, -4.57488209e-01, -8.80533159e-02]],\n",
              "   \n",
              "           [[-5.82750142e-03, -2.41864145e-01,  3.18555713e-01, ...,\n",
              "             -2.28457808e-01, -4.56747860e-02,  1.31476581e-01],\n",
              "            [-1.29174083e-01, -5.49212575e-01, -7.85232410e-02, ...,\n",
              "             -5.91924489e-01,  4.60150570e-01, -5.01730144e-01],\n",
              "            [-1.41991675e-01,  4.75257158e-01, -7.89966941e-01, ...,\n",
              "              2.84605205e-01, -9.02540535e-02, -9.78253409e-02],\n",
              "            ...,\n",
              "            [ 1.60378069e-01, -6.96575791e-02, -4.98365074e-01, ...,\n",
              "              2.66323145e-03, -6.47430599e-01,  2.84479886e-01],\n",
              "            [-2.34845638e-01, -2.62286305e-01,  1.66502222e-01, ...,\n",
              "             -2.63371438e-01, -2.63272285e-01,  3.09270956e-02],\n",
              "            [ 6.57959640e-01,  4.86605130e-02, -4.48151708e-01, ...,\n",
              "             -2.42799358e-03, -7.18423307e-01, -4.82602119e-01]],\n",
              "   \n",
              "           [[-3.63265485e-01, -2.88323760e-01,  4.95429933e-01, ...,\n",
              "             -1.99220907e-02, -1.70863062e-01,  4.59049314e-01],\n",
              "            [ 2.81812437e-02,  1.18758589e-01, -1.17806651e-01, ...,\n",
              "             -2.36826658e-01,  5.49137712e-01, -3.19427639e-01],\n",
              "            [ 2.76980072e-01,  4.95717287e-01, -3.48231018e-01, ...,\n",
              "              1.16651520e-01, -1.79123387e-01, -9.25059095e-02],\n",
              "            ...,\n",
              "            [ 2.42278174e-01,  8.62821862e-02, -6.85412467e-01, ...,\n",
              "              4.26259302e-02, -1.78116828e-01,  4.34627205e-01],\n",
              "            [ 2.00019613e-01, -9.43134502e-02, -6.34221077e-01, ...,\n",
              "              5.21364361e-02,  4.16678824e-02, -1.87922865e-01],\n",
              "            [ 1.19303513e+00, -5.10634109e-02, -2.41158865e-02, ...,\n",
              "              4.03079301e-01, -2.16853041e-02, -1.36434436e-01]]],\n",
              "   \n",
              "   \n",
              "          [[[ 2.54789472e-01,  5.15871584e-01,  4.75436568e-01, ...,\n",
              "             -2.16773987e-01,  5.65230131e-01, -1.65312678e-01],\n",
              "            [-2.32921522e-02, -2.82801300e-01, -5.23564041e-01, ...,\n",
              "             -3.57360065e-01, -1.48165256e-01, -1.10222769e+00],\n",
              "            [ 1.86956987e-01,  1.14827631e-02, -4.12610471e-01, ...,\n",
              "              3.59022081e-01, -8.80226433e-01, -9.59637821e-01],\n",
              "            ...,\n",
              "            [ 3.49205226e-01,  5.18787690e-02, -6.48970842e-01, ...,\n",
              "             -6.80925697e-02, -9.71033871e-01,  3.71053517e-01],\n",
              "            [ 1.06672093e-01,  4.56987411e-01, -5.50073385e-01, ...,\n",
              "             -3.83239277e-02, -6.82369620e-02, -1.29782349e-01],\n",
              "            [ 6.06342852e-01,  1.09749007e+00, -1.51097894e+00, ...,\n",
              "             -3.93215939e-02, -6.40115380e-01, -1.23985827e+00]],\n",
              "   \n",
              "           [[-5.34689426e-02, -7.45778382e-02, -6.25535697e-02, ...,\n",
              "              8.12848151e-01, -1.10639811e-01, -9.78362784e-02],\n",
              "            [-2.13911220e-01, -4.01134133e-01, -3.86692643e-01, ...,\n",
              "             -5.14686704e-01,  5.78713775e-01, -1.34710467e+00],\n",
              "            [ 4.46226358e-01, -1.72637329e-01, -5.44561565e-01, ...,\n",
              "             -1.14198053e+00, -6.33693755e-01, -2.46056929e-01],\n",
              "            ...,\n",
              "            [-1.77195206e-01,  5.78568220e-01, -3.06802273e-01, ...,\n",
              "             -3.48290890e-01, -4.47919846e-01,  9.87048090e-01],\n",
              "            [ 2.63735559e-02,  2.80437112e-01, -2.08486617e-01, ...,\n",
              "              3.53254117e-02, -1.59646019e-01,  5.54617643e-01],\n",
              "            [ 9.85325873e-01,  5.87071717e-01, -4.55872148e-01, ...,\n",
              "             -7.57255793e-01, -3.70901585e-01, -3.08952779e-01]],\n",
              "   \n",
              "           [[ 1.98557571e-01, -5.79725616e-02,  1.40944839e-01, ...,\n",
              "              2.16377288e-01, -9.22867805e-02,  3.97195071e-01],\n",
              "            [ 2.19786149e-02, -3.89066264e-02, -1.36507124e-01, ...,\n",
              "             -1.15258408e+00,  3.30686122e-01, -1.26942182e+00],\n",
              "            [ 3.29772830e-01,  3.78849596e-01,  5.41092753e-02, ...,\n",
              "             -1.36918277e-01, -4.15365279e-01,  2.38226742e-01],\n",
              "            ...,\n",
              "            [-3.73573452e-02,  1.94232255e-01,  1.62928045e-01, ...,\n",
              "             -6.87490255e-02, -2.26325721e-01,  5.54815590e-01],\n",
              "            [-5.53735197e-01,  3.62216741e-01, -1.61763415e-01, ...,\n",
              "              2.44010687e-02, -2.17370406e-01,  8.68129015e-01],\n",
              "            [ 8.08752179e-01,  2.56485850e-01, -3.90894443e-01, ...,\n",
              "             -8.84532571e-01, -5.27577214e-02,  1.45421743e-01]],\n",
              "   \n",
              "           [[-1.98046014e-01,  2.05180086e-02,  1.28753893e-02, ...,\n",
              "              4.01559144e-01, -2.74763227e-01, -1.01894677e-01],\n",
              "            [-4.07268740e-02,  1.44918665e-01, -2.72733212e-01, ...,\n",
              "             -8.11241329e-01,  3.62263083e-01, -6.21156335e-01],\n",
              "            [ 3.88089716e-01,  6.06497109e-01, -3.34994555e-01, ...,\n",
              "             -3.16238344e-01, -3.33681583e-01,  6.36197805e-01],\n",
              "            ...,\n",
              "            [-5.13920009e-01,  2.75131643e-01, -3.48353207e-01, ...,\n",
              "              1.63679525e-01, -1.73923224e-01,  1.10774326e+00],\n",
              "            [-7.56895006e-01,  2.76660860e-01, -5.82089186e-01, ...,\n",
              "              2.47609466e-01,  1.20833799e-01,  5.00535548e-01],\n",
              "            [ 6.73231602e-01, -1.86233252e-01,  2.73636281e-01, ...,\n",
              "             -1.90438673e-01, -4.74688172e-01,  5.33869505e-01]],\n",
              "   \n",
              "           [[-3.15676890e-02, -7.73656368e-01,  1.59730062e-01, ...,\n",
              "             -2.49515429e-01, -4.04987305e-01, -2.36988917e-01],\n",
              "            [ 2.68811006e-02,  5.72983146e-01, -2.77341038e-01, ...,\n",
              "             -3.64129245e-01,  7.14195549e-01,  1.77737931e-03],\n",
              "            [ 5.64896405e-01,  6.09612167e-01,  1.37356773e-01, ...,\n",
              "             -9.77230594e-02, -4.81889337e-01,  6.58134162e-01],\n",
              "            ...,\n",
              "            [-6.03392780e-01, -9.57781002e-02, -6.00157738e-01, ...,\n",
              "              4.11736399e-01,  3.52460653e-01,  1.18755233e+00],\n",
              "            [-1.07696688e+00,  7.77849495e-01, -4.79456961e-01, ...,\n",
              "              3.18215102e-01,  8.16803336e-01, -6.30724579e-02],\n",
              "            [ 6.07903898e-01, -8.68975103e-01,  5.83210945e-01, ...,\n",
              "              2.82276943e-02, -4.41761643e-01,  5.42042911e-01]]]],\n",
              "         dtype=float32)},\n",
              "  'Dense_0': {'bias': array([ 0.09042518,  0.9883207 , -0.5461252 ,  0.5591024 , -1.014932  ,\n",
              "           0.03685181, -0.5483948 ,  0.17893407,  0.09069423,  0.42663318],\n",
              "         dtype=float32),\n",
              "   'kernel': array([[ 1.0585502 , -1.3956808 ,  0.34636924, -0.9243253 ,  0.6699518 ,\n",
              "            0.3478436 , -1.3132898 ,  1.0497571 , -1.5761775 , -0.09053539],\n",
              "          [ 0.32554245,  0.65131235,  0.49186245,  0.80666155, -0.4104941 ,\n",
              "           -0.11780734, -0.0439073 , -1.242046  , -0.13078369, -2.0075583 ],\n",
              "          [-0.03930703,  0.42648882, -0.15188728, -0.28224364,  0.8943288 ,\n",
              "           -3.0429964 , -2.3474329 ,  1.0505046 ,  0.55783266,  0.8439393 ],\n",
              "          [-1.8106714 , -1.7483554 , -1.0383319 ,  0.70590925,  0.8624324 ,\n",
              "            0.6231594 , -1.6662395 ,  0.4543369 ,  0.55865437,  0.56678826],\n",
              "          [-0.8799365 ,  0.5406505 ,  1.0845698 ,  1.3094565 , -0.76142496,\n",
              "           -1.0557452 , -1.8899986 ,  1.1710124 , -0.07026887, -1.4067062 ],\n",
              "          [ 0.6540665 ,  0.6292495 , -2.1739986 , -1.8979679 , -0.40589637,\n",
              "            0.22870934,  0.25991988,  0.50809854,  0.76795137, -0.14191902],\n",
              "          [ 0.6679607 , -1.9748625 ,  0.98862076,  0.6099043 , -2.3751013 ,\n",
              "           -0.23683429, -1.6925834 ,  0.9011968 , -0.02695877,  0.33320114],\n",
              "          [ 0.2799459 ,  0.9547268 , -0.74576557,  1.2307259 , -0.01783373,\n",
              "            0.49792215, -0.47266635, -1.6590598 ,  0.7549098 , -2.2042775 ],\n",
              "          [ 0.5997754 , -2.9535203 ,  0.30706394,  0.2468266 , -2.768543  ,\n",
              "            0.67560655,  0.9843934 , -1.5188024 ,  0.6699628 ,  0.4690546 ],\n",
              "          [ 0.6613665 ,  0.16088574, -0.3168095 ,  0.8122926 , -0.8657408 ,\n",
              "            1.0023273 , -0.35324186,  0.24720134, -2.308821  , -0.20775384],\n",
              "          [ 0.7620061 , -1.4445834 ,  0.67909336,  0.9022798 , -1.9370795 ,\n",
              "           -1.1674767 , -1.6605536 ,  0.8177995 ,  0.22709715,  0.7079008 ],\n",
              "          [-1.2467344 , -1.6212808 ,  0.5991567 , -1.7618549 ,  1.0668769 ,\n",
              "            1.0039266 ,  0.38810116,  0.22538279,  0.25290293, -0.8203572 ],\n",
              "          [ 1.2433797 ,  0.29689768, -0.76886374, -0.8980173 ,  0.03523094,\n",
              "           -0.52442664, -1.575716  ,  0.27575853, -0.99056035,  0.6911181 ],\n",
              "          [ 0.7539889 ,  0.91173303,  0.77747923, -1.0377945 ,  0.29228503,\n",
              "           -1.0293409 ,  0.2873672 ,  0.11681654,  0.3716995 , -2.1477218 ],\n",
              "          [-0.5412877 , -0.958979  , -2.1284919 ,  1.5140238 ,  0.648908  ,\n",
              "            1.0241888 , -1.1419033 , -1.5348314 ,  0.8750265 , -0.26669705],\n",
              "          [-1.1223868 ,  1.1346172 , -0.3250557 , -1.2305187 , -0.07616247,\n",
              "            0.7045931 ,  1.035662  ,  0.7788254 , -0.02310352, -1.1854041 ],\n",
              "          [ 0.23997088,  0.8244198 , -1.1998576 , -2.548607  ,  0.74460965,\n",
              "           -0.10683431,  0.6283817 , -0.22918731, -0.7731416 , -0.38630196],\n",
              "          [-2.2417324 , -0.3725433 ,  0.23350902,  0.47917002,  0.64841974,\n",
              "            0.13434617, -2.549328  ,  0.6658044 ,  0.3580458 ,  0.59050244],\n",
              "          [-0.7369954 ,  0.5043032 ,  0.21876381, -1.0767151 ,  0.6187672 ,\n",
              "           -1.5783373 , -0.19088574,  0.8125462 , -2.3482482 ,  0.89920163],\n",
              "          [ 0.34502566, -2.5694413 ,  1.138908  , -0.60091656,  0.4023234 ,\n",
              "           -1.1248877 ,  0.6033688 , -2.1898944 ,  0.9333728 ,  0.0617527 ],\n",
              "          [-2.3115706 ,  0.87495077,  0.21941987,  0.49158254, -1.1033119 ,\n",
              "           -0.00610985,  0.15623653, -0.12987779, -0.0570875 ,  0.77911097],\n",
              "          [-0.01991823, -2.3508358 , -1.8154157 , -1.3452514 , -0.6187231 ,\n",
              "            1.1783906 ,  1.0292022 , -0.54034513,  0.7649426 ,  0.8226675 ],\n",
              "          [ 0.7452354 ,  0.07045756,  0.6198479 ,  1.0426222 , -0.6314761 ,\n",
              "            0.12317616,  0.7753757 , -1.6244898 , -1.0560359 , -0.5074863 ],\n",
              "          [ 0.63647234, -2.0846255 , -0.21029124, -1.2813647 ,  0.9048618 ,\n",
              "           -1.1205572 ,  1.1520034 , -2.8156273 ,  0.30465096,  0.8744729 ],\n",
              "          [-0.32400385, -2.8790267 , -0.82452756,  0.42916846, -1.2816621 ,\n",
              "            0.97144747,  1.049633  , -2.085277  ,  0.74591506,  0.1364304 ],\n",
              "          [-1.6981544 , -0.27329352, -0.48453346,  0.62416697,  0.9788648 ,\n",
              "            0.42749774, -2.0492291 ,  0.44086713, -0.8616808 ,  0.80357397],\n",
              "          [ 0.55568004,  0.07409288,  1.2333273 , -0.81862557, -1.4973769 ,\n",
              "            1.0674679 , -0.5029427 , -0.9597362 , -0.08249622, -0.7234108 ],\n",
              "          [-1.6893413 , -1.5363307 ,  1.4890915 , -0.11975233,  0.883288  ,\n",
              "           -1.7834784 ,  0.85656303,  0.5533208 ,  1.0144161 , -1.2230774 ],\n",
              "          [ 0.44524628, -0.32470143,  0.29902253, -1.467759  ,  1.2573394 ,\n",
              "           -0.97124434,  0.8314855 , -1.8486851 , -1.0785341 ,  0.94371533],\n",
              "          [ 0.6512383 ,  0.35397965, -0.41253412, -3.0363603 ,  0.6052733 ,\n",
              "           -0.6348416 ,  0.5386722 ,  0.40046528, -0.94667107, -0.15558401],\n",
              "          [ 0.07591704,  0.19094907,  0.63252354, -2.176439  ,  0.3965976 ,\n",
              "           -0.7874573 ,  0.6791022 , -2.0750954 ,  0.74295837, -0.20663945],\n",
              "          [-1.9893469 , -0.42655385,  0.7106501 ,  0.76565945, -0.3628365 ,\n",
              "            0.6943133 ,  0.85113996,  0.20940991, -1.2786822 , -1.9053285 ]],\n",
              "         dtype=float32)}},\n",
              " 'step': array(9370, dtype=int32)}"
            ]
          },
          "metadata": {},
          "execution_count": 19
        }
      ]
    }
  ]
}