Set data_size_initialized to true after data_size is initialized

This commit is contained in:
Kai Li 2014-02-12 16:15:08 +08:00
Родитель f4a0a6fcc2
Коммит ffd2d7c51c
1 изменённых файлов: 1 добавлений и 0 удалений

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

@ -78,6 +78,7 @@ int main(int argc, char** argv) {
};
if (!data_size_initialized) {
data_size = datum.channels() * datum.height() * datum.width();
data_size_initialized = true;
} else {
const string& data = datum.data();
CHECK_EQ(data.size(), data_size) << "Incorrect data field size " << data.size();