Skip to content

Commit

Permalink
Exact TSNE fewer log statements (rapidsai#3808)
Browse files Browse the repository at this point in the history
closes rapidsai#3792

Authors:
  - Divye Gala (https://github.com/divyegala)

Approvers:
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: rapidsai#3808
  • Loading branch information
divyegala authored Apr 30, 2021
1 parent 977b7e9 commit 1a5b5b1
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions cpp/src/tsne/exact_tsne.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -126,8 +126,10 @@ void Exact_TSNE(value_t *VAL, const value_idx *COL, const value_idx *ROW,
min_gain, gradient.data(), check_convergence, stream);

if (check_convergence) {
CUML_LOG_DEBUG("Z at iter = %d = %f and gradient norm = %f", iter, Z,
gradient_norm);
if (iter % 100 == 0) {
CUML_LOG_DEBUG("Z at iter = %d = %f and gradient norm = %f", iter, Z,
gradient_norm);
}
if (gradient_norm < min_grad_norm) {
CUML_LOG_DEBUG(
"Gradient norm = %f <= min_grad_norm = %f. Early stopped at iter = "
Expand All @@ -136,7 +138,9 @@ void Exact_TSNE(value_t *VAL, const value_idx *COL, const value_idx *ROW,
break;
}
} else {
CUML_LOG_DEBUG("Z at iter = %d = %f", iter, Z);
if (iter % 100 == 0) {
CUML_LOG_DEBUG("Z at iter = %d = %f", iter, Z);
}
}
}
}
Expand Down

0 comments on commit 1a5b5b1

Please sign in to comment.