Skip to content

GitLab

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

Bug fix for SQuAD example tuner. (#134)

parent 2921e143
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/tuners/ga_customer_tuner/customer_tuner.py
View file @ 46954344
...@@ -61,7 +61,8 @@ class Individual(object): ...@@ -61,7 +61,8 @@ class Individual(object):
def mutation(self, config=None, info=None, save_dir=None): def mutation(self, config=None, info=None, save_dir=None):
self.result = None self.result = None
self.config = config if config is not None:
self.config = config
self.config.mutation() self.config.mutation()
self.restore_dir = self.save_dir self.restore_dir = self.save_dir
self.save_dir = save_dir self.save_dir = save_dir
...@@ -121,8 +122,7 @@ class CustomerTuner(Tuner): ...@@ -121,8 +122,7 @@ class CustomerTuner(Tuner):
logger.debug(str(parameters)) logger.debug(str(parameters))
logger.debug(str(reward)) logger.debug(str(reward))
indiv = graph_loads(parameters) indiv = Individual(graph_loads(parameters), result=reward)
indiv.result = reward
self.population.append(indiv) self.population.append(indiv)
return 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