[src] Bug-fix in model-collapsing code (thanks: aarora8)

diff --git a/src/nnet3/nnet-simple-component.cc b/src/nnet3/nnet-simple-component.cc
index 3e589df029b6a541b3b1b5937f8f93d480011bce..09bbf4334e772d68fbc6f9bcdeab729e9e801974 100644 (file)
--- a/src/nnet3/nnet-simple-component.cc
+++ b/src/nnet3/nnet-simple-component.cc
@@ -5289,8 +5289,8 @@ void BatchNormComponent::ComputeDerived() {
   if (count_ == 0.0) {
     KALDI_WARN << "Test-mode is set but there is no data count.  "
         "Creating random counts.  This only makes sense "
-        "in unit-tests.  If you see this in real life, "
-        "something is very wrong.";
+        "in unit-tests (or compute_prob_*.0.log).  If you see this "
+        "elsewhere, something is very wrong.";
     count_ = 1.0;
     stats_sum_.SetRandn();
     stats_sumsq_.SetRandn();

diff --git a/src/nnet3/nnet-utils.cc b/src/nnet3/nnet-utils.cc
index 7ec155e074ad100602e4bf2a09cd614f027ccf9a..2450f37baff43f7694b906fbfc11ac6b9d7a35ff 100644 (file)
--- a/src/nnet3/nnet-utils.cc
+++ b/src/nnet3/nnet-utils.cc
@@ -1288,7 +1288,7 @@ class ModelCollapser {
       return component_index;
     std::ostringstream os;
     os << nnet_->GetComponentName(component_index)
-       << ":scale" << std::setprecision(3) << scale;
+       << ".scale" << std::setprecision(3) << scale;
     std::string new_component_name = os.str();  // e.g. foo.s2.0
     int32 ans = nnet_->GetComponentIndex(new_component_name);
     if (ans >= 0)