diff --git a/src/nnet3/nnet-chain-training.cc b/src/nnet3/nnet-chain-training.cc index c8dfff9e92a..3f08710fd38 100644 --- a/src/nnet3/nnet-chain-training.cc +++ b/src/nnet3/nnet-chain-training.cc @@ -186,7 +186,8 @@ void NnetChainTrainer::UpdateParamsWithMaxChange() { component_name_with_min_scale = delta_nnet_->GetComponentName(c); max_change_with_min_scale = max_param_change_per_comp; } - param_delta_squared += std::pow(scale_factors(i), 2.0) * dot_prod; + param_delta_squared += std::pow(scale_factors(i), + static_cast(2.0)) * dot_prod; i++; } } diff --git a/src/nnet3/nnet-training.cc b/src/nnet3/nnet-training.cc index 7251e4bc54b..87d64e27871 100644 --- a/src/nnet3/nnet-training.cc +++ b/src/nnet3/nnet-training.cc @@ -136,7 +136,8 @@ void NnetTrainer::UpdateParamsWithMaxChange() { component_name_with_min_scale = delta_nnet_->GetComponentName(c); max_change_with_min_scale = max_param_change_per_comp; } - param_delta_squared += std::pow(scale_factors(i), 2.0) * dot_prod; + param_delta_squared += std::pow(scale_factors(i), + static_cast(2.0)) * dot_prod; i++; } }