텐서보드 코드 분리
This commit is contained in:
@@ -1631,48 +1631,11 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 28,
|
||||
"execution_count": null,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"# 크롬 브라우저는 확인되었지만 다른 브라우저에서는 작동되지 않을 수 있습니다.\n",
|
||||
"\n",
|
||||
"from IPython.display import clear_output, Image, display, HTML\n",
|
||||
"\n",
|
||||
"def strip_consts(graph_def, max_const_size=32):\n",
|
||||
" \"\"\"Strip large constant values from graph_def.\"\"\"\n",
|
||||
" strip_def = tf.GraphDef()\n",
|
||||
" for n0 in graph_def.node:\n",
|
||||
" n = strip_def.node.add() \n",
|
||||
" n.MergeFrom(n0)\n",
|
||||
" if n.op == 'Const':\n",
|
||||
" tensor = n.attr['value'].tensor\n",
|
||||
" size = len(tensor.tensor_content)\n",
|
||||
" if size > max_const_size:\n",
|
||||
" tensor.tensor_content = b\"<stripped %d bytes>\"%size\n",
|
||||
" return strip_def\n",
|
||||
"\n",
|
||||
"def show_graph(graph_def, max_const_size=32):\n",
|
||||
" \"\"\"Visualize TensorFlow graph.\"\"\"\n",
|
||||
" if hasattr(graph_def, 'as_graph_def'):\n",
|
||||
" graph_def = graph_def.as_graph_def()\n",
|
||||
" strip_def = strip_consts(graph_def, max_const_size=max_const_size)\n",
|
||||
" code = \"\"\"\n",
|
||||
" <script>\n",
|
||||
" function load() {{\n",
|
||||
" document.getElementById(\"{id}\").pbtxt = {data};\n",
|
||||
" }}\n",
|
||||
" </script>\n",
|
||||
" <link rel=\"import\" href=\"https://tensorboard.appspot.com/tf-graph-basic.build.html\" onload=load()>\n",
|
||||
" <div style=\"height:600px\">\n",
|
||||
" <tf-graph-basic id=\"{id}\"></tf-graph-basic>\n",
|
||||
" </div>\n",
|
||||
" \"\"\".format(data=repr(str(strip_def)), id='graph'+str(np.random.rand()))\n",
|
||||
"\n",
|
||||
" iframe = \"\"\"\n",
|
||||
" <iframe seamless style=\"width:1200px;height:620px;border:0\" srcdoc=\"{}\"></iframe>\n",
|
||||
" \"\"\".format(code.replace('\"', '"'))\n",
|
||||
" display(HTML(iframe))"
|
||||
"from tensorflow_graph_in_jupyter import show_graph"
|
||||
]
|
||||
},
|
||||
{
|
||||
@@ -2237,7 +2200,7 @@
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.5.5"
|
||||
"version": "3.6.5"
|
||||
},
|
||||
"nav_menu": {
|
||||
"height": "264px",
|
||||
|
||||
@@ -312,46 +312,11 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 14,
|
||||
"execution_count": null,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"from IPython.display import clear_output, Image, display, HTML\n",
|
||||
"\n",
|
||||
"def strip_consts(graph_def, max_const_size=32):\n",
|
||||
" \"\"\"Strip large constant values from graph_def.\"\"\"\n",
|
||||
" strip_def = tf.GraphDef()\n",
|
||||
" for n0 in graph_def.node:\n",
|
||||
" n = strip_def.node.add() \n",
|
||||
" n.MergeFrom(n0)\n",
|
||||
" if n.op == 'Const':\n",
|
||||
" tensor = n.attr['value'].tensor\n",
|
||||
" size = len(tensor.tensor_content)\n",
|
||||
" if size > max_const_size:\n",
|
||||
" tensor.tensor_content = \"b<stripped %d bytes>\"%size\n",
|
||||
" return strip_def\n",
|
||||
"\n",
|
||||
"def show_graph(graph_def, max_const_size=32):\n",
|
||||
" \"\"\"Visualize TensorFlow graph.\"\"\"\n",
|
||||
" if hasattr(graph_def, 'as_graph_def'):\n",
|
||||
" graph_def = graph_def.as_graph_def()\n",
|
||||
" strip_def = strip_consts(graph_def, max_const_size=max_const_size)\n",
|
||||
" code = \"\"\"\n",
|
||||
" <script>\n",
|
||||
" function load() {{\n",
|
||||
" document.getElementById(\"{id}\").pbtxt = {data};\n",
|
||||
" }}\n",
|
||||
" </script>\n",
|
||||
" <link rel=\"import\" href=\"https://tensorboard.appspot.com/tf-graph-basic.build.html\" onload=load()>\n",
|
||||
" <div style=\"height:600px\">\n",
|
||||
" <tf-graph-basic id=\"{id}\"></tf-graph-basic>\n",
|
||||
" </div>\n",
|
||||
" \"\"\".format(data=repr(str(strip_def)), id='graph'+str(np.random.rand()))\n",
|
||||
"\n",
|
||||
" iframe = \"\"\"\n",
|
||||
" <iframe seamless style=\"width:1200px;height:620px;border:0\" srcdoc=\"{}\"></iframe>\n",
|
||||
" \"\"\".format(code.replace('\"', '"'))\n",
|
||||
" display(HTML(iframe))"
|
||||
"from tensorflow_graph_in_jupyter import show_graph"
|
||||
]
|
||||
},
|
||||
{
|
||||
@@ -3490,7 +3455,7 @@
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.5.5"
|
||||
"version": "3.6.5"
|
||||
},
|
||||
"nav_menu": {},
|
||||
"toc": {
|
||||
|
||||
Reference in New Issue
Block a user