Skip to content

Commit 1fe567f

Browse files
ensure keras implementation runs and tensorflow 2.0
1 parent 26701c6 commit 1fe567f

File tree

1 file changed

+24
-7
lines changed

1 file changed

+24
-7
lines changed

.travis.yml

Lines changed: 24 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,28 @@ language: r
44
dist: trusty
55
sudo: false
66

7-
env:
8-
- KERAS_BACKEND="tensorflow"
9-
- KERAS_BACKEND="theano" THEANO_FLAGS=optimizer=fast_compile
10-
- KERAS_IMPLEMENTATION="tensorflow"
11-
- KERAS_IMPLEMENTATION="tensorflow" TENSORFLOW_VERSION="1.12"
7+
matrix:
8+
include:
9+
- name: "TensorFlow (Keras)"
10+
env:
11+
- KERAS_BACKEND="tensorflow"
12+
- KERAS_IMPLEMENTATION="tensorflow"
13+
- name: "Theano"
14+
env:
15+
- KERAS_BACKEND="theano"
16+
- THEANO_FLAGS=optimizer=fast_compile
17+
- name: "TensorFlow (TensorFlow)"
18+
env:
19+
- KERAS_BACKEND="tensorflow"
20+
- KERAS_IMPLEMENTATION="tensorflow"
21+
- name: "TensorFlow (TensorFlow 1.12)"
22+
env:
23+
- KERAS_IMPLEMENTATION="tensorflow"
24+
- TENSORFLOW_WHEEL="https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.12.0-cp27-none-linux_x86_64.whl"
25+
- name: "TensorFlow (TensorFlow 2.0-preview)"
26+
env:
27+
- KERAS_IMPLEMENTATION="tensorflow"
28+
- TENSORFLOW_WHEEL="https://files.pythonhosted.org/packages/b8/48/983c8af6e63cfeebb6bb89866c38ea7d16491b8f72309f27df41b33a751e/tf_nightly_2.0_preview-1.13.0.dev20190117-cp27-cp27mu-manylinux1_x86_64.whl"
1229

1330
cache:
1431
packages: true
@@ -20,8 +37,8 @@ before_script:
2037
- pip2.7 install --upgrade --ignore-installed --user travis virtualenv
2138
- pip2.7 install --upgrade --ignore-installed --user travis keras h5py pyyaml requests Pillow scipy theano
2239
- |
23-
if [[ $TENSORFLOW_VERSION == "1.12" ]]; then
24-
pip2.7 install --upgrade --ignore-installed --user travis https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.12.0-cp27-none-linux_x86_64.whl
40+
if [[ ! -z "$TENSORFLOW_WHEEL" ]]; then
41+
pip2.7 install --upgrade --ignore-installed --user travis $TENSORFLOW_WHEEL
2542
else
2643
pip2.7 install --upgrade --ignore-installed --user travis tensorflow
2744
fi

0 commit comments

Comments
 (0)