Merge pull request #897 from Kaggle/tfa-upgrade-whl

Upgrade TensorFlow addons

Author: rosbo

Dockerfile

@@ -2,7 +2,7 @@
 ARG BASE_TAG=m46
 ARG TENSORFLOW_VERSION=2.3.1
 
-FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37 as tensorflow_whl
+FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37-2 as tensorflow_whl
 FROM gcr.io/deeplearning-platform-release/base-cpu:${BASE_TAG}
 
 ADD clean-layer.sh  /tmp/clean-layer.sh

gpu.Dockerfile

@@ -1,7 +1,7 @@
 ARG BASE_TAG=staging
 
 FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 AS nvidia
-FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.1-py37 as tensorflow_whl
+FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.1-py37-2 as tensorflow_whl
 FROM gcr.io/kaggle-images/python:${BASE_TAG}
 
 ADD clean-layer.sh  /tmp/clean-layer.sh

tensorflow-whl/CHANGELOG.md

@@ -17,4 +17,5 @@
 * `2.2.0-py37`: TensorFlow 2.2.0 with Python 3.7
 * `2.2.0-py37-2`: TensorFlow 2.2.0 with Python 3.7 & TFA
 * `2.3.0-py37`: TensorFlow 2.3.0 with Python 3.7
-* `2.3.1-py37`: TensorFlow 2.3.1 with Python 3.6
+* `2.3.1-py37`: TensorFlow 2.3.1 with Python 3.7
+* `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7

tensorflow-whl/Dockerfile

@@ -87,7 +87,7 @@ RUN pip install /tmp/tensorflow_cpu/tensorflow*.whl
 RUN cd /usr/local/src/ && \
     git clone https://github.com/tensorflow/addons && \
     cd addons && \
-    git checkout tags/v0.10.0 && \
+    git checkout tags/v0.11.2 && \
     python ./configure.py && \
     bazel build --enable_runfiles build_pip_pkg && \
     bazel-bin/build_pip_pkg /tmp/tfa_cpu && \