add epoch number in float in eval_accuracy

NVIDIA-Merlin · Jun 17, 2020 · c6179ab · c6179ab
1 parent 480071b
commit c6179ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HugeCTR/src/main.cpp b/HugeCTR/src/main.cpp
@@ -111,7 +111,7 @@ void train(std::string config_file) {
         for (auto& eval_metric : eval_metrics) {
           MESSAGE_("Evaluation, " + eval_metric.first + ": " + std::to_string(eval_metric.second));
 
-          HugeCTR::LOG(timer_log.elapsedMilliseconds(), "eval_accuracy", eval_metric.second, 1, i);
+          HugeCTR::LOG(timer_log.elapsedMilliseconds(), "eval_accuracy", eval_metric.second, float(i) / solver_config.max_iter, i);
 
           // early stop doesn't support multinodes
           if (!eval_metric.first.compare("auc") && eval_metric.second >= 0.8025) {