|
@@ -42,8 +42,7 @@
|
|
|
"from __future__ import absolute_import, division, print_function\n",
|
|
|
"\n",
|
|
|
"import numpy as np\n",
|
|
|
- "import tensorflow as tf\n",
|
|
|
- "import tensorflow.contrib.eager as tfe"
|
|
|
+ "import tensorflow as tf"
|
|
|
]
|
|
|
},
|
|
|
{
|
|
@@ -64,7 +63,8 @@
|
|
|
"source": [
|
|
|
"# Set Eager API\n",
|
|
|
"print(\"Setting Eager mode...\")\n",
|
|
|
- "tfe.enable_eager_execution()"
|
|
|
+ "tf.enable_eager_execution()\n",
|
|
|
+ "tfe = tf.contrib.eager"
|
|
|
]
|
|
|
},
|
|
|
{
|