Update REA, REINFORCE, and RANDOM

2020-07-14 06:10:34 +00:00
parent ebad9197f7
commit 168b08d9e6
6 changed files with 401 additions and 7 deletions
--- a/exps/algos/BOHB.py
+++ b/exps/algos/BOHB.py
@@ -192,7 +192,7 @@ def main(xargs, nas_bench):


 if __name__ == '__main__':
-  parser = argparse.ArgumentParser("Regularized Evolution Algorithm")
+  parser = argparse.ArgumentParser("BOHB: Robust and Efficient Hyperparameter Optimization at Scale")
  parser.add_argument('--data_path',          type=str,   help='Path to dataset')
  parser.add_argument('--dataset',            type=str,   choices=['cifar10', 'cifar100', 'ImageNet16-120'], help='Choose between Cifar10/100 and ImageNet-16.')
  # channels and number-of-cells