Commit 46954344 authored by Sinan Tan's avatar Sinan Tan Committed by xuehui
Browse files

Bug fix for SQuAD example tuner. (#134)

parent 2921e143
......@@ -61,7 +61,8 @@ class Individual(object):
def mutation(self, config=None, info=None, save_dir=None):
self.result = None
self.config = config
if config is not None:
self.config = config
self.config.mutation()
self.restore_dir = self.save_dir
self.save_dir = save_dir
......@@ -121,8 +122,7 @@ class CustomerTuner(Tuner):
logger.debug(str(parameters))
logger.debug(str(reward))
indiv = graph_loads(parameters)
indiv.result = reward
indiv = Individual(graph_loads(parameters), result=reward)
self.population.append(indiv)
return
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment