Update create_experiments.py

create_experiments.py

@@ -48,10 +48,10 @@ def write_csv(experiments, filename):
 
 def main():
     parser = argparse.ArgumentParser(description='Generate a CSV file with a variety of layer counts and widths.')
-    parser.add_argument('--max_layers', type=int, default=4, help='Maximum number of layers (default: 4)')
-    parser.add_argument('--max_width', type=int, default=2048, help='Maximum width (default: 2048)')
+    parser.add_argument('--max_layers', type=int, default=36, help='Maximum number of layers (default: 4)')
+    parser.add_argument('--max_width', type=int, default=4096, help='Maximum width (default: 2048)')
     parser.add_argument('--output_file', type=str, default='experiments.csv', help='Output CSV file (default: experiments.csv)')
-    parser.add_argument('--input_size', type=int, default=224*224*3, help='Input size (default: 224*224*3)')
+    parser.add_argument('--input_size', type=int, default=64*64*3, help='Input size (default: 224*224*3)')
     parser.add_argument('--output_size', type=int, default=10, help='Output size (default: 10)')
     args = parser.parse_args()