adjust example parameters

Author: tolhq

examples/config/GRIT_PF_datakit_case14.yaml

@@ -28,15 +28,15 @@ model:
   # edge_dim must match the bus-bus edge feature count after transforms
   # (P_E, Q_E, YFF_TT_R, YFF_TT_I, YFT_TF_R, YFT_TF_I, TAP, ANG_MIN, ANG_MAX, RATE_A)
   edge_dim: 10
-  hidden_size: 116
+  hidden_size: 496
   # input_dim = bus feature count (used by GRIT core FeatureEncoder)
   input_dim: 15
   # Hetero adapter head dimensions
   input_bus_dim: 15
   input_gen_dim: 6
   output_bus_dim: 2
   output_gen_dim: 1
-  num_layers: 10
+  num_layers: 7
   type: GRIT
   act: relu
   encoder:

scripts/benchmark_model_inference.py

@@ -96,6 +96,8 @@
 
 config_args = NestedNamespace(**base_config)
 model = load_model(config_args).to(device).eval()
+tot_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
+print("**Total model trainable params: {}".format(tot_params))
 
 # ----------------------------
 # Parameters

scripts/run_benchmark.sh

@@ -7,7 +7,7 @@ CONFIGS=(
 )
 
 CONFIG_PATHS=(
-    "../examples/config/r2-1_grit_pretraining_RWSE_multi.yaml"
+    "../examples/config/GRIT_PF_datakit_case14.yaml"
 )
 
 GRAPH_SIZES=(