iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update NATS-Bench (tss version 0.9)

Browse Source
This commit is contained in:
D-X-Y
2020-09-02 07:34:12 +00:00
parent e04808c14e
commit 8d64afd4a3
5 changed files with 88 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/NATS-algos/reinforce.py
View File

@@ -192,7 +192,7 @@ if __name__ == '__main__':
parser.add_argument('--rand_seed', type=int, default=-1, help='manual seed')
args = parser.parse_args()
api = create(None, args.search_space, verbose=False)
api = create(None, args.search_space, fast_mode=True, verbose=False)
args.save_dir = os.path.join('{:}-{:}'.format(args.save_dir, args.search_space), args.dataset, 'REINFORCE-{:}'.format(args.learning_rate))
print('save-dir : {:}'.format(args.save_dir))