This commit is contained in:
Nikolay Bogoychev 2021-04-14 09:56:07 +01:00 коммит произвёл GitHub
Родитель 3daa024eb3
Коммит e4b58357db
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

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

@ -36,7 +36,7 @@ void BatchTranslator::initialize() {
}
}
graph_ = New<ExpressionGraph>(true); // always optimize
graph_ = New<ExpressionGraph>(true); // set the graph to be inference only
auto prec = options_->get<std::vector<std::string>>("precision", {"float32"});
graph_->setDefaultElementType(typeFromString(prec[0]));
graph_->setDevice(device_);