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/experimental/vis-bench-algos.py
+++ b/exps/experimental/vis-bench-algos.py
@@ -30,10 +30,10 @@ def fetch_data(root_dir='./output/search', search_space='tss', dataset=None):
  alg2name, alg2path = OrderedDict(), OrderedDict()
  alg2name['REA'] = 'R-EA-SS3'
  alg2name['REINFORCE'] = 'REINFORCE-0.001'
-  # alg2name['RANDOM'] = 'RANDOM'
+  alg2name['RANDOM'] = 'RANDOM'
  for alg, name in alg2name.items():
    alg2path[alg] = os.path.join(ss_dir, dataset, name, 'results.pth')
-    assert os.path.isfile(alg2path[alg])
+    assert os.path.isfile(alg2path[alg]), 'invalid path : {:}'.format(alg2path[alg])
  alg2data = OrderedDict()
  for alg, path in alg2path.items():
    data = torch.load(path)