Update REA, REINFORCE, and RANDOM

This commit is contained in:
D-X-Y
2020-07-14 06:10:34 +00:00
parent ebad9197f7
commit 168b08d9e6
6 changed files with 401 additions and 7 deletions

View File

@@ -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)