Add some new examples in chapter5/5.5

kalfazed / tensorrt_starter

This repository give a guidline to learn CUDA and TensorRT from the beginning.

127 stars 35 forks source link

Add some new examples in chapter5/5.5 #8

Closed Phoenix8215 closed 2 months ago

Phoenix8215 commented 3 months ago

Overview

Hi kalfazed,i have added some new examples in chapter5/5.5.And i have already demonstrated that the network output results between Pytorch and TRT almost the same:-) please review it, 3Q~

What should reviewers check

i will provide some pictures from the output results(one from Pytorch, another is from TRT)

How to check

pooling
upsample
deconv
concat
elementwise & const
reduce
slice

Related issues

Phoenix8215 commented 2 months ago

Hi @kalfazed,

I have made the requested changes by adding dynamic parameters to the build_concat and build_slice functions. Additionally, I have removed the unnecessary weights as per your suggestion.

Could you please review the changes again?

Thank you! 🥰

Phoenix8215 commented 2 months ago

and some results for my test🥰(actually the same) img_v3_02c7_c0fe94fe-6dfb-462e-ac1d-7eb36e34d8dg img_v3_02c7_ed7267d5-a4e5-47cb-90c5-e4c81c275c8g

Phoenix8215 commented 2 months ago

Hi kalfazed,

Thank you for the approval! Unfortunately, I do not have write access to this repository, so I am unable to merge this pull request myself. Could you merge it for me?

Thank you!🥰