mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-30 14:35:15 +02:00
small changes
This commit is contained in:
parent
a9ce363c3c
commit
1b3b56eaa2
2 changed files with 2 additions and 2 deletions
|
|
@ -291,7 +291,7 @@ gpu_code = """
|
||||||
|
|
||||||
class PSICOMP_SSRBF_GPU(PSICOMP_RBF):
|
class PSICOMP_SSRBF_GPU(PSICOMP_RBF):
|
||||||
|
|
||||||
def __init__(self, threadnum=128, blocknum=15, GPU_direct=False):
|
def __init__(self, threadnum=64, blocknum=16, GPU_direct=False):
|
||||||
self.GPU_direct = GPU_direct
|
self.GPU_direct = GPU_direct
|
||||||
self.gpuCache = None
|
self.gpuCache = None
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -64,7 +64,7 @@ class SSGPLVM(SparseGP_MPI):
|
||||||
[self.X.gamma[:,i].tie_together() for i in xrange(self.X.gamma.shape[1])] # Tie columns together
|
[self.X.gamma[:,i].tie_together() for i in xrange(self.X.gamma.shape[1])] # Tie columns together
|
||||||
|
|
||||||
def _init_X(self, input_dim, Y=None, X=None, X_variance=None, Gamma=None, init='PCA'):
|
def _init_X(self, input_dim, Y=None, X=None, X_variance=None, Gamma=None, init='PCA'):
|
||||||
if X == None:
|
if X is None:
|
||||||
from ..util.initialization import initialize_latent
|
from ..util.initialization import initialize_latent
|
||||||
X, fracs = initialize_latent(init, input_dim, Y)
|
X, fracs = initialize_latent(init, input_dim, Y)
|
||||||
else:
|
else:
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue