diff --git a/src/main.cu b/src/main.cu index 8c30e833de4c89b30b115cf73f276b6168f10e72..f05b489598b3cb0208251230e359310954a8b2d3 100644 --- a/src/main.cu +++ b/src/main.cu @@ -176,6 +176,8 @@ int main_func(const std::vector<std::string>& arguments) { tlog::info() << "iteration=" << testbed.m_training_step << " loss=" << testbed.m_loss_scalar.val(); } } + + return 0; } NGP_NAMESPACE_END