search

kuroko1t / nne

convert a pytorch model to a model for edge device
Apache License 2.0
18 stars 3 forks source link

update fp16_mode in TensorRT function #13

Closed kuroko1t closed 4 years ago

kuroko1t commented 4 years ago

refactaring tensorrt function and , add fp16_mode setting about TensorRT