Skip to content

[CATS] upgrade to tensorflow 2.19.0 #558

@JulienPeloton

Description

@JulienPeloton

The current CATS model has been trained on tensorflow 2.15.0, which uses Keras 2. While moving to tensorflow 2.19 (using Keras 3), we need to slightly modify the code (assuming the model is not yet retrained):

# to be put before importing tensorflow
import os
os.environ["TF_USE_LEGACY_KERAS"] = "1"

import tensorflow as tf

...

https://keras.io/getting_started/#tensorflow--keras-2-backwards-compatibility

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    Status

    New

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions