loading tensorflow model inside step error

[michael135]

Getting error, when loading saved

  @step
    def model_loading_step(self):
        """
        Here we are loading model Tensorflow model
        """
        print("Model loading")
        from tensorflow.keras.models import load_model
        self.model = load_model(path_to_model)
        self.next(self.end)

2022-05-01 19:25:25.928 [1651422297658984/model_loading_step/2 (pid 166635)] _pickle.PicklingError: Can't pickle <class 'tensorflow.python.keras.saving.saved_model.load.PreProcess_DensNet'>: attribute lookup PreProcess_DensNet on tensorflow.python.keras.saving.saved_model.load failed 2022-05-01 19:25:25.928 [1651422297658984/model_loading_step/2 (pid 166635)] 2022-05-01 19:25:25.928 [1651422297658984/model_loading_step/2 (pid 166635)] Path to model: /home/michaeld/fl/ATool/core/image_classification/pcp566_model 2022-05-01 19:25:25.928 [1651422297658984/model_loading_step/2 (pid 166635)] Task failed. 2022-05-01 19:25:25.929 Workflow failed. 2022-05-01 19:25:25.929 Terminating 0 active tasks... 2022-05-01 19:25:25.929 Flushing logs... Step failure: Step model_loading_step (task-id 2) failed. 2022-05-01 19:25:25.929 Workflow failed. 2022-05-01 19:25:25.929 Terminating 0 active tasks... 2022-05-01 19:25:25.929 Flushing logs...