schihei/ascend-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update model_process.cpp

Browse Source
This commit is contained in:
chen68
2020-10-19 01:45:32 -07:00
parent 3746263ae2
commit bf21c4c872
1 changed files with 42 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msame/src/model_process.cpp
+42 -41
View File
@@ -440,10 +440,10 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
}
if (f_isTXT) {
ofstream outstr(times + "/" + modelName + "_output_" + to_string(i) + ".txt", ios::out);
int amount_onebatch = 1;
int amount_onebatch = 1;
for (int j = 1; j < dim->dimCount; j++) {
amount_onebatch *= dim->dims[j];
}
amount_onebatch *= dim->dims[j];
}
switch (datatype) {
case 0:
for (int i = 0; i < len / sizeof(float); i++) {
@@ -451,11 +451,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 1:
@@ -464,12 +464,13 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
}
break;
case 2:
for (int i = 0; i < len / sizeof(int8_t); i++) {
@@ -477,12 +478,12 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
}
break;
case 3:
for (int i = 0; i < len / sizeof(int); i++) {
@@ -490,11 +491,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 4:
@@ -503,11 +504,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 6:
@@ -516,11 +517,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 7:
@@ -529,11 +530,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 8:
@@ -542,11 +543,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 9:
@@ -555,11 +556,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 10:
@@ -568,11 +569,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 11:
@@ -581,11 +582,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
case 12:
@@ -594,11 +595,11 @@ void ModelProcess::OutputModelResult(std::string& s, std::string& modelName, siz
outstr << out << " ";
if (i != 0 && (i + 1) % amount_onebatch == 0 && i != len / sizeof(float)-1){
outstr << "\n\n";
}else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
} else{
if ((i + 1) % 100 == 0 && i != len / sizeof(float)-1){
outstr << "\n";
}
}
}
}
break;
default: