Add TKDNN_MODE variable to the name of the network.
This commit permits to obtain different .rt files for different precision optimizations of the same network. Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
@@ -32,7 +32,7 @@ int main() {
|
||||
std::cout<<"\n";
|
||||
|
||||
//convert network to tensorRT
|
||||
tk::dnn::NetworkRT netRT(&net, "simple.rt");
|
||||
tk::dnn::NetworkRT netRT(&net, net.getNetworkRTName("simple"));
|
||||
|
||||
dnnType *out_data, *out_data2; // cudnn output, tensorRT output
|
||||
|
||||
|
||||
Reference in New Issue
Block a user