Skip to content

Commit 0bc5f8f

Browse files
authored
Merge pull request #1623 from huynhnguyen/master
fix issue with advi when running on GPU
2 parents cad8ec2 + 5376771 commit 0bc5f8f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc3/variational/advi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def _calc_elbo(vars, model, n_mcsamples, random_seed):
182182

183183
uw = tt.vector('uw')
184184
uw.tag.test_value = np.concatenate([inarray.tag.test_value,
185-
inarray.tag.test_value])
185+
inarray.tag.test_value]).astype(theano.config.floatX)
186186

187187
elbo = _elbo_t(logp, uw, inarray, n_mcsamples, random_seed)
188188

0 commit comments

Comments
 (0)