зеркало из https://github.com/microsoft/caffe.git
Set data_size_initialized to true after data_size is initialized
This commit is contained in:
Родитель
f4a0a6fcc2
Коммит
ffd2d7c51c
|
@ -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();
|
||||
|
|
Загрузка…
Ссылка в новой задаче