apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-10 12:32:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[GPU] psicommputation

Browse source
This commit is contained in:
Zhenwen Dai 2014-03-27 17:12:17 +00:00
parent bc59cb8b22
commit 4465c5be8d
5 changed files with 153 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

4
GPy/models/ss_gplvm.py
View file

@ -62,7 +62,9 @@ class SSGPLVM(SparseGP):
if group_spike:
kernel.group_spike_prob = True
self.variational_prior.group_spike_prob = True
if isinstance(inference_method, VarDTC_GPU) and self.kern._support_GPU:
self.kern.useGPU = True
SparseGP.__init__(self, X, Y, Z, kernel, likelihood, inference_method, name, **kwargs)
self.add_parameter(self.X, index=0)