Added missing try-catch statement
resolve 457555, 457556, 457557, 457558
Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>
return 1;
};
- std::shared_ptr<ml::train::Optimizer> optimizer =
- ml::train::optimizer::SGD({"learning_rate=0.1"});
+ std::shared_ptr<ml::train::Optimizer> optimizer;
+ try {
+ optimizer = ml::train::optimizer::SGD({"learning_rate=0.1"});
+ } catch (...) {
+ std::cerr << "creating optimizer failed";
+ return 1;
+ }
if (model->setOptimizer(optimizer) != 0) {
std::cerr << "failed to set optimizer" << std::endl;
* @brief Main gtest
*/
int main(int argc, char **argv) {
- nntrainer::AppContext::Global().setWorkingDirectory(getTestResPath(""));
+ try {
+ nntrainer::AppContext::Global().setWorkingDirectory(getTestResPath(""));
+ } catch (std::invalid_argument &e) {
+ std::cout << "failed to get test res path\n";
+ }
+
int result = -1;
try {
* @brief Main gtest
*/
int main(int argc, char **argv) {
- nntrainer::AppContext::Global().setWorkingDirectory(getTestResPath(""));
+ try {
+ nntrainer::AppContext::Global().setWorkingDirectory(getTestResPath(""));
+ } catch (std::invalid_argument &e) {
+ ml_loge("Failed to get test res path\n");
+ }
int result = -1;
try {