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

fix in sparse_gp_mpi optimizer

Browse source
This commit is contained in:
Alex Feldstein 2017-03-20 13:27:25 -04:00 • committed by Akash Kumar Dhaka
parent ef924eba6c
commit 6f738bd2b4
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
GPy/core/sparse_gp_mpi.py
View file

@ -88,9 +88,9 @@ class SparseGP_MPI(SparseGP):
def optimize(self, optimizer=None, start=None, **kwargs): def optimize(self, optimizer=None, start=None, **kwargs):
self._IN_OPTIMIZATION_ = True self._IN_OPTIMIZATION_ = True
if self.mpi_comm==None: if self.mpi_comm==None:
super(SparseGP_MPI, self).optimize(optimizer,start,**kwargs) ret = super(SparseGP_MPI, self).optimize(optimizer,start,**kwargs)
elif self.mpi_comm.rank==0: elif self.mpi_comm.rank==0:
super(SparseGP_MPI, self).optimize(optimizer,start,**kwargs) ret = super(SparseGP_MPI, self).optimize(optimizer,start,**kwargs)
self.mpi_comm.Bcast(np.int32(-1),root=0) self.mpi_comm.Bcast(np.int32(-1),root=0)
elif self.mpi_comm.rank>0: elif self.mpi_comm.rank>0:
x = self.optimizer_array.copy() x = self.optimizer_array.copy()
@ -111,6 +111,7 @@ class SparseGP_MPI(SparseGP):
self._IN_OPTIMIZATION_ = False self._IN_OPTIMIZATION_ = False
raise Exception("Unrecognizable flag for synchronization!") raise Exception("Unrecognizable flag for synchronization!")
self._IN_OPTIMIZATION_ = False self._IN_OPTIMIZATION_ = False
return ret
def parameters_changed(self): def parameters_changed(self):
if isinstance(self.inference_method,VarDTC_minibatch): if isinstance(self.inference_method,VarDTC_minibatch):