fix stick man example

This commit is contained in:
Zhenwen Dai 2014-04-09 12:22:46 +01:00
parent 5cfc250ad1
commit 9d12c83935
6 changed files with 15 additions and 12 deletions

View file

@ -32,7 +32,7 @@ class ExactGaussianInference(object):
return Y
else:
#if Y in self.cache, return self.Cache[Y], else store Y in cache and return L.
print "WARNING: N>D of Y, we need caching of L, such that L*L^T = Y, returning Y still!"
#print "WARNING: N>D of Y, we need caching of L, such that L*L^T = Y, returning Y still!"
return Y
def inference(self, kern, X, likelihood, Y, Y_metadata=None):

View file

@ -109,7 +109,9 @@ class VarDTC_GPU(object):
x0, x1 = 0.,0.
y0, y1 = self._estimateMemoryOccupation(N, M, D)
return int((self.gpu_memory-y0-x0)/(x1+y1))
opt_batchsize = min(int((self.gpu_memory-y0-x0)/(x1+y1)), N)
return opt_batchsize
def _get_YYTfactor(self, Y):
"""