diff --git a/tests/tools/onert_train/src/args.cc b/tests/tools/onert_train/src/args.cc index dbdd384b502..4ba73498468 100644 --- a/tests/tools/onert_train/src/args.cc +++ b/tests/tools/onert_train/src/args.cc @@ -212,7 +212,7 @@ void Args::Initialize(void) ("mem_poll,m", po::value()->default_value(false)->notifier([&](const auto &v) { _mem_poll = v; }), "Check memory polling") ("epoch", po::value()->default_value(5)->notifier([&](const auto &v) { _epoch = v; }), "Epoch number (default: 5)") ("batch_size", po::value()->default_value(32)->notifier([&](const auto &v) { _batch_size = v; }), "Batch size (default: 32)") - ("learning_rate", po::value()->default_value(1.0e-4)->notifier([&](const auto &v) { _learning_rate = v; }), "Learning rate (default: 1.0e-4)") + ("learning_rate", po::value()->default_value(0.001)->notifier([&](const auto &v) { _learning_rate = v; }), "Learning rate (default: 0.001)") ("loss", po::value()->default_value(0)->notifier([&] (const auto &v) { _loss_type = v; }), "Loss type\n" "0: MEAN_SQUARED_ERROR (default)\n"