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

add debug util module and try to debug sparsegp_mpi

Browse source
This commit is contained in:
Zhenwen Dai 2014-08-29 15:21:21 +01:00
parent 313a238b15
commit a6e7d01ad2
3 changed files with 45 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
GPy/core/sparse_gp_mpi.py
View file

@ -3,12 +3,7 @@
import numpy as np
from sparse_gp import SparseGP
from parameterization.param import Param
from ..inference.latent_function_inference import var_dtc
from .. import likelihoods
from parameterization.variational import VariationalPosterior
from ..inference.latent_function_inference.var_dtc_parallel import update_gradients, VarDTC_minibatch
from ..core.parameterization.parameter_core import OptimizationHandlable
import logging
logger = logging.getLogger("sparse gp mpi")
@ -80,11 +75,15 @@ class SparseGP_MPI(SparseGP):
#=====================================================
@SparseGP.optimizer_array.setter
def optimizer_array(self, p):
def optimizer_array(self, p):
if self.mpi_comm != None:
if self._IN_OPTIMIZATION_ and self.mpi_comm.rank==0:
self.mpi_comm.Bcast(np.int32(1),root=0)
self.mpi_comm.Bcast(p, root=0)
from ..util.debug import checkFinite
checkFinite(p, 'optimizer_array')
SparseGP.optimizer_array.fset(self,p)
def optimize(self, optimizer=None, start=None, **kwargs):