Skip to content

Commit 49eb95a

Browse files
committed
imx500: Swap pytorch/tf tab ordering. Pytorch is preferred.
1 parent 3540385 commit 49eb95a

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

documentation/asciidoc/accessories/ai-camera/model-conversion.adoc

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -26,21 +26,21 @@ The Edge-MDT package takes a parameter to select between installing the PyTorch
2626

2727
[tabs]
2828
======
29-
TensorFlow::
29+
PyTorch::
3030
+
3131
[source,console]
3232
----
33-
$ pip install edge-mdt[tf]
33+
$ pip install edge-mdt[pt]
3434
----
35-
+
36-
TIP: Always use the same version of TensorFlow you used to compress your model.
3735
38-
PyTorch::
36+
TensorFlow::
3937
+
4038
[source,console]
4139
----
42-
$ pip install edge-mdt[pt]
40+
$ pip install edge-mdt[tf]
4341
----
42+
+
43+
TIP: Always use the same version of TensorFlow you used to compress your model.
4444
======
4545

4646
If you need to install both packages, use two separate Python virtual environments. This prevents TensorFlow and PyTorch from causing conflicts with each other.
@@ -62,18 +62,18 @@ To convert a model model:
6262

6363
[tabs]
6464
======
65-
TensorFlow::
65+
PyTorch::
6666
+
6767
[source,console]
6868
----
69-
$ imxconv-tf -i <compressed Keras model> -o <output folder>
69+
$ imxconv-pt -i <compressed ONNX model> -o <output folder>
7070
----
7171
72-
PyTorch::
72+
TensorFlow::
7373
+
7474
[source,console]
7575
----
76-
$ imxconv-pt -i <compressed ONNX model> -o <output folder>
76+
$ imxconv-tf -i <compressed Keras model> -o <output folder>
7777
----
7878
======
7979

0 commit comments

Comments
 (0)