apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-30 14:35:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

enable gpu for inference X

Browse source
This commit is contained in:
Zhenwen Dai 2014-10-24 02:42:40 +01:00
parent 6c0d3a8a16
commit 8856144890
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
GPy/inference/latent_function_inference/inference_X.py
View file

@ -25,6 +25,8 @@ class Inference_X(Model):
super(Inference_X, self).__init__(name)
self.likelihood = model.likelihood.copy()
self.kern = model.kern.copy()
if model.kern.useGPU:
self.kern.GPU(True)
from copy import deepcopy
self.posterior = deepcopy(model.posterior)
self.variational_prior = model.variational_prior.copy()