Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
felixdittrich92 committed Apr 15, 2024
1 parent dfee857 commit cc6db23
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions references/classification/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ python references/classification/train_pytorch_character.py mobilenet_v3_large -
You can start your training in TensorFlow:

```shell
"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
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:
Expand Down Expand Up @@ -60,12 +60,12 @@ Feel free to inspect the multiple script option to customize your training to yo

Character classification:

```python
"TF_USE_LEGACY_KERAS"=1 python references/classification/train_tensorflow_character.py --help
```shell
TF_USE_LEGACY_KERAS=1 python references/classification/train_tensorflow_character.py --help
```

Orientation classification:

```python
```shell
python references/classification/train_tensorflow_orientation.py --help
```

0 comments on commit cc6db23

Please sign in to comment.