Please provide the following information when requesting support.
• Hardware (RGX 3080)
• Network Type (Classification)
• TLT Version format_version: 2.0
toolkit_version: 3.22.05
published_date: 05/25/2022()
• Training spec file
• How to reproduce the issue ?
I am attempting to retrain the pretrained Resnet-50 model with my own dataset. Running “tao classification train” is successful and produces the following:
Yes I am using BYOM.
I am hoping to pick up the BYOM project which I had already started when a series of error messages led me to reach out to you for guidance. However if there is something wrong with my resnet50 model I realise that I may need to begin the exercise again.
In the gitub ReadMe for classification it says that a resnet model can be easily exported to ONNX using the command:
python export_torchvision.py -m resnet18
The resnet_50 hdf5 file from ngc is not an onnx model or pytorch model. We do not need to run BYOM against it.
BYOM is a Python-based package that converts any open-source ONNX model to a TAO-comaptible model. The TAO BYOM Converter provides a CLI to import an ONNX model and convert it to Keras. The converted model is stored in .tltb format, which is based on EFF.
If you want to run classification with resnet_50 hdf5 file from ngc, there is another notebook under classification folder. You can refer to it. The resnet_50.hdf5 file can be loaded as pretrained weights directly.
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
In previous TAO versions, only tao_voc is available because the BYOM feature is not available.
Since BYOM is available for classification in latest TAO 22.05, so we make a new notebook(BYOM_voc) for end user to get familiar with BYOM in classification.