diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 506b734..201d13f 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -5,7 +5,7 @@ MAINTAINER Sai Soundararaj ARG THEANO_VERSION=rel-0.8.2 ARG TENSORFLOW_VERSION=0.12.1 ARG TENSORFLOW_ARCH=cpu -ARG KERAS_VERSION=1.2.0 +ARG KERAS_VERSION=2.2.4 ARG LASAGNE_VERSION=v0.1 ARG TORCH_VERSION=latest ARG CAFFE_VERSION=master