my system info: (GPU , dockerfile of tlt version 2)
I have a feature extraction model which is based on Tensorflow. I want to use this model in tlt and then prune and quantize it.
(for example we can convert our pytorch model into tensorflow model by using ONNX, I’m looking for something like that to convert my model into tlt)
- Is it possible to use our none tlt model in tlt platform (v 2.0 or 3.0)?
None tlt model such as :
1. classification model
2. feature extraction model
- for this purpose what can I do (which steps should be taken)?
any idea will be appreciated