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

fix stick man example

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
GPy/inference/latent_function_inference/var_dtc_gpu.py
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):
"""