modeld: remove redundant pred_mu assignment (#33152)
remove redundant pred_mu assignment
This commit is contained in:
parent
c40d129f70
commit
2cd9975e4f
|
@ -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])
|
||||
|
|
Loading…
Reference in New Issue