update

mindee · Apr 15, 2024 · 0067628 · 0067628
1 parent 0a108c1
commit 0067628
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/.github/workflows/references.yml b/.github/workflows/references.yml
@@ -263,7 +263,7 @@ jobs:
           pip install -e .[torch] --upgrade
       - if: matrix.framework == 'tensorflow'
         name: Benchmark latency (TF)
-        run: python references/recognition/latency_tensorflow.py crnn_mobilenet_v3_small --it 5
+        run: TF_USE_LEGACY_KERAS=1 python references/recognition/latency_tensorflow.py crnn_mobilenet_v3_small --it 5
       - if: matrix.framework == 'pytorch'
         name: Benchmark latency (PT)
         run: python references/recognition/latency_pytorch.py crnn_mobilenet_v3_small --it 5
@@ -408,7 +408,7 @@ jobs:
           pip install -e .[torch] --upgrade
       - if: matrix.framework == 'tensorflow'
         name: Benchmark latency (TF)
-        run: python references/detection/latency_tensorflow.py db_mobilenet_v3_large --it 5 --size 512
+        run: TF_USE_LEGACY_KERAS=1 python references/detection/latency_tensorflow.py db_mobilenet_v3_large --it 5 --size 512
       - if: matrix.framework == 'pytorch'
         name: Benchmark latency (PT)
         run: python references/detection/latency_pytorch.py db_mobilenet_v3_large --it 5 --size 512

diff --git a/references/classification/README.md b/references/classification/README.md
@@ -16,7 +16,7 @@ pip install -r references/requirements.txt
 You can start your training in TensorFlow:
 
 ```shell
-python references/classification/train_tensorflow_character.py mobilenet_v3_large --epochs 5
+"TF_USE_LEGACY_KERAS"=1 python references/classification/train_tensorflow_character.py mobilenet_v3_large --epochs 5
 ```
 
 or PyTorch:
@@ -30,7 +30,7 @@ python references/classification/train_pytorch_character.py mobilenet_v3_large -
 You can start your training in TensorFlow:
 
 ```shell
-python references/classification/train_tensorflow_orientation.py path/to/your/train_set path/to/your/val_set resnet18 page --epochs 5
+"TF_USE_LEGACY_KERAS"=1 python references/classification/train_tensorflow_orientation.py path/to/your/train_set path/to/your/val_set resnet18 page --epochs 5
 ```
 
 or PyTorch:
@@ -61,7 +61,7 @@ Feel free to inspect the multiple script option to customize your training to yo
 Character classification:
 
 ```python
-python references/classification/train_tensorflow_character.py --help
+"TF_USE_LEGACY_KERAS"=1 python references/classification/train_tensorflow_character.py --help
 ```
 
 Orientation classification:

diff --git a/references/detection/README.md b/references/detection/README.md
@@ -16,7 +16,7 @@ pip install -r references/requirements.txt
 You can start your training in TensorFlow:
 
 ```shell
-python references/detection/train_tensorflow.py path/to/your/train_set path/to/your/val_set db_resnet50 --epochs 5
+TF_USE_LEGACY_KERAS=1 python references/detection/train_tensorflow.py path/to/your/train_set path/to/your/val_set db_resnet50 --epochs 5
 ```
 
 or PyTorch:

diff --git a/references/recognition/README.md b/references/recognition/README.md
@@ -16,7 +16,7 @@ pip install -r references/requirements.txt
 You can start your training in TensorFlow:
 
 ```shell
-python references/recognition/train_tensorflow.py crnn_vgg16_bn --train_path path/to/your/train_set --val_path path/to/your/val_set  --epochs 5
+TF_USE_LEGACY_KERAS=1 python references/recognition/train_tensorflow.py crnn_vgg16_bn --train_path path/to/your/train_set --val_path path/to/your/val_set  --epochs 5
 ```
 
 or PyTorch:
@@ -81,7 +81,7 @@ When typing your labels, be aware that the VOCAB doesn't handle spaces. Also mak
 
 Feel free to inspect the multiple script option to customize your training to your own needs!
 
-```python
+```shell
 python references/recognition/train_pytorch.py --help
 ```