Update Keras script

lutzroeder · Nov 3, 2023 · e40e6e6 · e40e6e6
1 parent 746f78c
commit e40e6e6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/tf b/tools/tf
@@ -21,6 +21,7 @@ venv() {
 clean() {
     echo "tf clean"
     rm -rf "./third_party/env/tensorflow"
+    rm -rf "./third_party/source/keras"
     rm -rf "./third_party/source/tensorflow"
     rm -rf "./third_party/source/tflite-support"
 }
@@ -29,6 +30,8 @@ sync() {
     echo "tf sync"
     [ -d "./third_party/source/tensorflow" ] || git clone --quiet https://github.com/tensorflow/tensorflow.git "./third_party/source/tensorflow"
     git -C "./third_party/source/tensorflow" pull --quiet --prune
+    [ -d "./third_party/source/keras" ] || git clone --quiet https://github.com/keras-team/keras.git "./third_party/source/keras"
+    git -C "./third_party/source/keras" pull --quiet --prune
     mkdir -p "./third_party/source/tflite-support/tensorflow_lite_support/metadata"
     curl --silent --location --output "./third_party/source/tflite-support/tensorflow_lite_support/metadata/metadata_schema.fbs" "https://github.com/tensorflow/tflite-support/raw/master/tensorflow_lite_support/metadata/metadata_schema.fbs"
 }