small changes

This commit is contained in:
Zhenwen Dai 2014-10-20 11:08:26 +01:00
parent a9ce363c3c
commit 1b3b56eaa2
2 changed files with 2 additions and 2 deletions

View file

@ -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

View file

@ -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: