TensorRT
TensorRT copied to clipboard
[Question]:How to add a Dropout layer
Description
How to add Dropout layer
Environment
TensorRT Version:8.6.1.6
Question
I added a Dropout operation in the VGG16 network layer,.
After generate .wts, I want to build engine files , but I couldn't find any relevant methods to add dropout layer in TensorRT API and Tensorrtx projects.
How can I add a Dropout layer in my c++ projects?
Dropout is disabled during inference so I don't think you need it.
Dropout is disabled during inference so I don't think you need it.
Yes, after ignoring the Dropout operation, I still successfully generated the engine and implemented the classification function I wanted.Although some warnings were generated.
Thanks for your reply.