Please provide the following information when requesting support. • Hardware (T4/V100/Xavier/Nano/etc) • Network Type (Detectnet_v2/Faster_rcnn/Yolo_v4/LPRnet/Mask_rcnn/Classification/etc) Classification TF2 • TLT Version (Please run “tlt info --verbose” and share “docker_tag” here) v5.0.0 • Tra…

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 You can trigger docker via below way. $ docker run --runtime=nvidia -it -p 8888:8888 --rm nvcr.io/nvidia/tao/tao-toolkit:5.0.0-tf…

In TAO 5.0.0, BYOM with TF1 (Classification and UNet) has been deprecated because the source code of TAO Toolkit is now fully open-sourced. To use BYOM with TF1, you will need to continue using TAO 4.0. Classification TF2 still supports BYOM with the same workflow as TAO 4.0. If you wish to bring y…

But it’s not just model weights, it’s also model structure. Can I modify the source code to load a .hdf5 even if the model structure isn’t supported?

Just so I understand correctly, that would mean a lot of changes right? Add a backbone, change the list of accepted model architectures, and then add a function to load the model. Is there a plan to add a low code version to do this Keras to .tltb conversion or do I have to rely on changing the sou…

Yes, users can do the modification to meet custom requirement based on the open sourced code.

How do I run training and evaluation if I pull the git repository locally and make changes to the source code? I instantiate the docker and then run the Jupyter notebook inside the docker? OR Can I just launch the notebooks from my conda env without the docker? My issue is, how do I make sure tao …

You can login the docker via $ docker run --runtime=nvidia -it --rm docker_name /bin/bash Then find the original file. For example, if going to modify one train.py $ find /usr |grep train.py Backup it, and then copy the modified version of train.py to replace it.

Thanks for the quick reply So the confusion I have is that I want to modify a file in https://github.com/NVIDIA/tao_tensorflow2_backend/blob/main/nvidia_tao_tf2/cv/classification/model/model_builder.py and then run the classification tf2 notebook in conda env. So how do I access the tao_tensorflow…

Train custom Keras model using BYOM

Accelerated Computing Intelligent Video Analytics TAO Toolkit

Morganh August 5, 2023, 3:58pm 4

You can modify code to support your own model architecture.

Topic		Replies	Views
TAO BYOM Mobilenetv2 model TAO Toolkit	5	115	May 8, 2025
BYOM model trainer with Tao 5.5 toolkit TAO Toolkit tao	9	163	October 21, 2024
How to apply ONNX models to TAO Toolkit? TAO Toolkit tao	5	334	May 27, 2024
Prune and transform an existing TF2 model TAO Toolkit	2	444	October 31, 2023
BYOM support TAO Toolkit	2	448	October 20, 2023
Cannot convert ONNX model using tao_byom TAO Toolkit	4	1155	September 25, 2022
Error TAO BYOM - ONNX conversion TAO Toolkit onnx , tao	5	124	October 15, 2024
Is tao able to retrain existing caffe model/onnx model? TAO Toolkit	2	384	February 4, 2024
Does BYOM in TAO Toolkit support object detection models TAO Toolkit yolo	5	109	August 9, 2024
TAO Toolkit Converter TAO Toolkit	10	446	January 5, 2024

Train custom Keras model using BYOM

Related topics