Skip to content
Snippets Groups Projects
Commit 543b6aee authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 146927bc
No related merge requests found
...@@ -26,8 +26,9 @@ with open(args.production_hyperparameters) as fd: ...@@ -26,8 +26,9 @@ with open(args.production_hyperparameters) as fd:
def transform_to_single_hidden(hyperparameters): def transform_to_single_hidden(hyperparameters):
result = [] result = []
for size in [64, 128, 256, 1024]: for size in [64, 128, 256, 1024]:
hyperparameters['layer_sizes'] = [size] new_hyperparameters = deepcopy(hyperparameters)
result.append(deepcopy(hyperparameters)) new_hyperparameters['layer_sizes'] = [size]
result.append(new_hyperparameters)
return result return result
...@@ -46,8 +47,9 @@ transform = TRANSFORMS[args.kind] ...@@ -46,8 +47,9 @@ transform = TRANSFORMS[args.kind]
result_list = [] result_list = []
for item in production_hyperparameters_list: for item in production_hyperparameters_list:
for result_item in transform(item): results = transform(item)
for result_item in results:
if result_item not in result_list: if result_item not in result_list:
result_list.append(item) result_list.append(result_item)
dump(result_list, stdout) dump(result_list, stdout)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment