Update create_experiments.py

create_experiments.py

@@ -5,7 +5,7 @@ def generate_ratios(max_layers, max_width):
     ratios = []
     for i in range(1, 9):  # 1/8 steps from 1:1 to 1:3 and vice versa
         ratio = i / 8
-        if ratio <= 3:  # Ensure the ratio is not greater than 3
+        if ratio <= 6:  # Ensure the ratio is not greater than 3
             ratios.append(ratio)
     return ratios
 
@@ -48,8 +48,8 @@ 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=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('--max_layers', type=int, default=72, 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=64*64*3, help='Input size (default: 224*224*3)')
     parser.add_argument('--output_size', type=int, default=10, help='Output size (default: 10)')