From 3a3f6cee44af185979fdd7e6303816e4e0978f6a Mon Sep 17 00:00:00 2001 From: Zhenwen Dai Date: Thu, 12 Nov 2015 15:11:20 +0000 Subject: [PATCH] enhance optimize parallel --- GPy/util/parallel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/GPy/util/parallel.py b/GPy/util/parallel.py index 0c99287c..de36f780 100644 --- a/GPy/util/parallel.py +++ b/GPy/util/parallel.py @@ -29,14 +29,14 @@ def divide_data(datanum, rank, size): offset = size*rank+residue return offset, offset+size, datanum_list -def optimize_parallel(model, optimizer=None, messages=True, max_iters=1000, outpath='.', interval=100, name=None): +def optimize_parallel(model, optimizer=None, messages=True, max_iters=1000, outpath='.', interval=100, name=None, **kwargs): from math import ceil from datetime import datetime import os if name is None: name = model.name stop = 0 for iter in range(int(ceil(float(max_iters)/interval))): - model.optimize(optimizer=optimizer, messages= True if messages and model.mpi_comm.rank==model.mpi_root else False, max_iters=interval) + model.optimize(optimizer=optimizer, messages= True if messages and model.mpi_comm.rank==model.mpi_root else False, max_iters=interval, **kwargs) if model.mpi_comm.rank==model.mpi_root: timenow = datetime.now() timestr = timenow.strftime('%Y:%m:%d_%H:%M:%S')