Exemplo n.º 1
0
 def setUp(self):
     import os
     os.environ['TF_EAGER'] = '1'
     tf_perhaps_eager_execution()
     self.om = Omega()
     self.om.models.register_backend(TFDatasetBackend.KIND, TFDatasetBackend)
     self.clean()
Exemplo n.º 2
0
    def setUp(self):
        from omegaml.backends.tensorflow.tfestimatormodel import TFEstimatorModelBackend

        self.om = Omega()
        self.om.models.register_backend(TFEstimatorModelBackend.KIND,
                                        TFEstimatorModelBackend)
        self.clean()
        tf_perhaps_eager_execution()
Exemplo n.º 3
0
 def setUp(self):
     self.om = Omega()
     self.om.models.register_backend(TensorflowKerasBackend.KIND,
                                     TensorflowKerasBackend)
     self.om.models.register_backend(TensorflowKerasSavedModelBackend.KIND,
                                     TensorflowKerasSavedModelBackend)
     self.clean()
     tf_perhaps_eager_execution()
Exemplo n.º 4
0
    def setUp(self):
        from omegaml.backends.tensorflow.tfkeras import TensorflowKerasBackend
        from omegaml.backends.tensorflow.tfkerassavedmodel import TensorflowKerasSavedModelBackend

        self.om = Omega()
        self.om.models.register_backend(TensorflowKerasBackend.KIND,
                                        TensorflowKerasBackend)
        self.om.models.register_backend(TensorflowKerasSavedModelBackend.KIND,
                                        TensorflowKerasSavedModelBackend)
        self.clean()
        tf_perhaps_eager_execution()
Exemplo n.º 5
0
 def setUp(self):
     self.om = Omega()
     self.om.models.register_backend(TFEstimatorModelBackend.KIND,
                                     TFEstimatorModelBackend)
     self.clean()
     tf_perhaps_eager_execution()
Exemplo n.º 6
0
 def setUp(self):
     self.om = Omega()
     self.om.datasets.register_backend(ProtobufDataBackend.KIND, ProtobufDataBackend)
     self.clean()
     self._images = None
     tf_perhaps_eager_execution()