You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: documentation/asciidoc/accessories/ai-camera/model-conversion.adoc
+10-10Lines changed: 10 additions & 10 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -26,21 +26,21 @@ The Edge-MDT package takes a parameter to select between installing the PyTorch
26
26
27
27
[tabs]
28
28
======
29
-
TensorFlow::
29
+
PyTorch::
30
30
+
31
31
[source,console]
32
32
----
33
-
$ pip install edge-mdt[tf]
33
+
$ pip install edge-mdt[pt]
34
34
----
35
-
+
36
-
TIP: Always use the same version of TensorFlow you used to compress your model.
37
35
38
-
PyTorch::
36
+
TensorFlow::
39
37
+
40
38
[source,console]
41
39
----
42
-
$ pip install edge-mdt[pt]
40
+
$ pip install edge-mdt[tf]
43
41
----
42
+
+
43
+
TIP: Always use the same version of TensorFlow you used to compress your model.
44
44
======
45
45
46
46
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:
62
62
63
63
[tabs]
64
64
======
65
-
TensorFlow::
65
+
PyTorch::
66
66
+
67
67
[source,console]
68
68
----
69
-
$ imxconv-tf -i <compressed Keras model> -o <output folder>
0 commit comments