diff --git a/nntrainer/dataset/iteration_queue.cpp b/nntrainer/dataset/iteration_queue.cpp index b7ff26234a..c3eab1a5b4 100644 --- a/nntrainer/dataset/iteration_queue.cpp +++ b/nntrainer/dataset/iteration_queue.cpp @@ -173,9 +173,11 @@ IterationQueue::MarkableIteration::MarkableIteration( iq(iq) {} IterationQueue::MarkableIteration::MarkableIteration(MarkableIteration &&rhs) : - num_observed(rhs.num_observed), iteration(std::move(rhs.iteration)), - iq(rhs.iq) {} + iq(rhs.iq) { + std::lock_guard notify_lock_guard(notify_mutex); + num_observed = rhs.num_observed; +} void IterationQueue::MarkableIteration::reset() { std::lock_guard notify_lock_guard(notify_mutex); diff --git a/test/unittest/models/models_test_utils.cpp b/test/unittest/models/models_test_utils.cpp index 753467675e..6471e14a6d 100644 --- a/test/unittest/models/models_test_utils.cpp +++ b/test/unittest/models/models_test_utils.cpp @@ -142,8 +142,11 @@ class IterationForGolden { auto to_tensors = [](sharedConstTensors &sts) { std::vector ts; ts.reserve(sts.size()); - std::transform(sts.begin(), sts.end(), std::back_inserter(ts), - [](const auto &ts) { return *ts; }); + std::transform( + sts.begin(), sts.end(), + std::back_inserter(ts), [](const auto &ts) -> const auto & { + return *ts; + }); return ts; };