clean up

Author: tolhq

scripts/benchmark_model_inference.py

@@ -129,7 +129,7 @@
     RWSE_TIMES = 0
 
 # Keep original batch sizes list
-batch_sizes = [1, 2, 4, 8, 16, 32, 64, 96, 128, 256, 512, 640, 768, 1024, 2048, 2560, 3072, 3584, 4096, 6144, 9216, 13824, 17280, 20736, 30000, 35000, 40000, 45000, 50000, 55000, 60000, 65000, 70000, 75000, 80000, 85000, 90000]
+batch_sizes = [1, 2, 4, 8, 16, 32]
 iterations = args.iterations
 
 # ----------------------------

scripts/run_benchmark.sh

@@ -3,13 +3,11 @@
 set +e  # Do NOT exit on error
 
 CONFIGS=(
-    "gridfm01"
-    "gridfm02"
+    "grit01"
 )
 
 CONFIG_PATHS=(
-    "../examples/config/gridFMv0.1_pretraining.yaml"
-    "../examples/config/gridFMv0.2_pretraining.yaml"
+    "../examples/config/r2-1_grit_pretraining_RWSE_multi.yaml"
 )
 
 GRAPH_SIZES=(
@@ -37,4 +35,4 @@ for i in "${!CONFIGS[@]}"; do
             --num_nodes "$nodes" \
             --num_edges "$edges" || echo "Failed for $config_name with $nodes nodes"
     done
-done
+done