diff --git a/notebooks/alexnet_in_keras.ipynb b/notebooks/alexnet_in_keras.ipynb index 9c98392..0360626 100644 --- a/notebooks/alexnet_in_keras.ipynb +++ b/notebooks/alexnet_in_keras.ipynb @@ -45,7 +45,7 @@ "import keras\n", "from keras.models import Sequential\n", "from keras.layers import Dense, Dropout, Flatten, Conv2D, MaxPooling2D\n", - "from keras.layers.normalization import BatchNormalization" + "from keras.layers import BatchNormalization" ] }, { @@ -86,6 +86,7 @@ } ], "source": [ + "!pip install tflearn\n", "import tflearn.datasets.oxflower17 as oxflower17\n", "X, Y = oxflower17.load_data(one_hot=True)" ]