Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Add unit tests for TensorRT integration and fix some bugs #15399

Merged
merged 20 commits into from
Oct 20, 2019
Merged

Add unit tests for TensorRT integration and fix some bugs #15399

merged 20 commits into from
Oct 20, 2019

Commits on Oct 2, 2019

  1. add unit tests and fix a bunch of bugs

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    518f5bf View commit details
    Browse the repository at this point in the history

  2. avoid init_tensorrt_params to modify inputs arg/aux_params

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    01f9da0 View commit details
    Browse the repository at this point in the history

  3. fix include in tensorrt-inl.h, fix bug following modification of buil… 

    …dCudaEngine, changer assert_allclose to assert_almost_equal
    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    0ce9e87 View commit details
    Browse the repository at this point in the history

  4. remove line for lint

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    36e6fc7 View commit details
    Browse the repository at this point in the history

  5. update onnx-tensorrt

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    04a5764 View commit details
    Browse the repository at this point in the history

  6. remove commented out, planned but not implemented work

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    063dbad View commit details
    Browse the repository at this point in the history

  7. restrict compatibility of deconvolution for tensorrt and make proper … 

    …unit test, remove test_tensorrt_deconvolution.py
    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    064938d View commit details
    Browse the repository at this point in the history

  8. update test_tensorrt.py adding with_seed and fixing fp16 weights loading

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    96440fe View commit details
    Browse the repository at this point in the history

  9. update libnvinfer to 5.1.5

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    f7e8e5c View commit details
    Browse the repository at this point in the history

  10. fix typo

    @Caenorst
    Caenorst committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    99575b2 View commit details
    Browse the repository at this point in the history

Commits on Oct 3, 2019

  1. get assert_allclose from numpy.testing

    @Caenorst
    Caenorst committed Oct 3, 2019
    Configuration menu
    Copy the full SHA
    8f8d4d8 View commit details
    Browse the repository at this point in the history

  2. using NVIDIA machine learning repo instead of just libnvinfer

    @Caenorst
    Caenorst committed Oct 3, 2019
    Configuration menu
    Copy the full SHA
    76bbcc7 View commit details
    Browse the repository at this point in the history

  3. dereference trt_network for createParser

    @Caenorst
    Caenorst committed Oct 3, 2019
    Configuration menu
    Copy the full SHA
    61f109c View commit details
    Browse the repository at this point in the history

  4. fix for python2

    @Caenorst
    Caenorst committed Oct 3, 2019
    Configuration menu
    Copy the full SHA
    fddac76 View commit details
    Browse the repository at this point in the history

Commits on Oct 4, 2019

  1. fix further for python2 syntax

    @Caenorst
    Caenorst committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    5c8c601 View commit details
    Browse the repository at this point in the history

  2. move test_tensorrt.py to tests/python/tensorrt/

    @Caenorst
    Caenorst committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    82565ce View commit details
    Browse the repository at this point in the history

Commits on Oct 5, 2019

  1. update unit tests

    @Caenorst
    Caenorst committed Oct 5, 2019
    Configuration menu
    Copy the full SHA
    ca332a5 View commit details
    Browse the repository at this point in the history

  2. update lenet5_train.py

    @Caenorst
    Caenorst committed Oct 5, 2019
    Configuration menu
    Copy the full SHA
    2ebc4bd View commit details
    Browse the repository at this point in the history

Commits on Oct 7, 2019

  1. remove redundant fix to batchnorm gamma correction

    @Caenorst
    Caenorst committed Oct 7, 2019
    Configuration menu
    Copy the full SHA
    60cd171 View commit details
    Browse the repository at this point in the history

  2. update test_resnet18.py

    @Caenorst
    Caenorst committed Oct 7, 2019
    Configuration menu
    Copy the full SHA
    cdd3b2e View commit details
    Browse the repository at this point in the history