diff --git a/cscs-checks/apps/tensorflow/tensorflow_check.py b/cscs-checks/apps/tensorflow/tensorflow_check.py index c79f8ad3dd..05a06b4ad0 100644 --- a/cscs-checks/apps/tensorflow/tensorflow_check.py +++ b/cscs-checks/apps/tensorflow/tensorflow_check.py @@ -20,7 +20,10 @@ def __init__(self, model_name): # Checkout to the branch corresponding to the module version of # TensorFlow - self.pre_run = ['git checkout r%s' % tf_version] + # FIXME: Currently the branch for Tensorflow 1.14.0 is not + # available, we use the one for 1.13.0 + # self.pre_run = ['git checkout r%s' % tf_version] + self.pre_run = ['git checkout r1.13.0'] self.variables = {'PYTHONPATH': '$PYTHONPATH:.'} diff --git a/cscs-checks/apps/tensorflow/tf_horovod_check.py b/cscs-checks/apps/tensorflow/tf_horovod_check.py index 117f0b3dd2..586de33a01 100644 --- a/cscs-checks/apps/tensorflow/tf_horovod_check.py +++ b/cscs-checks/apps/tensorflow/tf_horovod_check.py @@ -6,13 +6,12 @@ @rfm.parameterized_test(['small'], ['large']) class TensorFlowHorovodTest(rfm.RunOnlyRegressionTest): def __init__(self, variant): - super().__init__() self.descr = 'Distributed training with TensorFlow and Horovod' self.valid_systems = ['daint:gpu'] self.valid_prog_environs = ['PrgEnv-gnu'] - tfshortver = '1.12' + tfshortver = '1.14' self.sourcesdir = 'https://github.com/tensorflow/benchmarks' - self.modules = ['Horovod/0.16.0-CrayGNU-19.03-tf-%s.0' % tfshortver] + self.modules = ['Horovod/0.16.4-CrayGNU-19.06-tf-%s.0' % tfshortver] if variant == 'small': self.valid_systems += ['dom:gpu'] self.num_tasks = 8