mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

fix

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-04 10:45:07 +02:00
parent 0ff47ad6ba
commit b20a2e2902
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/yolo/yolo.cpp
+3 -4
View File
@@ -129,9 +129,8 @@ int main() {
value_type *out, *out_h;
int out_dim = net.getOutputDim().tot();
readBinaryFile(output_bin, out_dim, &out_h, &out);
std::cout<<"CUDNN vs correct Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"TRT vs correct Diffs: "<<checkResult(out_dim, out_data2, out)<<"\n";
std::cout<<"CUDNN vs TRT Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"CUDNN vs correct Wrongs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"TRT vs correct Wrongs: "<<checkResult(out_dim, out_data2, out)<<"\n";
std::cout<<"CUDNN vs TRT Wrongs: "<<checkResult(out_dim, out_data, out)<<"\n";
return 0;
}