Changed the maximum possible amount of units (#19)

Changed the maximum possible amount of units for the optimization to return to 70 units.
pull/23/merge
Mikhail Dmitrienko 2018-07-16 11:57:43 -05:00 committed by Matthew D. Scholefield
parent b7ae1472f1
commit 75e0769026
1 changed files with 1 additions and 1 deletions

View File

@ -82,7 +82,7 @@ class OptimizeTrainer(Trainer):
print("\n= %d = (example #%d)" % (i + 1, len(self.bb.get_data()["examples"]) + 1))
params = ModelParams(
recurrent_units=self.bb.randint("units", 1, 100, guess=50),
recurrent_units=self.bb.randint("units", 1, 70, guess=50),
dropout=self.bb.uniform("dropout", 0.1, 0.9, guess=0.6),
extra_metrics=self.args.extra_metrics,
skip_acc=self.args.no_validation,