modeld: remove redundant pred_mu assignment (#33152)

remove redundant pred_mu assignment
This commit is contained in:
Dean Lee 2024-08-03 04:40:33 +08:00 committed by GitHub
parent c40d129f70
commit 2cd9975e4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 1 deletions

View File

@ -42,7 +42,6 @@ class Parser:
raw = outs[name]
raw = raw.reshape((raw.shape[0], max(in_N, 1), -1))
pred_mu = raw[:,:,:(raw.shape[2] - out_N)//2]
n_values = (raw.shape[2] - out_N)//2
pred_mu = raw[:,:,:n_values]
pred_std = np.exp(raw[:,:,n_values: 2*n_values])