This commit is contained in:
linmin 2014-02-21 11:25:22 +08:00 коммит произвёл Evan Shelhamer
Родитель 53db54aa25
Коммит a2c2f3ce57
1 изменённых файлов: 7 добавлений и 6 удалений

Просмотреть файл

@ -52,12 +52,13 @@ int main(int argc, char** argv) {
const string& data = datum.data(); const string& data = datum.data();
size_in_datum = std::max<int>(datum.data().size(), datum.float_data_size()); size_in_datum = std::max<int>(datum.data().size(), datum.float_data_size());
CHECK_EQ(size_in_datum, data_size) << "Incorrect data field size " << size_in_datum; CHECK_EQ(size_in_datum, data_size) << "Incorrect data field size " << size_in_datum;
<< data.size(); if (data.size() != 0) {
for (int i = 0; i < size_in_datum; ++i) { for (int i = 0; i < size_in_datum; ++i) {
if (data.size() != 0) { sum_blob.set_data(i, sum_blob.data(i) + (uint8_t)data[i]);
sum_blob.set_data(i, sum_blob.data(i) + (uint8_t)data[i]); }
} else { } else {
sum_blob.set_data(i, sum_blob.data(i) + (float)datum.float_data(i)); for (int i = 0; i < size_in_datum; ++i) {
sum_blob.set_data(i, sum_blob.data(i) + (float)datum.float_data(i));
} }
} }
++count; ++count;