The accuracy is almost 75 to 80% of my trained model.
Now I want to use the TAO toolkit for pruning purpose (Only pruning, not optimization or any other thing).
Please provide a detailed video or complete guide on how to prune my customized developed ONNX model via the TAO tool kit.
If there is any other info needed please let me know
Thanks.
Is it a classification onnx model? If yes, you can use TAO BYOM feature to convert it to .tltb model. Then train with very low learning rate to get the .tlt model and further prune it.
Thanks for the reply,
I gave it a try and the following error is happening.
ERROR: Command errored out with exit status 1:
command: 'C:\Users\wasee\miniconda3\envs\byom_dev\python.exe' -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'C:\\Users\\wasee\\AppData\\Local\\Temp\\pip-install-rjprvti6\\nvidia-tensorflow_23ae8ad94a254207ab6c8c5c4608a653\\setup.py'"'"'; __file__='"'"'C:\\Users\\wasee\\AppData\\Local\\Temp\\pip-install-rjprvti6\\nvidia-tensorflow_23ae8ad94a254207ab6c8c5c4608a653\\setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' egg_info --egg-base 'C:\Users\wasee\AppData\Local\Temp\pip-pip-egg-info-o3xo5cxa'
cwd: C:\Users\wasee\AppData\Local\Temp\pip-install-rjprvti6\nvidia-tensorflow_23ae8ad94a254207ab6c8c5c4608a653\
Complete output (18 lines):
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "C:\Users\wasee\AppData\Local\Temp\pip-install-rjprvti6\nvidia-tensorflow_23ae8ad94a254207ab6c8c5c4608a653\setup.py", line 150, in <module>
raise RuntimeError(open("ERROR.txt", "r").read())
RuntimeError:
###########################################################################################
The package you are trying to install is only a placeholder project on PyPI.org repository.
This package is hosted on NVIDIA Python Package Index.
This package can be installed as:
```
$ pip install nvidia-pyindex
$ pip install nvidia-tensorflow
```
Please refer to NVIDIA instructions: https://github.com/NVIDIA/tensorflow#install.
###########################################################################################
----------------------------------------
WARNING: Discarding https://files.pythonhosted.org/packages/c7/b8/b96bd09cc7c9cf9dbd8b82313c4f9860fe615b3ce7f3f6efc962d166178b/nvidia-tensorflow-0.0.1.dev5.tar.gz#sha256=65f44f1b583b950925a3617a75fd299c29975f1f14327bb2d946cb52e84baef5 (from https://pypi.org/simple/nvidia-tensorflow/). Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.
Downloading nvidia-tensorflow-0.0.1.dev4.tar.gz (3.8 kB)
ERROR: Command errored out with exit status 1:
command: 'C:\Users\wasee\miniconda3\envs\byom_dev\python.exe' -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'C:\\Users\\wasee\\AppData\\Local\\Temp\\pip-install-rjprvti6\\nvidia-tensorflow_14cae1d2da3345a18768265786b44465\\setup.py'"'"'; __file__='"'"'C:\\Users\\wasee\\AppData\\Local\\Temp\\pip-install-rjprvti6\\nvidia-tensorflow_14cae1d2da3345a18768265786b44465\\setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' egg_info --egg-base 'C:\Users\wasee\AppData\Local\Temp\pip-pip-egg-info-4uu05pme'
cwd: C:\Users\wasee\AppData\Local\Temp\pip-install-rjprvti6\nvidia-tensorflow_14cae1d2da3345a18768265786b44465\
Complete output (17 lines):
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "C:\Users\wasee\AppData\Local\Temp\pip-install-rjprvti6\nvidia-tensorflow_14cae1d2da3345a18768265786b44465\setup.py", line 150, in <module>
raise RuntimeError(open("ERROR.txt", "r").read())
RuntimeError:
###########################################################################################
The package you are trying to install is only a placeholder project on PyPI.org repository.
This package is hosted on NVIDIA Python Package Index.
This package can be installed as:
```
$ pip install nvidia-pyindex
$ pip install nvidia-tensorflow
```
Please refer to NVIDIA instructions: https://github.com/NVIDIA/tensorflow#install.
###########################################################################################
----------------------------------------
WARNING: Discarding https://files.pythonhosted.org/packages/97/39/b1aa7fe84e2ad1f183f5fc851bea78bb46a73ebb9a91ffed0878f656f2f6/nvidia-tensorflow-0.0.1.dev4.tar.gz#sha256=1be0b140aadedcd7a05f736566587bb1aacc6baa1797d7fda7d8f23e59f35142 (from https://pypi.org/simple/nvidia-tensorflow/). Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output.
ERROR: Could not find a version that satisfies the requirement nvidia-tensorflow (from versions: 0.0.1.dev4, 0.0.1.dev5)
ERROR: No matching distribution found for nvidia-tensorflow
And please provide the command to run pruning after conversion.
There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks