modeld: remove redundant pred_mu assignment (#33152)

remove redundant pred_mu assignment
old-commit-hash: 2cd9975e4f
This commit is contained in:
Dean Lee 2024-08-03 04:40:33 +08:00 committed by GitHub
parent 7adc4219f0
commit 74a9e64c2c
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])