Fix bug regarding G_prime being zero

This commit is contained in:
Daniel Povey 2022-07-25 06:57:52 +08:00
parent 3acdf3b395
commit 854c2965a9

View File

@ -1091,7 +1091,7 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of
G_prime_min = group["cov_min"][3] G_prime_min = group["cov_min"][3]
# make sure G_prime has no zero eigs, and is unit mean. # make sure G_prime has no zero eigs, and is unit mean.
#G_prime = _soft_floor(G_prime, G_prime_min * G_prime_mean, depth=3) #G_prime = _soft_floor(G_prime, G_prime_min * G_prime_mean, depth=3)
G_prime += G_prime_min * G_prime_mean G_prime = G_prime + G_prime_min * G_prime_mean + 1.0e-20
G_prime /= _mean(G_prime, exclude_dims=[0], keepdim=True) G_prime /= _mean(G_prime, exclude_dims=[0], keepdim=True)
# it now has unit mean.. # it now has unit mean..
G_prime_max = group["cov_max"][3] G_prime_max = group["cov_max"][3]