Skip to content

Commit

Permalink
Merge pull request #1 from TianhongDai/update-challenge
Browse files Browse the repository at this point in the history
Update Challenge
  • Loading branch information
TianhongDai authored Nov 12, 2024
2 parents b42e165 + 2424523 commit 31c3ee2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ python ./utils/download_tiny_imagenet.py

### Instructions 📃

1. Select a complex computer vision model from Torch Hub (e.g., Dinov2).
1. Select a Vision Transformer (ViT) based computer vision model from Torch Hub (e.g., Dinov2).
2. Prepare a small subset of ImageNet images for inference.
3. Apply dynamic quantisation to the model.
4. Measure and compare the inference time of the original and quantized models.
Expand Down Expand Up @@ -58,7 +58,7 @@ Objective: Convert a trained model to TensorRT format and serialize it in ONNX f

### Instructions 📃

1. Train or use a pre-trained model (it can be the model from Part 1 or another model).
1. Use the pre-trained model from Part 1.
2. Export the model to ONNX format.
3. Convert the ONNX model to TensorRT using TensorRT tools.
4. Measure the inference time of the TensorRT model on an Nvidia GPU.
Expand Down

0 comments on commit 31c3ee2

Please sign in to comment.