diff --git a/fbgemm_gpu/fbgemm_gpu/tbe/stats/bench_params_reporter.py b/fbgemm_gpu/fbgemm_gpu/tbe/stats/bench_params_reporter.py
index 5df112257e..9dc3372ce1 100644
--- a/fbgemm_gpu/fbgemm_gpu/tbe/stats/bench_params_reporter.py
+++ b/fbgemm_gpu/fbgemm_gpu/tbe/stats/bench_params_reporter.py
@@ -209,7 +209,7 @@ def extract_params(
                                     for bs in batch_size_per_feature_per_rank
                                     for b in bs
                                 ]
-                            )
+                            ).float()
                         )
                     )
                 )
diff --git a/fbgemm_gpu/src/tbe/eeg/indices_estimator.cpp b/fbgemm_gpu/src/tbe/eeg/indices_estimator.cpp
index 73b330c17d..0f949603ea 100644
--- a/fbgemm_gpu/src/tbe/eeg/indices_estimator.cpp
+++ b/fbgemm_gpu/src/tbe/eeg/indices_estimator.cpp
@@ -163,16 +163,12 @@ ZipfParameters IndicesEstimator::zipfParams(
 
       if ((ratio < kHeavyHitterLowerBound_) ||
           (ratio > kHeavyHitterUpperBound_)) {
-        std::cout << "Skipping (s,q) (" << s << ", " << q
-                  << "): " << " inconsistent with heavy hitters!" << "\n";
         continue;
       }
 
       double logLikelihood = -zipfTotalFreq * log(normalizeConst) +
           s * freqTerm - kQRegularizer_ * q;
       if (logLikelihood > maxLogLikelihood) {
-        std::cout << "Found best Log likelihood so far on (s,q) (" << s << ", "
-                  << q << "): " << logLikelihood << "\n";
         maxLogLikelihood = logLikelihood;
         zipfParams.q = q;
         zipfParams.s = s;