diff --git a/code/train/Train_adv.ipynb b/code/train/Train_adv.ipynb
index 5e903bb..0f49708 100644
--- a/code/train/Train_adv.ipynb
+++ b/code/train/Train_adv.ipynb
@@ -24,7 +24,13 @@
     "from skimage.transform import rescale, resize\n",
     "import os\n",
     "from keras import optimizers\n",
-    "from keras.models import load_model\n"
+    "from keras.models import load_model\n",
+    "\n",
+    "\n",
+    "import keras\n",
+    "\n",
+    "print(\"Keras Version:      \"+keras.__version__)\n",
+    "print(\"Tensorflow Version: \"+tf.__version__)\n"
    ]
   },
   {
diff --git a/code/train/Train_phase1.ipynb b/code/train/Train_phase1.ipynb
index dfc4ae9..4b81e65 100644
--- a/code/train/Train_phase1.ipynb
+++ b/code/train/Train_phase1.ipynb
@@ -24,7 +24,12 @@
     "from skimage.transform import rescale, resize\n",
     "import os\n",
     "from keras.models import load_model\n",
-    "from sklearn.model_selection import train_test_split"
+    "from sklearn.model_selection import train_test_split\n",
+    "\n",
+    "import keras\n",
+    "\n",
+    "print(\"Keras Version:      \"+keras.__version__)\n",
+    "print(\"Tensorflow Version: \"+tf.__version__)"
    ]
   },
   {
diff --git a/code/train/Train_phase2.ipynb b/code/train/Train_phase2.ipynb
index ca9afad..f1c494c 100644
--- a/code/train/Train_phase2.ipynb
+++ b/code/train/Train_phase2.ipynb
@@ -26,7 +26,12 @@
     "\n",
     "import os\n",
     "import numpy as np\n",
-    "from keras.models import load_model"
+    "from keras.models import load_model\n",
+    "\n",
+    "import keras\n",
+    "\n",
+    "print(\"Keras Version:      \"+keras.__version__)\n",
+    "print(\"Tensorflow Version: \"+tf.__version__)"
    ]
   },
   {