Update REA and REINFORCE

2020-07-13 03:43:10 +00:00
parent 88a5be1368
commit 041a9aa4a3
4 changed files with 239 additions and 14 deletions
--- a/exps/algos/reinforce.py
+++ b/exps/algos/reinforce.py
@@ -184,7 +184,7 @@ def main(xargs, nas_bench):


 if __name__ == '__main__':
-  parser = argparse.ArgumentParser("Regularized Evolution Algorithm")
+  parser = argparse.ArgumentParser("The REINFORCE Algorithm")
  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