Update alpha_feature_vis.ipynb

fixes: tensorflow#236
abhinavsp0730 · Mar 29, 2020 · 8c71cf2 · 8c71cf2
1 parent 3e95f64
commit 8c71cf2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/notebooks/misc/alpha_feature_vis.ipynb b/notebooks/misc/alpha_feature_vis.ipynb
@@ -104,10 +104,13 @@
       "source": [
         "# Install Lucid\n",
         "!pip install --quiet lucid==0.1.0\n",
+        "#tensorflow_version only works on colab\n",
+        "%tensorflow_version 1.x\n",
         "\n",
         "# Import libraries\n",
         "import numpy as np\n",
         "import tensorflow as tf\n",
+        "assert tf.__version__.startswith('1')\n",
         "\n",
         "import lucid.modelzoo.vision_models as models\n",
         "import lucid.optvis.objectives as objectives\n",