diff --git a/bob/bio/gmm/tools/isv.py b/bob/bio/gmm/tools/isv.py index 7f42bea4bf81605660271d4a2df8a0eefe33ecc9..83e87aaea4b8c4effb675afcf960ff452e376f89 100644 --- a/bob/bio/gmm/tools/isv.py +++ b/bob/bio/gmm/tools/isv.py @@ -51,6 +51,7 @@ def isv_estep(algorithm, iteration, indices, force=False): # Load data training_list = fs.training_list('projected_gmm', 'train_projector', arrange_by_client=True) data = [algorithm.read_gmm_stats(training_list[i]) for i in range(indices[0], indices[1])] + data_initialize = [algorithm.read_gmm_stats(training_list[i]) for i in range(0,len(training_list))] # Load machine if iteration: @@ -61,8 +62,8 @@ def isv_estep(algorithm, iteration, indices, force=False): # create new ISV Base isv_base = bob.learn.em.ISVBase(algorithm.ubm, algorithm.subspace_dimension_of_u) - # Perform the E-step - trainer.initialize(isv_base, data, rng = algorithm.rng) #Just to reset the accumulators + # Perform the E-step + trainer.initialize(isv_base, data_initialize, rng = algorithm.rng) #Just to reset the accumulators trainer.e_step(isv_base, data) # write results to file @@ -130,8 +131,8 @@ def isv_mstep(algorithm, iteration, number_of_parallel_jobs, force=False, clean= # Creates the IVectorTrainer and initialize values trainer = algorithm.isv_trainer - data = [algorithm.read_gmm_stats(training_list[0])]#Loading data just to allocate memory - trainer.initialize(isv_base, data) #Just to allocate memory + data_initialize = [algorithm.read_gmm_stats(training_list[i]) for i in range(0,len(training_list))] + trainer.initialize(isv_base, data_initialize) #Just to allocate memory trainer.acc_u_a1 = statistics[0] trainer.acc_u_a2 = statistics[1] trainer.m_step(isv_base) # data is not used in M-step diff --git a/buildout.cfg b/buildout.cfg index 6f84c9a7d526317e56e1c9243e49d21c08fa1049..aae19151bee5521c394860ae71d47f26b457dd91 100644 --- a/buildout.cfg +++ b/buildout.cfg @@ -12,3 +12,4 @@ verbose = true [scripts] recipe = bob.buildout:scripts dependent-scripts = true +