diff --git a/examples/reuploading_classifier/main.py b/examples/reuploading_classifier/main.py index 66b800daa7..35573aefb7 100644 --- a/examples/reuploading_classifier/main.py +++ b/examples/reuploading_classifier/main.py @@ -26,7 +26,7 @@ def main(dataset, layers): data = pickle.load(f) except: data = {} - data[dataset] = {} + data = {dataset: {}} try: parameters = data[dataset][layers]