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

fix pickle for RBF GPU kernel

Browse source
This commit is contained in:
Zhenwen Dai 2014-06-23 12:02:36 +01:00
parent 59172435e2
commit 7a35c9a6ad
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
GPy/kern/_src/rbf.py
View file

@ -38,6 +38,15 @@ class RBF(Stationary):
def dK_dr(self, r): def dK_dr(self, r):
return -r*self.K_of_r(r) return -r*self.K_of_r(r)
def __getstate__(self):
dc = super(RBF, self).__getstate__()
if self.useGPU:
dc['psicomp'] = PSICOMP_RBF()
return dc
def __setstate__(self, state):
return super(RBF, self).__setstate__(state)
#---------------------------------------# #---------------------------------------#
# PSI statistics # # PSI statistics #
#---------------------------------------# #---------------------------------------#