TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

[Question]:How to add a Dropout layer

Open John-ReleaseVersion opened this issue 1 year ago • 1 comments

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?

John-ReleaseVersion avatar Feb 29 '24 09:02 John-ReleaseVersion

Dropout is disabled during inference so I don't think you need it.

zerollzeng avatar Mar 03 '24 01:03 zerollzeng

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.

John-ReleaseVersion avatar Mar 06 '24 02:03 John-ReleaseVersion