README.md 847 B
TAO_Transfer_Learning: This directory contains a single lab for Transfer Learning using NVIDIA TAO toolkit. It covers concepts such as:
- TAO toolkit Architecture
- NGC API KEY registration
- Image classification Pretrained Models
- Yolov4 object detection
- Training, Evaluation, Pruning, Retraining, Inferencing, and INT8 Optimization using resnet18 pretrained model
The lab contains docker and singularity definition files. Follow the readme files inside each on how to build the container and run the labs inside it.
Questions?
Please join OpenACC Slack Channel for questions.