Improve version number and docker-image

README.md

@@ -36,7 +36,7 @@ $ pip install tf-keras-vis tensorflow
 * Docker
 
 ```bash
-$ docker pull keisen/tf-keras-vis:0.3.0
+$ docker pull keisen/tf-keras-vis:0.2.4
 ```
 
 > You can find other images (that's nvidia-docker images) at [dockerhub](https://hub.docker.com/repository/docker/keisen/tf-keras-vis/tags).
@@ -50,13 +50,13 @@ Please see [examples/attentions.ipynb](https://github.com/keisen/tf-keras-vis/bl
 * Run Jupyter notebooks on Docker
 
 ```bash
-$ docker run -itd -v /PATH/TO/tf-keras-vis:/tf-keras-vis -p 8888:8888 keisen/tf-keras-vis:0.3.0
+$ docker run -itd -v /PATH/TO/tf-keras-vis:/tf-keras-vis -p 8888:8888 keisen/tf-keras-vis:0.2.4
 ```
 
 Or, if you have GPU processors,
 
 ```bash
-$ docker run -itd --runtime=nvidia -v /PATH/TO/tf-keras-vis:/tf-keras-vis -p 8888:8888 keisen/tf-keras-vis:0.3.0-gpu
+$ docker run -itd --runtime=nvidia -v /PATH/TO/tf-keras-vis:/tf-keras-vis -p 8888:8888 keisen/tf-keras-vis:0.2.4-gpu
 ```
 
 

dockerfiles/cpu.Dockerfile

@@ -1,7 +1,7 @@
-FROM tensorflow/tensorflow:2.2.0-py3
+FROM tensorflow/tensorflow:2.2.0
 
 # Default ENV Settings
-ARG TF_KERAS_VIS_VERSION=0.3.0
+ARG TF_KERAS_VIS_VERSION=0.2.4
 ARG JUPYTER_ALLOW_IP="0.0.0.0"
 ARG JUPYTER_TOKEN=""
 

dockerfiles/gpu.Dockerfile

@@ -1,7 +1,7 @@
-FROM tensorflow/tensorflow:2.2.0-gpu-py3
+FROM tensorflow/tensorflow:2.2.0-gpu
 
 # Default ENV Settings
-ARG TF_KERAS_VIS_VERSION=0.3.0
+ARG TF_KERAS_VIS_VERSION=0.2.4
 ARG JUPYTER_ALLOW_IP="0.0.0.0"
 ARG JUPYTER_TOKEN=""
 

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="tf-keras-vis",
-    version="0.3.0",
+    version="0.2.4",
     author="keisen",
     author_email="[email protected]",
     description="Neural network visualization toolkit for tf.keras",