텐서보드 코드 분리
This commit is contained in:
@@ -1631,48 +1631,11 @@
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"execution_count": 28,
|
"execution_count": null,
|
||||||
"metadata": {},
|
"metadata": {},
|
||||||
"outputs": [],
|
"outputs": [],
|
||||||
"source": [
|
"source": [
|
||||||
"# 크롬 브라우저는 확인되었지만 다른 브라우저에서는 작동되지 않을 수 있습니다.\n",
|
"from tensorflow_graph_in_jupyter import show_graph"
|
||||||
"\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))"
|
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -2237,7 +2200,7 @@
|
|||||||
"name": "python",
|
"name": "python",
|
||||||
"nbconvert_exporter": "python",
|
"nbconvert_exporter": "python",
|
||||||
"pygments_lexer": "ipython3",
|
"pygments_lexer": "ipython3",
|
||||||
"version": "3.5.5"
|
"version": "3.6.5"
|
||||||
},
|
},
|
||||||
"nav_menu": {
|
"nav_menu": {
|
||||||
"height": "264px",
|
"height": "264px",
|
||||||
|
|||||||
@@ -312,46 +312,11 @@
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"execution_count": 14,
|
"execution_count": null,
|
||||||
"metadata": {},
|
"metadata": {},
|
||||||
"outputs": [],
|
"outputs": [],
|
||||||
"source": [
|
"source": [
|
||||||
"from IPython.display import clear_output, Image, display, HTML\n",
|
"from tensorflow_graph_in_jupyter import show_graph"
|
||||||
"\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))"
|
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -3490,7 +3455,7 @@
|
|||||||
"name": "python",
|
"name": "python",
|
||||||
"nbconvert_exporter": "python",
|
"nbconvert_exporter": "python",
|
||||||
"pygments_lexer": "ipython3",
|
"pygments_lexer": "ipython3",
|
||||||
"version": "3.5.5"
|
"version": "3.6.5"
|
||||||
},
|
},
|
||||||
"nav_menu": {},
|
"nav_menu": {},
|
||||||
"toc": {
|
"toc": {
|
||||||
|
|||||||
Reference in New Issue
Block a user