Update main.py

qiboteam · Apr 30, 2021 · b2ce3c7 · b2ce3c7
1 parent 6a98430
commit b2ce3c7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/examples/reuploading_classifier/main.py b/examples/reuploading_classifier/main.py
@@ -6,9 +6,9 @@
 #TODO: fix issue with .pkl
 
 parser = argparse.ArgumentParser()
-parser.add_argument("--dataset", default='circle',
+parser.add_argument("--dataset", default='tricrown',
                     help="Name of the example", type=str)
-parser.add_argument("--layers", default=2, help="Number of layers.", type=int)
+parser.add_argument("--layers", default=10, help="Number of layers.", type=int)
 
 
 def main(dataset, layers):
@@ -25,7 +25,6 @@ def main(dataset, layers):
             # Load previous results. Have we ever run these problem?
             data = pickle.load(f)
     except:
-        data = {}
         data = {dataset: {}}
 
     try: