From 419c400a9be0423b03564e58da2b4ac0d527dcf3 Mon Sep 17 00:00:00 2001
From: inaniloquentee <3051000145@qq.com>
Date: Sat, 14 Feb 2026 23:22:01 +0800
Subject: [PATCH 1/3] Add PIGU_Hybrid model computation graph

---
 samples/pigu_hybrid/graph_hash.txt            |    1 +
 samples/pigu_hybrid/graph_net.json            |    7 +
 samples/pigu_hybrid/input_meta.py             |    0
 .../pigu_hybrid/input_tensor_constraints.py   |    0
 samples/pigu_hybrid/model.py                  |  237 ++++
 samples/pigu_hybrid/weight_meta.py            | 1061 +++++++++++++++++
 6 files changed, 1306 insertions(+)
 create mode 100644 samples/pigu_hybrid/graph_hash.txt
 create mode 100644 samples/pigu_hybrid/graph_net.json
 create mode 100644 samples/pigu_hybrid/input_meta.py
 create mode 100644 samples/pigu_hybrid/input_tensor_constraints.py
 create mode 100644 samples/pigu_hybrid/model.py
 create mode 100644 samples/pigu_hybrid/weight_meta.py

diff --git a/samples/pigu_hybrid/graph_hash.txt b/samples/pigu_hybrid/graph_hash.txt
new file mode 100644
index 000000000..359b0a72d
--- /dev/null
+++ b/samples/pigu_hybrid/graph_hash.txt
@@ -0,0 +1 @@
+b0d0d50c6f76c99353442db322e1a6b1994b47187e1ce995eaee1ef8c104400a
\ No newline at end of file
diff --git a/samples/pigu_hybrid/graph_net.json b/samples/pigu_hybrid/graph_net.json
new file mode 100644
index 000000000..660acd178
--- /dev/null
+++ b/samples/pigu_hybrid/graph_net.json
@@ -0,0 +1,7 @@
+{
+    "framework": "torch",
+    "num_devices_required": 1,
+    "num_nodes_required": 1,
+    "dynamic": true,
+    "model_name": "pigu_hybrid"
+}
\ No newline at end of file
diff --git a/samples/pigu_hybrid/input_meta.py b/samples/pigu_hybrid/input_meta.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/samples/pigu_hybrid/input_tensor_constraints.py b/samples/pigu_hybrid/input_tensor_constraints.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/samples/pigu_hybrid/model.py b/samples/pigu_hybrid/model.py
new file mode 100644
index 000000000..84f97b558
--- /dev/null
+++ b/samples/pigu_hybrid/model.py
@@ -0,0 +1,237 @@
+import torch
+
+class GraphModule(torch.nn.Module):
+    
+    
+    
+    def forward(self, s10 : torch.SymInt, L_sensor_vals_ : torch.Tensor, L_sensor_coords_ : torch.Tensor, L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps : torch.Tensor, L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_dropout : torch.Tensor, L_self_modules_sensor_encoder_modules_norm_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_norm_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_norm_eps : torch.Tensor, L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_grid_coords_ : torch.Tensor, L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_projector_transition_x : torch.Tensor, L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ : torch.nn.parameter.Parameter, G_np_pi : torch.Tensor, L_self_modules_projector_modules_sensor_embedding_buffers_B_high_ : torch.Tensor, L_self_modules_projector_modules_key_proj_smooth_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_key_proj_smooth_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_key_proj_detail_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_key_proj_detail_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_projector_scale : torch.Tensor, L_base_flow_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_up1_scale_factor : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_up2_scale_factor : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_out_conv_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_out_conv_parameters_bias_ : torch.nn.parameter.Parameter):
+        l_sensor_vals_ = L_sensor_vals_
+        l_sensor_coords_ = L_sensor_coords_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ = L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ = L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ = L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps = L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps
+        l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ = L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ = L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_
+        l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_
+        l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ = L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_
+        l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ = L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_
+        l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ = L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_
+        l_self_modules_sensor_encoder_modules_attn_dropout = L_self_modules_sensor_encoder_modules_attn_dropout
+        l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = L_self_modules_sensor_encoder_modules_norm_parameters_weight_
+        l_self_modules_sensor_encoder_modules_norm_parameters_bias_ = L_self_modules_sensor_encoder_modules_norm_parameters_bias_
+        l_self_modules_sensor_encoder_modules_norm_eps = L_self_modules_sensor_encoder_modules_norm_eps
+        l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ = L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_
+        l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ = L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_
+        l_grid_coords_ = L_grid_coords_
+        l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ = L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_
+        l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ = L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_
+        l_self_modules_projector_transition_x = L_self_modules_projector_transition_x
+        l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_
+        l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ = L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_
+        g_np_pi = G_np_pi
+        l_self_modules_projector_modules_sensor_embedding_buffers_b_high_ = L_self_modules_projector_modules_sensor_embedding_buffers_B_high_
+        l_self_modules_projector_modules_key_proj_smooth_parameters_weight_ = L_self_modules_projector_modules_key_proj_smooth_parameters_weight_
+        l_self_modules_projector_modules_key_proj_smooth_parameters_bias_ = L_self_modules_projector_modules_key_proj_smooth_parameters_bias_
+        l_self_modules_projector_modules_key_proj_detail_parameters_weight_ = L_self_modules_projector_modules_key_proj_detail_parameters_weight_
+        l_self_modules_projector_modules_key_proj_detail_parameters_bias_ = L_self_modules_projector_modules_key_proj_detail_parameters_bias_
+        l_self_modules_projector_scale = L_self_modules_projector_scale
+        l_base_flow_ = L_base_flow_
+        l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_
+        l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_
+        l_self_modules_in_conv_modules_double_conv_modules_1_momentum = L_self_modules_in_conv_modules_double_conv_modules_1_momentum
+        l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_
+        l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_
+        l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_
+        l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_
+        l_self_modules_in_conv_modules_double_conv_modules_1_eps = L_self_modules_in_conv_modules_double_conv_modules_1_eps
+        l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_
+        l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_
+        l_self_modules_in_conv_modules_double_conv_modules_4_momentum = L_self_modules_in_conv_modules_double_conv_modules_4_momentum
+        l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_
+        l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_
+        l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_
+        l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_
+        l_self_modules_in_conv_modules_double_conv_modules_4_eps = L_self_modules_in_conv_modules_double_conv_modules_4_eps
+        l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum = L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps = L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps
+        l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum = L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps = L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps
+        l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum = L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps = L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps
+        l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum = L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps = L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps
+        l_self_modules_up1_scale_factor = L_self_modules_up1_scale_factor
+        l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_
+        l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_
+        l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = L_self_modules_conv_up1_modules_double_conv_modules_1_momentum
+        l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_
+        l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_
+        l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_
+        l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_
+        l_self_modules_conv_up1_modules_double_conv_modules_1_eps = L_self_modules_conv_up1_modules_double_conv_modules_1_eps
+        l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_
+        l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_
+        l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = L_self_modules_conv_up1_modules_double_conv_modules_4_momentum
+        l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_
+        l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_
+        l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_
+        l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_
+        l_self_modules_conv_up1_modules_double_conv_modules_4_eps = L_self_modules_conv_up1_modules_double_conv_modules_4_eps
+        l_self_modules_up2_scale_factor = L_self_modules_up2_scale_factor
+        l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_
+        l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_
+        l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = L_self_modules_conv_up2_modules_double_conv_modules_1_momentum
+        l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_
+        l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_
+        l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_
+        l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_
+        l_self_modules_conv_up2_modules_double_conv_modules_1_eps = L_self_modules_conv_up2_modules_double_conv_modules_1_eps
+        l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_
+        l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_
+        l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = L_self_modules_conv_up2_modules_double_conv_modules_4_momentum
+        l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_
+        l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_
+        l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_
+        l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_
+        l_self_modules_conv_up2_modules_double_conv_modules_4_eps = L_self_modules_conv_up2_modules_double_conv_modules_4_eps
+        l_self_modules_out_conv_parameters_weight_ = L_self_modules_out_conv_parameters_weight_
+        l_self_modules_out_conv_parameters_bias_ = L_self_modules_out_conv_parameters_bias_
+        node_in = torch.cat([l_sensor_vals_, l_sensor_coords_], dim = -1);  l_sensor_vals_ = None
+        input_1 = torch.nn.functional.linear(node_in, l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_, l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_);  node_in = l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ = l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ = None
+        item = l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps.item();  l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps = None
+        input_2 = torch.nn.functional.layer_norm(input_1, (64,), l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_, l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_, item);  input_1 = l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ = l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = item = None
+        input_3 = torch.nn.functional.silu(input_2, inplace = False);  input_2 = None
+        input_4 = torch.nn.functional.linear(input_3, l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_, l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_);  input_3 = l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ = l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ = None
+        query = input_4.transpose(1, 0)
+        item_1 = l_self_modules_sensor_encoder_modules_attn_dropout.item();  l_self_modules_sensor_encoder_modules_attn_dropout = None
+        multi_head_attention_forward = torch.nn.functional.multi_head_attention_forward(query, query, query, 64, 4, l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_, l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_, None, None, False, item_1, l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_, l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_, training = False, key_padding_mask = None, need_weights = True, attn_mask = None, average_attn_weights = True, is_causal = False);  query = l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ = l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = item_1 = l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ = l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ = None
+        attn_output = multi_head_attention_forward[0];  multi_head_attention_forward = None
+        h_attn = attn_output.transpose(1, 0);  attn_output = None
+        add = input_4 + h_attn;  input_4 = h_attn = None
+        item_2 = l_self_modules_sensor_encoder_modules_norm_eps.item();  l_self_modules_sensor_encoder_modules_norm_eps = None
+        h = torch.nn.functional.layer_norm(add, (64,), l_self_modules_sensor_encoder_modules_norm_parameters_weight_, l_self_modules_sensor_encoder_modules_norm_parameters_bias_, item_2);  add = l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = l_self_modules_sensor_encoder_modules_norm_parameters_bias_ = item_2 = None
+        global_context = torch.mean(h, dim = 1)
+        input_5 = torch.nn.functional.linear(l_grid_coords_, l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_, l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_);  l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ = l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ = None
+        input_6 = torch.nn.functional.silu(input_5, inplace = False);  input_5 = None
+        input_7 = torch.nn.functional.linear(input_6, l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_, l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_);  input_6 = l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ = l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ = None
+        Q_smooth = input_7[(Ellipsis, slice(None, 64, None))]
+        Q_detail = input_7[(Ellipsis, slice(64, None, None))];  input_7 = None
+        grid_x = l_grid_coords_[(Ellipsis, slice(0, 1, None))]
+        item_3 = l_self_modules_projector_transition_x.item();  l_self_modules_projector_transition_x = None
+        sub = grid_x - item_3;  grid_x = item_3 = None
+        mul = 50.0 * sub;  sub = None
+        beta = torch.sigmoid(mul);  mul = None
+        Q_detail_1 = Q_detail * beta;  Q_detail = beta = None
+        linear_4 = torch.nn.functional.linear(l_sensor_coords_, l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_, l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_);  l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ = None
+        feat_smooth = torch.nn.functional.silu(linear_4);  linear_4 = None
+        item_4 = g_np_pi.item();  g_np_pi = None
+        mul_2 = 2 * item_4;  item_4 = None
+        mul_3 = mul_2 * l_sensor_coords_;  mul_2 = l_sensor_coords_ = None
+        proj_high = mul_3 @ l_self_modules_projector_modules_sensor_embedding_buffers_b_high_;  mul_3 = l_self_modules_projector_modules_sensor_embedding_buffers_b_high_ = None
+        sin = torch.sin(proj_high)
+        cos = torch.cos(proj_high);  proj_high = None
+        feat_detail = torch.cat([sin, cos], dim = -1);  sin = cos = None
+        K_smooth = torch.nn.functional.linear(feat_smooth, l_self_modules_projector_modules_key_proj_smooth_parameters_weight_, l_self_modules_projector_modules_key_proj_smooth_parameters_bias_);  feat_smooth = l_self_modules_projector_modules_key_proj_smooth_parameters_weight_ = l_self_modules_projector_modules_key_proj_smooth_parameters_bias_ = None
+        K_detail = torch.nn.functional.linear(feat_detail, l_self_modules_projector_modules_key_proj_detail_parameters_weight_, l_self_modules_projector_modules_key_proj_detail_parameters_bias_);  feat_detail = l_self_modules_projector_modules_key_proj_detail_parameters_weight_ = l_self_modules_projector_modules_key_proj_detail_parameters_bias_ = None
+        score_smooth = torch.functional.einsum('bhwc,bsc->bhws', Q_smooth, K_smooth);  Q_smooth = K_smooth = None
+        score_detail = torch.functional.einsum('bhwc,bsc->bhws', Q_detail_1, K_detail);  Q_detail_1 = K_detail = None
+        add_1 = score_smooth + score_detail;  score_smooth = score_detail = None
+        item_5 = l_self_modules_projector_scale.item();  l_self_modules_projector_scale = None
+        attn_logits = add_1 * item_5;  add_1 = item_5 = None
+        attn = torch.nn.functional.softmax(attn_logits, dim = -1);  attn_logits = None
+        out = torch.functional.einsum('bhws,bsc->bhwc', attn, h);  attn = h = None
+        grid_feats = out.permute(0, 3, 1, 2);  out = None
+        view = global_context.view(1, 64, 1, 1);  global_context = None
+        global_map = view.expand(1, 64, 128, 256);  view = None
+        coords_map = l_grid_coords_.permute(0, 3, 1, 2);  l_grid_coords_ = None
+        coords_map_1 = torch.nn.functional.interpolate(coords_map, size = (128, 256), mode = 'bilinear', align_corners = True);  coords_map = None
+        base_flow = torch.nn.functional.interpolate(l_base_flow_, size = (128, 256), mode = 'bilinear', align_corners = True);  l_base_flow_ = None
+        x = torch.cat([grid_feats, coords_map_1, global_map, base_flow], dim = 1);  grid_feats = coords_map_1 = global_map = None
+        input_8 = torch.conv2d(x, l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  x = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ = None
+        item_6 = l_self_modules_in_conv_modules_double_conv_modules_1_momentum.item();  l_self_modules_in_conv_modules_double_conv_modules_1_momentum = None
+        item_7 = l_self_modules_in_conv_modules_double_conv_modules_1_eps.item();  l_self_modules_in_conv_modules_double_conv_modules_1_eps = None
+        input_9 = torch.nn.functional.batch_norm(input_8, l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_, False, item_6, item_7);  input_8 = l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ = item_6 = item_7 = None
+        input_10 = torch.nn.functional.silu(input_9, inplace = True);  input_9 = None
+        input_11 = torch.conv2d(input_10, l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_10 = l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ = None
+        item_8 = l_self_modules_in_conv_modules_double_conv_modules_4_momentum.item();  l_self_modules_in_conv_modules_double_conv_modules_4_momentum = None
+        item_9 = l_self_modules_in_conv_modules_double_conv_modules_4_eps.item();  l_self_modules_in_conv_modules_double_conv_modules_4_eps = None
+        input_12 = torch.nn.functional.batch_norm(input_11, l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_, False, item_8, item_9);  input_11 = l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ = item_8 = item_9 = None
+        input_13 = torch.nn.functional.silu(input_12, inplace = True);  input_12 = None
+        input_14 = torch.nn.functional.max_pool2d(input_13, 2, 2, 0, 1, ceil_mode = False, return_indices = False)
+        input_15 = torch.conv2d(input_14, l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_14 = l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ = None
+        item_10 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum = None
+        item_11 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps = None
+        input_16 = torch.nn.functional.batch_norm(input_15, l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_, False, item_10, item_11);  input_15 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ = item_10 = item_11 = None
+        input_17 = torch.nn.functional.silu(input_16, inplace = True);  input_16 = None
+        input_18 = torch.conv2d(input_17, l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_17 = l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ = None
+        item_12 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum = None
+        item_13 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps = None
+        input_19 = torch.nn.functional.batch_norm(input_18, l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_, False, item_12, item_13);  input_18 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ = item_12 = item_13 = None
+        input_20 = torch.nn.functional.silu(input_19, inplace = True);  input_19 = None
+        input_21 = torch.nn.functional.max_pool2d(input_20, 2, 2, 0, 1, ceil_mode = False, return_indices = False)
+        input_22 = torch.conv2d(input_21, l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_21 = l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ = None
+        item_14 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum = None
+        item_15 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps = None
+        input_23 = torch.nn.functional.batch_norm(input_22, l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_, False, item_14, item_15);  input_22 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ = item_14 = item_15 = None
+        input_24 = torch.nn.functional.silu(input_23, inplace = True);  input_23 = None
+        input_25 = torch.conv2d(input_24, l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_24 = l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ = None
+        item_16 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum = None
+        item_17 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps = None
+        input_26 = torch.nn.functional.batch_norm(input_25, l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_, False, item_16, item_17);  input_25 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ = item_16 = item_17 = None
+        input_27 = torch.nn.functional.silu(input_26, inplace = True);  input_26 = None
+        item_18 = l_self_modules_up1_scale_factor.item();  l_self_modules_up1_scale_factor = None
+        x_1 = torch.nn.functional.interpolate(input_27, None, item_18, 'bilinear', True, recompute_scale_factor = None);  input_27 = item_18 = None
+        x_2 = torch.cat([x_1, input_20], dim = 1);  x_1 = input_20 = None
+        input_28 = torch.conv2d(x_2, l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  x_2 = l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = None
+        item_19 = l_self_modules_conv_up1_modules_double_conv_modules_1_momentum.item();  l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = None
+        item_20 = l_self_modules_conv_up1_modules_double_conv_modules_1_eps.item();  l_self_modules_conv_up1_modules_double_conv_modules_1_eps = None
+        input_29 = torch.nn.functional.batch_norm(input_28, l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_, False, item_19, item_20);  input_28 = l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ = item_19 = item_20 = None
+        input_30 = torch.nn.functional.silu(input_29, inplace = True);  input_29 = None
+        input_31 = torch.conv2d(input_30, l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_30 = l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = None
+        item_21 = l_self_modules_conv_up1_modules_double_conv_modules_4_momentum.item();  l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = None
+        item_22 = l_self_modules_conv_up1_modules_double_conv_modules_4_eps.item();  l_self_modules_conv_up1_modules_double_conv_modules_4_eps = None
+        input_32 = torch.nn.functional.batch_norm(input_31, l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_, False, item_21, item_22);  input_31 = l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ = item_21 = item_22 = None
+        input_33 = torch.nn.functional.silu(input_32, inplace = True);  input_32 = None
+        item_23 = l_self_modules_up2_scale_factor.item();  l_self_modules_up2_scale_factor = None
+        x_3 = torch.nn.functional.interpolate(input_33, None, item_23, 'bilinear', True, recompute_scale_factor = None);  input_33 = item_23 = None
+        x_4 = torch.cat([x_3, input_13], dim = 1);  x_3 = input_13 = None
+        input_34 = torch.conv2d(x_4, l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  x_4 = l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = None
+        item_24 = l_self_modules_conv_up2_modules_double_conv_modules_1_momentum.item();  l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = None
+        item_25 = l_self_modules_conv_up2_modules_double_conv_modules_1_eps.item();  l_self_modules_conv_up2_modules_double_conv_modules_1_eps = None
+        input_35 = torch.nn.functional.batch_norm(input_34, l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_, False, item_24, item_25);  input_34 = l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ = item_24 = item_25 = None
+        input_36 = torch.nn.functional.silu(input_35, inplace = True);  input_35 = None
+        input_37 = torch.conv2d(input_36, l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_36 = l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = None
+        item_26 = l_self_modules_conv_up2_modules_double_conv_modules_4_momentum.item();  l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = None
+        item_27 = l_self_modules_conv_up2_modules_double_conv_modules_4_eps.item();  l_self_modules_conv_up2_modules_double_conv_modules_4_eps = None
+        input_38 = torch.nn.functional.batch_norm(input_37, l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_, False, item_26, item_27);  input_37 = l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ = item_26 = item_27 = None
+        input_39 = torch.nn.functional.silu(input_38, inplace = True);  input_38 = None
+        fluc_out = torch.conv2d(input_39, l_self_modules_out_conv_parameters_weight_, l_self_modules_out_conv_parameters_bias_, (1, 1), (0, 0), (1, 1), 1);  input_39 = l_self_modules_out_conv_parameters_weight_ = l_self_modules_out_conv_parameters_bias_ = None
+        add_2 = fluc_out + base_flow;  fluc_out = base_flow = None
+        return (add_2,)
+        
\ No newline at end of file
diff --git a/samples/pigu_hybrid/weight_meta.py b/samples/pigu_hybrid/weight_meta.py
new file mode 100644
index 000000000..676c00349
--- /dev/null
+++ b/samples/pigu_hybrid/weight_meta.py
@@ -0,0 +1,1061 @@
+class Program_weight_tensor_meta_s10:
+	name = "s10"
+	shape = []
+	dtype = "torch.int64"
+	device = "cpu"
+	mean = None
+	std = None
+	data = [4]
+
+class Program_weight_tensor_meta_L_sensor_vals_:
+	name = "L_sensor_vals_"
+	shape = [1, 65, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.023
+	std = 0.933
+	data = None
+
+class Program_weight_tensor_meta_L_sensor_coords_:
+	name = "L_sensor_coords_"
+	shape = [1, 65, 2]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.012
+	std = 0.964
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_"
+	shape = [64, 5]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.013
+	std = 0.249
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.003
+	std = 0.258
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_"
+	shape = [64, 64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.002
+	std = 0.072
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_:
+	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.002
+	std = 0.067
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_:
+	name = "L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_"
+	shape = [192]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_:
+	name = "L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_"
+	shape = [192, 64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.001
+	std = 0.088
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_:
+	name = "L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_"
+	shape = [64, 64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.072
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_:
+	name = "L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_dropout:
+	name = "L_self_modules_sensor_encoder_modules_attn_dropout"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000000]
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_norm_parameters_weight_:
+	name = "L_self_modules_sensor_encoder_modules_norm_parameters_weight_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_norm_parameters_bias_:
+	name = "L_self_modules_sensor_encoder_modules_norm_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_norm_eps:
+	name = "L_self_modules_sensor_encoder_modules_norm_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_:
+	name = "L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_"
+	shape = [64, 2]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.072
+	std = 0.392
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_:
+	name = "L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.033
+	std = 0.412
+	data = None
+
+class Program_weight_tensor_meta_L_grid_coords_:
+	name = "L_grid_coords_"
+	shape = [1, 128, 256, 2]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.008
+	std = 1.002
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_:
+	name = "L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_"
+	shape = [128, 64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.001
+	std = 0.072
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_:
+	name = "L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.006
+	std = 0.075
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_transition_x:
+	name = "L_self_modules_projector_transition_x"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.250
+	std = 0.000
+	data = [0.250000]
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_:
+	name = "L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_"
+	shape = [32, 2]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.071
+	std = 0.406
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_:
+	name = "L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_"
+	shape = [32]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.044
+	std = 0.406
+	data = [-0.109570, -0.500612, 0.499090, -0.595455, -0.314464, 0.311158, 0.427396, -0.394249, 0.155079, 0.525865, 0.152357, 0.517181, 0.421748, -0.172479, 0.703390, -0.336485, -0.533521, 0.561451, -0.191133, -0.205178, -0.142233, -0.455833, -0.541973, -0.376394, -0.008022, 0.528279, 0.360097, 0.309068, 0.316133, -0.304521, 0.450108, 0.365386]
+
+class Program_weight_tensor_meta_G_np_pi:
+	name = "G_np_pi"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 3.142
+	std = 0.000
+	data = [3.141593]
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_sensor_embedding_buffers_B_high_:
+	name = "L_self_modules_projector_modules_sensor_embedding_buffers_B_high_"
+	shape = [2, 32]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.810
+	std = 11.562
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_smooth_parameters_weight_:
+	name = "L_self_modules_projector_modules_key_proj_smooth_parameters_weight_"
+	shape = [64, 32]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.004
+	std = 0.101
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_smooth_parameters_bias_:
+	name = "L_self_modules_projector_modules_key_proj_smooth_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.008
+	std = 0.097
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_detail_parameters_weight_:
+	name = "L_self_modules_projector_modules_key_proj_detail_parameters_weight_"
+	shape = [64, 64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.001
+	std = 0.072
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_detail_parameters_bias_:
+	name = "L_self_modules_projector_modules_key_proj_detail_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.075
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_projector_scale:
+	name = "L_self_modules_projector_scale"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.125
+	std = 0.000
+	data = [0.125000]
+
+class Program_weight_tensor_meta_L_base_flow_:
+	name = "L_base_flow_"
+	shape = [1, 3, 128, 256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.002
+	std = 0.998
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_"
+	shape = [64, 133, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.001
+	std = 0.016
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_momentum:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_1_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_eps:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_1_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_"
+	shape = [64, 64, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.024
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.001
+	std = 0.022
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_momentum:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_4_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_eps:
+	name = "L_self_modules_in_conv_modules_double_conv_modules_4_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_"
+	shape = [128, 64, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.024
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.001
+	std = 0.024
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_"
+	shape = [128, 128, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps:
+	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_"
+	shape = [256, 128, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.001
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_"
+	shape = [256, 256, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.012
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.002
+	std = 0.012
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_"
+	shape = [256]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps:
+	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_up1_scale_factor:
+	name = "L_self_modules_up1_scale_factor"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 2.000
+	std = 0.000
+	data = [2.000000]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_"
+	shape = [128, 384, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.010
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.001
+	std = 0.011
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_momentum:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_eps:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_"
+	shape = [128, 128, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.001
+	std = 0.017
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_momentum:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_"
+	shape = [128]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_eps:
+	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_up2_scale_factor:
+	name = "L_self_modules_up2_scale_factor"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 2.000
+	std = 0.000
+	data = [2.000000]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_"
+	shape = [64, 192, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.000
+	std = 0.014
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = -0.001
+	std = 0.014
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_momentum:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_eps:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_"
+	shape = [64, 64, 3, 3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.024
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.022
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_momentum:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_momentum"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.100
+	std = 0.000
+	data = [0.100000]
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 1.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_"
+	shape = [64]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_eps:
+	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_eps"
+	shape = []
+	dtype = "torch.float64"
+	device = "cpu"
+	mean = 0.000
+	std = 0.000
+	data = [0.000010]
+
+class Program_weight_tensor_meta_L_self_modules_out_conv_parameters_weight_:
+	name = "L_self_modules_out_conv_parameters_weight_"
+	shape = [3, 64, 1, 1]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.001
+	std = 0.073
+	data = None
+
+class Program_weight_tensor_meta_L_self_modules_out_conv_parameters_bias_:
+	name = "L_self_modules_out_conv_parameters_bias_"
+	shape = [3]
+	dtype = "torch.float32"
+	device = "cpu"
+	mean = 0.019
+	std = 0.047
+	data = [0.064352, -0.028943, 0.022260]

From ac49bb595409988249aff2980b6b2ff66b9193ff Mon Sep 17 00:00:00 2001
From: inaniloquentee <3051000145@qq.com>
Date: Sun, 15 Feb 2026 11:06:20 +0800
Subject: [PATCH 2/3] style: manually format code with black

---
 samples/pigu_hybrid/model.py       | 1172 +++++++++++++++---
 samples/pigu_hybrid/weight_meta.py | 1822 +++++++++++++++-------------
 2 files changed, 1975 insertions(+), 1019 deletions(-)

diff --git a/samples/pigu_hybrid/model.py b/samples/pigu_hybrid/model.py
index 84f97b558..2f52bca76 100644
--- a/samples/pigu_hybrid/model.py
+++ b/samples/pigu_hybrid/model.py
@@ -1,237 +1,1023 @@
 import torch
 
+
 class GraphModule(torch.nn.Module):
-    
-    
-    
-    def forward(self, s10 : torch.SymInt, L_sensor_vals_ : torch.Tensor, L_sensor_coords_ : torch.Tensor, L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps : torch.Tensor, L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_attn_dropout : torch.Tensor, L_self_modules_sensor_encoder_modules_norm_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_norm_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_sensor_encoder_modules_norm_eps : torch.Tensor, L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_grid_coords_ : torch.Tensor, L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_projector_transition_x : torch.Tensor, L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ : torch.nn.parameter.Parameter, G_np_pi : torch.Tensor, L_self_modules_projector_modules_sensor_embedding_buffers_B_high_ : torch.Tensor, L_self_modules_projector_modules_key_proj_smooth_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_key_proj_smooth_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_key_proj_detail_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_projector_modules_key_proj_detail_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_projector_scale : torch.Tensor, L_base_flow_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_in_conv_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_up1_scale_factor : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up1_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_up2_scale_factor : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_1_momentum : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_1_eps : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_4_momentum : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ : torch.Tensor, L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ : torch.nn.parameter.Parameter, L_self_modules_conv_up2_modules_double_conv_modules_4_eps : torch.Tensor, L_self_modules_out_conv_parameters_weight_ : torch.nn.parameter.Parameter, L_self_modules_out_conv_parameters_bias_ : torch.nn.parameter.Parameter):
+
+    def forward(
+        self,
+        s10: torch.SymInt,
+        L_sensor_vals_: torch.Tensor,
+        L_sensor_coords_: torch.Tensor,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps: torch.Tensor,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_attn_dropout: torch.Tensor,
+        L_self_modules_sensor_encoder_modules_norm_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_norm_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_sensor_encoder_modules_norm_eps: torch.Tensor,
+        L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_grid_coords_: torch.Tensor,
+        L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_transition_x: torch.Tensor,
+        L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_: torch.nn.parameter.Parameter,
+        G_np_pi: torch.Tensor,
+        L_self_modules_projector_modules_sensor_embedding_buffers_B_high_: torch.Tensor,
+        L_self_modules_projector_modules_key_proj_smooth_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_modules_key_proj_smooth_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_modules_key_proj_detail_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_modules_key_proj_detail_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_projector_scale: torch.Tensor,
+        L_base_flow_: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_1_momentum: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_1_eps: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_4_momentum: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_: torch.Tensor,
+        L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_in_conv_modules_double_conv_modules_4_eps: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_: torch.Tensor,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_: torch.Tensor,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps: torch.Tensor,
+        L_self_modules_up1_scale_factor: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_1_momentum: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_1_eps: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_4_momentum: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_: torch.Tensor,
+        L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up1_modules_double_conv_modules_4_eps: torch.Tensor,
+        L_self_modules_up2_scale_factor: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_1_momentum: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_1_eps: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_4_momentum: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_: torch.Tensor,
+        L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_: torch.nn.parameter.Parameter,
+        L_self_modules_conv_up2_modules_double_conv_modules_4_eps: torch.Tensor,
+        L_self_modules_out_conv_parameters_weight_: torch.nn.parameter.Parameter,
+        L_self_modules_out_conv_parameters_bias_: torch.nn.parameter.Parameter,
+    ):
         l_sensor_vals_ = L_sensor_vals_
         l_sensor_coords_ = L_sensor_coords_
-        l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ = L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_
-        l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ = L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_
-        l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ = L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_
-        l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_
-        l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps = L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps
-        l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ = L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_
-        l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ = L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_
-        l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_
-        l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ = L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_
+        l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_
+        )
+        l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_
+        )
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_
+        )
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_
+        )
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps
+        )
+        l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_
+        )
+        l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ = (
+            L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_
+        )
+        l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = (
+            L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_
+        )
+        l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ = (
+            L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_
+        )
         l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ = L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_
-        l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ = L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_
-        l_self_modules_sensor_encoder_modules_attn_dropout = L_self_modules_sensor_encoder_modules_attn_dropout
-        l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = L_self_modules_sensor_encoder_modules_norm_parameters_weight_
-        l_self_modules_sensor_encoder_modules_norm_parameters_bias_ = L_self_modules_sensor_encoder_modules_norm_parameters_bias_
-        l_self_modules_sensor_encoder_modules_norm_eps = L_self_modules_sensor_encoder_modules_norm_eps
-        l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ = L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_
-        l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ = L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_
+        l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ = (
+            L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_
+        )
+        l_self_modules_sensor_encoder_modules_attn_dropout = (
+            L_self_modules_sensor_encoder_modules_attn_dropout
+        )
+        l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = (
+            L_self_modules_sensor_encoder_modules_norm_parameters_weight_
+        )
+        l_self_modules_sensor_encoder_modules_norm_parameters_bias_ = (
+            L_self_modules_sensor_encoder_modules_norm_parameters_bias_
+        )
+        l_self_modules_sensor_encoder_modules_norm_eps = (
+            L_self_modules_sensor_encoder_modules_norm_eps
+        )
+        l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ = (
+            L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_
+        )
+        l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ = (
+            L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_
+        )
         l_grid_coords_ = L_grid_coords_
-        l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ = L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_
-        l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ = L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_
+        l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ = (
+            L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_
+        )
+        l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ = (
+            L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_
+        )
         l_self_modules_projector_transition_x = L_self_modules_projector_transition_x
         l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_
         l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ = L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_
         g_np_pi = G_np_pi
-        l_self_modules_projector_modules_sensor_embedding_buffers_b_high_ = L_self_modules_projector_modules_sensor_embedding_buffers_B_high_
-        l_self_modules_projector_modules_key_proj_smooth_parameters_weight_ = L_self_modules_projector_modules_key_proj_smooth_parameters_weight_
-        l_self_modules_projector_modules_key_proj_smooth_parameters_bias_ = L_self_modules_projector_modules_key_proj_smooth_parameters_bias_
-        l_self_modules_projector_modules_key_proj_detail_parameters_weight_ = L_self_modules_projector_modules_key_proj_detail_parameters_weight_
-        l_self_modules_projector_modules_key_proj_detail_parameters_bias_ = L_self_modules_projector_modules_key_proj_detail_parameters_bias_
+        l_self_modules_projector_modules_sensor_embedding_buffers_b_high_ = (
+            L_self_modules_projector_modules_sensor_embedding_buffers_B_high_
+        )
+        l_self_modules_projector_modules_key_proj_smooth_parameters_weight_ = (
+            L_self_modules_projector_modules_key_proj_smooth_parameters_weight_
+        )
+        l_self_modules_projector_modules_key_proj_smooth_parameters_bias_ = (
+            L_self_modules_projector_modules_key_proj_smooth_parameters_bias_
+        )
+        l_self_modules_projector_modules_key_proj_detail_parameters_weight_ = (
+            L_self_modules_projector_modules_key_proj_detail_parameters_weight_
+        )
+        l_self_modules_projector_modules_key_proj_detail_parameters_bias_ = (
+            L_self_modules_projector_modules_key_proj_detail_parameters_bias_
+        )
         l_self_modules_projector_scale = L_self_modules_projector_scale
         l_base_flow_ = L_base_flow_
-        l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_
-        l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_
-        l_self_modules_in_conv_modules_double_conv_modules_1_momentum = L_self_modules_in_conv_modules_double_conv_modules_1_momentum
-        l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_
-        l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_
-        l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_
-        l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_
-        l_self_modules_in_conv_modules_double_conv_modules_1_eps = L_self_modules_in_conv_modules_double_conv_modules_1_eps
-        l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_
-        l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_
-        l_self_modules_in_conv_modules_double_conv_modules_4_momentum = L_self_modules_in_conv_modules_double_conv_modules_4_momentum
-        l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_
-        l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_
-        l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_
-        l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_
-        l_self_modules_in_conv_modules_double_conv_modules_4_eps = L_self_modules_in_conv_modules_double_conv_modules_4_eps
+        l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_1_momentum = (
+            L_self_modules_in_conv_modules_double_conv_modules_1_momentum
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_1_eps = (
+            L_self_modules_in_conv_modules_double_conv_modules_1_eps
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_4_momentum = (
+            L_self_modules_in_conv_modules_double_conv_modules_4_momentum
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ = (
+            L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_
+        )
+        l_self_modules_in_conv_modules_double_conv_modules_4_eps = (
+            L_self_modules_in_conv_modules_double_conv_modules_4_eps
+        )
         l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_
         l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_
-        l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum = L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum = (
+            L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum
+        )
         l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_
         l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_
         l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_
         l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_
-        l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps = L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps = (
+            L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps
+        )
         l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_
         l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_
-        l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum = L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum = (
+            L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum
+        )
         l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_
         l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_
         l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_
         l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_
-        l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps = L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps = (
+            L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps
+        )
         l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_
         l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_
-        l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum = L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum = (
+            L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum
+        )
         l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_
         l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_
         l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_
         l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_
-        l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps = L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps = (
+            L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps
+        )
         l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_
         l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_
-        l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum = L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum = (
+            L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum
+        )
         l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_
         l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_
         l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_
         l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_
-        l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps = L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps = (
+            L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps
+        )
         l_self_modules_up1_scale_factor = L_self_modules_up1_scale_factor
-        l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_
-        l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_
-        l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = L_self_modules_conv_up1_modules_double_conv_modules_1_momentum
-        l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_
-        l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_
-        l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_
-        l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_
-        l_self_modules_conv_up1_modules_double_conv_modules_1_eps = L_self_modules_conv_up1_modules_double_conv_modules_1_eps
-        l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_
-        l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_
-        l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = L_self_modules_conv_up1_modules_double_conv_modules_4_momentum
-        l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_
-        l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_
-        l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_
-        l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_
-        l_self_modules_conv_up1_modules_double_conv_modules_4_eps = L_self_modules_conv_up1_modules_double_conv_modules_4_eps
+        l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = (
+            L_self_modules_conv_up1_modules_double_conv_modules_1_momentum
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_1_eps = (
+            L_self_modules_conv_up1_modules_double_conv_modules_1_eps
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = (
+            L_self_modules_conv_up1_modules_double_conv_modules_4_momentum
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ = (
+            L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_
+        )
+        l_self_modules_conv_up1_modules_double_conv_modules_4_eps = (
+            L_self_modules_conv_up1_modules_double_conv_modules_4_eps
+        )
         l_self_modules_up2_scale_factor = L_self_modules_up2_scale_factor
-        l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_
-        l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_
-        l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = L_self_modules_conv_up2_modules_double_conv_modules_1_momentum
-        l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ = L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_
-        l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ = L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_
-        l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_
-        l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_
-        l_self_modules_conv_up2_modules_double_conv_modules_1_eps = L_self_modules_conv_up2_modules_double_conv_modules_1_eps
-        l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_
-        l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_
-        l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = L_self_modules_conv_up2_modules_double_conv_modules_4_momentum
-        l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ = L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_
-        l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ = L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_
-        l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_
-        l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ = L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_
-        l_self_modules_conv_up2_modules_double_conv_modules_4_eps = L_self_modules_conv_up2_modules_double_conv_modules_4_eps
-        l_self_modules_out_conv_parameters_weight_ = L_self_modules_out_conv_parameters_weight_
-        l_self_modules_out_conv_parameters_bias_ = L_self_modules_out_conv_parameters_bias_
-        node_in = torch.cat([l_sensor_vals_, l_sensor_coords_], dim = -1);  l_sensor_vals_ = None
-        input_1 = torch.nn.functional.linear(node_in, l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_, l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_);  node_in = l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_ = l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ = None
-        item = l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps.item();  l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps = None
-        input_2 = torch.nn.functional.layer_norm(input_1, (64,), l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_, l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_, item);  input_1 = l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_ = l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = item = None
-        input_3 = torch.nn.functional.silu(input_2, inplace = False);  input_2 = None
-        input_4 = torch.nn.functional.linear(input_3, l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_, l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_);  input_3 = l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_ = l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ = None
+        l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = (
+            L_self_modules_conv_up2_modules_double_conv_modules_1_momentum
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_1_eps = (
+            L_self_modules_conv_up2_modules_double_conv_modules_1_eps
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = (
+            L_self_modules_conv_up2_modules_double_conv_modules_4_momentum
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ = (
+            L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_
+        )
+        l_self_modules_conv_up2_modules_double_conv_modules_4_eps = (
+            L_self_modules_conv_up2_modules_double_conv_modules_4_eps
+        )
+        l_self_modules_out_conv_parameters_weight_ = (
+            L_self_modules_out_conv_parameters_weight_
+        )
+        l_self_modules_out_conv_parameters_bias_ = (
+            L_self_modules_out_conv_parameters_bias_
+        )
+        node_in = torch.cat([l_sensor_vals_, l_sensor_coords_], dim=-1)
+        l_sensor_vals_ = None
+        input_1 = torch.nn.functional.linear(
+            node_in,
+            l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_,
+            l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_,
+        )
+        node_in = (
+            l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_
+        ) = l_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_ = None
+        item = l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps.item()
+        l_self_modules_sensor_encoder_modules_mlp1_modules_1_eps = None
+        input_2 = torch.nn.functional.layer_norm(
+            input_1,
+            (64,),
+            l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_,
+            l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_,
+            item,
+        )
+        input_1 = (
+            l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_
+        ) = l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = (
+            item
+        ) = None
+        input_3 = torch.nn.functional.silu(input_2, inplace=False)
+        input_2 = None
+        input_4 = torch.nn.functional.linear(
+            input_3,
+            l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_,
+            l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_,
+        )
+        input_3 = (
+            l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_
+        ) = l_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_ = None
         query = input_4.transpose(1, 0)
-        item_1 = l_self_modules_sensor_encoder_modules_attn_dropout.item();  l_self_modules_sensor_encoder_modules_attn_dropout = None
-        multi_head_attention_forward = torch.nn.functional.multi_head_attention_forward(query, query, query, 64, 4, l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_, l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_, None, None, False, item_1, l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_, l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_, training = False, key_padding_mask = None, need_weights = True, attn_mask = None, average_attn_weights = True, is_causal = False);  query = l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_ = l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = item_1 = l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ = l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_ = None
-        attn_output = multi_head_attention_forward[0];  multi_head_attention_forward = None
-        h_attn = attn_output.transpose(1, 0);  attn_output = None
-        add = input_4 + h_attn;  input_4 = h_attn = None
-        item_2 = l_self_modules_sensor_encoder_modules_norm_eps.item();  l_self_modules_sensor_encoder_modules_norm_eps = None
-        h = torch.nn.functional.layer_norm(add, (64,), l_self_modules_sensor_encoder_modules_norm_parameters_weight_, l_self_modules_sensor_encoder_modules_norm_parameters_bias_, item_2);  add = l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = l_self_modules_sensor_encoder_modules_norm_parameters_bias_ = item_2 = None
-        global_context = torch.mean(h, dim = 1)
-        input_5 = torch.nn.functional.linear(l_grid_coords_, l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_, l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_);  l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ = l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_ = None
-        input_6 = torch.nn.functional.silu(input_5, inplace = False);  input_5 = None
-        input_7 = torch.nn.functional.linear(input_6, l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_, l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_);  input_6 = l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_ = l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ = None
+        item_1 = l_self_modules_sensor_encoder_modules_attn_dropout.item()
+        l_self_modules_sensor_encoder_modules_attn_dropout = None
+        multi_head_attention_forward = torch.nn.functional.multi_head_attention_forward(
+            query,
+            query,
+            query,
+            64,
+            4,
+            l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_,
+            l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_,
+            None,
+            None,
+            False,
+            item_1,
+            l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_,
+            l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_,
+            training=False,
+            key_padding_mask=None,
+            need_weights=True,
+            attn_mask=None,
+            average_attn_weights=True,
+            is_causal=False,
+        )
+        query = (
+            l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_
+        ) = l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = (
+            item_1
+        ) = (
+            l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_
+        ) = (
+            l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_
+        ) = None
+        attn_output = multi_head_attention_forward[0]
+        multi_head_attention_forward = None
+        h_attn = attn_output.transpose(1, 0)
+        attn_output = None
+        add = input_4 + h_attn
+        input_4 = h_attn = None
+        item_2 = l_self_modules_sensor_encoder_modules_norm_eps.item()
+        l_self_modules_sensor_encoder_modules_norm_eps = None
+        h = torch.nn.functional.layer_norm(
+            add,
+            (64,),
+            l_self_modules_sensor_encoder_modules_norm_parameters_weight_,
+            l_self_modules_sensor_encoder_modules_norm_parameters_bias_,
+            item_2,
+        )
+        add = l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = (
+            l_self_modules_sensor_encoder_modules_norm_parameters_bias_
+        ) = item_2 = None
+        global_context = torch.mean(h, dim=1)
+        input_5 = torch.nn.functional.linear(
+            l_grid_coords_,
+            l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_,
+            l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_,
+        )
+        l_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_ = (
+            l_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_
+        ) = None
+        input_6 = torch.nn.functional.silu(input_5, inplace=False)
+        input_5 = None
+        input_7 = torch.nn.functional.linear(
+            input_6,
+            l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_,
+            l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_,
+        )
+        input_6 = (
+            l_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_
+        ) = l_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_ = None
         Q_smooth = input_7[(Ellipsis, slice(None, 64, None))]
-        Q_detail = input_7[(Ellipsis, slice(64, None, None))];  input_7 = None
+        Q_detail = input_7[(Ellipsis, slice(64, None, None))]
+        input_7 = None
         grid_x = l_grid_coords_[(Ellipsis, slice(0, 1, None))]
-        item_3 = l_self_modules_projector_transition_x.item();  l_self_modules_projector_transition_x = None
-        sub = grid_x - item_3;  grid_x = item_3 = None
-        mul = 50.0 * sub;  sub = None
-        beta = torch.sigmoid(mul);  mul = None
-        Q_detail_1 = Q_detail * beta;  Q_detail = beta = None
-        linear_4 = torch.nn.functional.linear(l_sensor_coords_, l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_, l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_);  l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ = None
-        feat_smooth = torch.nn.functional.silu(linear_4);  linear_4 = None
-        item_4 = g_np_pi.item();  g_np_pi = None
-        mul_2 = 2 * item_4;  item_4 = None
-        mul_3 = mul_2 * l_sensor_coords_;  mul_2 = l_sensor_coords_ = None
-        proj_high = mul_3 @ l_self_modules_projector_modules_sensor_embedding_buffers_b_high_;  mul_3 = l_self_modules_projector_modules_sensor_embedding_buffers_b_high_ = None
+        item_3 = l_self_modules_projector_transition_x.item()
+        l_self_modules_projector_transition_x = None
+        sub = grid_x - item_3
+        grid_x = item_3 = None
+        mul = 50.0 * sub
+        sub = None
+        beta = torch.sigmoid(mul)
+        mul = None
+        Q_detail_1 = Q_detail * beta
+        Q_detail = beta = None
+        linear_4 = torch.nn.functional.linear(
+            l_sensor_coords_,
+            l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_,
+            l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_,
+        )
+        l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = (
+            l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_
+        ) = None
+        feat_smooth = torch.nn.functional.silu(linear_4)
+        linear_4 = None
+        item_4 = g_np_pi.item()
+        g_np_pi = None
+        mul_2 = 2 * item_4
+        item_4 = None
+        mul_3 = mul_2 * l_sensor_coords_
+        mul_2 = l_sensor_coords_ = None
+        proj_high = (
+            mul_3 @ l_self_modules_projector_modules_sensor_embedding_buffers_b_high_
+        )
+        mul_3 = l_self_modules_projector_modules_sensor_embedding_buffers_b_high_ = None
         sin = torch.sin(proj_high)
-        cos = torch.cos(proj_high);  proj_high = None
-        feat_detail = torch.cat([sin, cos], dim = -1);  sin = cos = None
-        K_smooth = torch.nn.functional.linear(feat_smooth, l_self_modules_projector_modules_key_proj_smooth_parameters_weight_, l_self_modules_projector_modules_key_proj_smooth_parameters_bias_);  feat_smooth = l_self_modules_projector_modules_key_proj_smooth_parameters_weight_ = l_self_modules_projector_modules_key_proj_smooth_parameters_bias_ = None
-        K_detail = torch.nn.functional.linear(feat_detail, l_self_modules_projector_modules_key_proj_detail_parameters_weight_, l_self_modules_projector_modules_key_proj_detail_parameters_bias_);  feat_detail = l_self_modules_projector_modules_key_proj_detail_parameters_weight_ = l_self_modules_projector_modules_key_proj_detail_parameters_bias_ = None
-        score_smooth = torch.functional.einsum('bhwc,bsc->bhws', Q_smooth, K_smooth);  Q_smooth = K_smooth = None
-        score_detail = torch.functional.einsum('bhwc,bsc->bhws', Q_detail_1, K_detail);  Q_detail_1 = K_detail = None
-        add_1 = score_smooth + score_detail;  score_smooth = score_detail = None
-        item_5 = l_self_modules_projector_scale.item();  l_self_modules_projector_scale = None
-        attn_logits = add_1 * item_5;  add_1 = item_5 = None
-        attn = torch.nn.functional.softmax(attn_logits, dim = -1);  attn_logits = None
-        out = torch.functional.einsum('bhws,bsc->bhwc', attn, h);  attn = h = None
-        grid_feats = out.permute(0, 3, 1, 2);  out = None
-        view = global_context.view(1, 64, 1, 1);  global_context = None
-        global_map = view.expand(1, 64, 128, 256);  view = None
-        coords_map = l_grid_coords_.permute(0, 3, 1, 2);  l_grid_coords_ = None
-        coords_map_1 = torch.nn.functional.interpolate(coords_map, size = (128, 256), mode = 'bilinear', align_corners = True);  coords_map = None
-        base_flow = torch.nn.functional.interpolate(l_base_flow_, size = (128, 256), mode = 'bilinear', align_corners = True);  l_base_flow_ = None
-        x = torch.cat([grid_feats, coords_map_1, global_map, base_flow], dim = 1);  grid_feats = coords_map_1 = global_map = None
-        input_8 = torch.conv2d(x, l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  x = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ = None
-        item_6 = l_self_modules_in_conv_modules_double_conv_modules_1_momentum.item();  l_self_modules_in_conv_modules_double_conv_modules_1_momentum = None
-        item_7 = l_self_modules_in_conv_modules_double_conv_modules_1_eps.item();  l_self_modules_in_conv_modules_double_conv_modules_1_eps = None
-        input_9 = torch.nn.functional.batch_norm(input_8, l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_, False, item_6, item_7);  input_8 = l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_ = item_6 = item_7 = None
-        input_10 = torch.nn.functional.silu(input_9, inplace = True);  input_9 = None
-        input_11 = torch.conv2d(input_10, l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_10 = l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ = None
-        item_8 = l_self_modules_in_conv_modules_double_conv_modules_4_momentum.item();  l_self_modules_in_conv_modules_double_conv_modules_4_momentum = None
-        item_9 = l_self_modules_in_conv_modules_double_conv_modules_4_eps.item();  l_self_modules_in_conv_modules_double_conv_modules_4_eps = None
-        input_12 = torch.nn.functional.batch_norm(input_11, l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_, l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_, False, item_8, item_9);  input_11 = l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_ = item_8 = item_9 = None
-        input_13 = torch.nn.functional.silu(input_12, inplace = True);  input_12 = None
-        input_14 = torch.nn.functional.max_pool2d(input_13, 2, 2, 0, 1, ceil_mode = False, return_indices = False)
-        input_15 = torch.conv2d(input_14, l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_14 = l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ = None
-        item_10 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum = None
-        item_11 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps = None
-        input_16 = torch.nn.functional.batch_norm(input_15, l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_, False, item_10, item_11);  input_15 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ = item_10 = item_11 = None
-        input_17 = torch.nn.functional.silu(input_16, inplace = True);  input_16 = None
-        input_18 = torch.conv2d(input_17, l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_17 = l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ = None
-        item_12 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum = None
-        item_13 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps.item();  l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps = None
-        input_19 = torch.nn.functional.batch_norm(input_18, l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_, l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_, False, item_12, item_13);  input_18 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ = item_12 = item_13 = None
-        input_20 = torch.nn.functional.silu(input_19, inplace = True);  input_19 = None
-        input_21 = torch.nn.functional.max_pool2d(input_20, 2, 2, 0, 1, ceil_mode = False, return_indices = False)
-        input_22 = torch.conv2d(input_21, l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_21 = l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ = None
-        item_14 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum = None
-        item_15 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps = None
-        input_23 = torch.nn.functional.batch_norm(input_22, l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_, False, item_14, item_15);  input_22 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ = item_14 = item_15 = None
-        input_24 = torch.nn.functional.silu(input_23, inplace = True);  input_23 = None
-        input_25 = torch.conv2d(input_24, l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_24 = l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ = None
-        item_16 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum = None
-        item_17 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps.item();  l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps = None
-        input_26 = torch.nn.functional.batch_norm(input_25, l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_, l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_, False, item_16, item_17);  input_25 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ = item_16 = item_17 = None
-        input_27 = torch.nn.functional.silu(input_26, inplace = True);  input_26 = None
-        item_18 = l_self_modules_up1_scale_factor.item();  l_self_modules_up1_scale_factor = None
-        x_1 = torch.nn.functional.interpolate(input_27, None, item_18, 'bilinear', True, recompute_scale_factor = None);  input_27 = item_18 = None
-        x_2 = torch.cat([x_1, input_20], dim = 1);  x_1 = input_20 = None
-        input_28 = torch.conv2d(x_2, l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  x_2 = l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = None
-        item_19 = l_self_modules_conv_up1_modules_double_conv_modules_1_momentum.item();  l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = None
-        item_20 = l_self_modules_conv_up1_modules_double_conv_modules_1_eps.item();  l_self_modules_conv_up1_modules_double_conv_modules_1_eps = None
-        input_29 = torch.nn.functional.batch_norm(input_28, l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_, False, item_19, item_20);  input_28 = l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_ = item_19 = item_20 = None
-        input_30 = torch.nn.functional.silu(input_29, inplace = True);  input_29 = None
-        input_31 = torch.conv2d(input_30, l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_30 = l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = None
-        item_21 = l_self_modules_conv_up1_modules_double_conv_modules_4_momentum.item();  l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = None
-        item_22 = l_self_modules_conv_up1_modules_double_conv_modules_4_eps.item();  l_self_modules_conv_up1_modules_double_conv_modules_4_eps = None
-        input_32 = torch.nn.functional.batch_norm(input_31, l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_, l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_, False, item_21, item_22);  input_31 = l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_ = item_21 = item_22 = None
-        input_33 = torch.nn.functional.silu(input_32, inplace = True);  input_32 = None
-        item_23 = l_self_modules_up2_scale_factor.item();  l_self_modules_up2_scale_factor = None
-        x_3 = torch.nn.functional.interpolate(input_33, None, item_23, 'bilinear', True, recompute_scale_factor = None);  input_33 = item_23 = None
-        x_4 = torch.cat([x_3, input_13], dim = 1);  x_3 = input_13 = None
-        input_34 = torch.conv2d(x_4, l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  x_4 = l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = None
-        item_24 = l_self_modules_conv_up2_modules_double_conv_modules_1_momentum.item();  l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = None
-        item_25 = l_self_modules_conv_up2_modules_double_conv_modules_1_eps.item();  l_self_modules_conv_up2_modules_double_conv_modules_1_eps = None
-        input_35 = torch.nn.functional.batch_norm(input_34, l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_, l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_, l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_, False, item_24, item_25);  input_34 = l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_ = item_24 = item_25 = None
-        input_36 = torch.nn.functional.silu(input_35, inplace = True);  input_35 = None
-        input_37 = torch.conv2d(input_36, l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_, (1, 1), (1, 1), (1, 1), 1);  input_36 = l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = None
-        item_26 = l_self_modules_conv_up2_modules_double_conv_modules_4_momentum.item();  l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = None
-        item_27 = l_self_modules_conv_up2_modules_double_conv_modules_4_eps.item();  l_self_modules_conv_up2_modules_double_conv_modules_4_eps = None
-        input_38 = torch.nn.functional.batch_norm(input_37, l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_, l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_, l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_, l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_, False, item_26, item_27);  input_37 = l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_ = item_26 = item_27 = None
-        input_39 = torch.nn.functional.silu(input_38, inplace = True);  input_38 = None
-        fluc_out = torch.conv2d(input_39, l_self_modules_out_conv_parameters_weight_, l_self_modules_out_conv_parameters_bias_, (1, 1), (0, 0), (1, 1), 1);  input_39 = l_self_modules_out_conv_parameters_weight_ = l_self_modules_out_conv_parameters_bias_ = None
-        add_2 = fluc_out + base_flow;  fluc_out = base_flow = None
+        cos = torch.cos(proj_high)
+        proj_high = None
+        feat_detail = torch.cat([sin, cos], dim=-1)
+        sin = cos = None
+        K_smooth = torch.nn.functional.linear(
+            feat_smooth,
+            l_self_modules_projector_modules_key_proj_smooth_parameters_weight_,
+            l_self_modules_projector_modules_key_proj_smooth_parameters_bias_,
+        )
+        feat_smooth = (
+            l_self_modules_projector_modules_key_proj_smooth_parameters_weight_
+        ) = l_self_modules_projector_modules_key_proj_smooth_parameters_bias_ = None
+        K_detail = torch.nn.functional.linear(
+            feat_detail,
+            l_self_modules_projector_modules_key_proj_detail_parameters_weight_,
+            l_self_modules_projector_modules_key_proj_detail_parameters_bias_,
+        )
+        feat_detail = (
+            l_self_modules_projector_modules_key_proj_detail_parameters_weight_
+        ) = l_self_modules_projector_modules_key_proj_detail_parameters_bias_ = None
+        score_smooth = torch.functional.einsum("bhwc,bsc->bhws", Q_smooth, K_smooth)
+        Q_smooth = K_smooth = None
+        score_detail = torch.functional.einsum("bhwc,bsc->bhws", Q_detail_1, K_detail)
+        Q_detail_1 = K_detail = None
+        add_1 = score_smooth + score_detail
+        score_smooth = score_detail = None
+        item_5 = l_self_modules_projector_scale.item()
+        l_self_modules_projector_scale = None
+        attn_logits = add_1 * item_5
+        add_1 = item_5 = None
+        attn = torch.nn.functional.softmax(attn_logits, dim=-1)
+        attn_logits = None
+        out = torch.functional.einsum("bhws,bsc->bhwc", attn, h)
+        attn = h = None
+        grid_feats = out.permute(0, 3, 1, 2)
+        out = None
+        view = global_context.view(1, 64, 1, 1)
+        global_context = None
+        global_map = view.expand(1, 64, 128, 256)
+        view = None
+        coords_map = l_grid_coords_.permute(0, 3, 1, 2)
+        l_grid_coords_ = None
+        coords_map_1 = torch.nn.functional.interpolate(
+            coords_map, size=(128, 256), mode="bilinear", align_corners=True
+        )
+        coords_map = None
+        base_flow = torch.nn.functional.interpolate(
+            l_base_flow_, size=(128, 256), mode="bilinear", align_corners=True
+        )
+        l_base_flow_ = None
+        x = torch.cat([grid_feats, coords_map_1, global_map, base_flow], dim=1)
+        grid_feats = coords_map_1 = global_map = None
+        input_8 = torch.conv2d(
+            x,
+            l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_,
+            l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        x = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = (
+            l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_
+        ) = None
+        item_6 = l_self_modules_in_conv_modules_double_conv_modules_1_momentum.item()
+        l_self_modules_in_conv_modules_double_conv_modules_1_momentum = None
+        item_7 = l_self_modules_in_conv_modules_double_conv_modules_1_eps.item()
+        l_self_modules_in_conv_modules_double_conv_modules_1_eps = None
+        input_9 = torch.nn.functional.batch_norm(
+            input_8,
+            l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_,
+            l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_,
+            l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_,
+            l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_,
+            False,
+            item_6,
+            item_7,
+        )
+        input_8 = (
+            l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_
+        ) = (
+            l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_
+        ) = l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = (
+            l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_
+        ) = item_6 = item_7 = None
+        input_10 = torch.nn.functional.silu(input_9, inplace=True)
+        input_9 = None
+        input_11 = torch.conv2d(
+            input_10,
+            l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_,
+            l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_10 = (
+            l_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_
+        ) = l_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_ = None
+        item_8 = l_self_modules_in_conv_modules_double_conv_modules_4_momentum.item()
+        l_self_modules_in_conv_modules_double_conv_modules_4_momentum = None
+        item_9 = l_self_modules_in_conv_modules_double_conv_modules_4_eps.item()
+        l_self_modules_in_conv_modules_double_conv_modules_4_eps = None
+        input_12 = torch.nn.functional.batch_norm(
+            input_11,
+            l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_,
+            l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_,
+            l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_,
+            l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_,
+            False,
+            item_8,
+            item_9,
+        )
+        input_11 = (
+            l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_
+        ) = (
+            l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_
+        ) = l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = (
+            l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_
+        ) = item_8 = item_9 = None
+        input_13 = torch.nn.functional.silu(input_12, inplace=True)
+        input_12 = None
+        input_14 = torch.nn.functional.max_pool2d(
+            input_13, 2, 2, 0, 1, ceil_mode=False, return_indices=False
+        )
+        input_15 = torch.conv2d(
+            input_14,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_14 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_
+        ) = None
+        item_10 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum.item()
+        )
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum = None
+        item_11 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps.item()
+        )
+        l_self_modules_down1_modules_1_modules_double_conv_modules_1_eps = None
+        input_16 = torch.nn.functional.batch_norm(
+            input_15,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_,
+            False,
+            item_10,
+            item_11,
+        )
+        input_15 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_
+        ) = item_10 = item_11 = None
+        input_17 = torch.nn.functional.silu(input_16, inplace=True)
+        input_16 = None
+        input_18 = torch.conv2d(
+            input_17,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_17 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_
+        ) = None
+        item_12 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum.item()
+        )
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum = None
+        item_13 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps.item()
+        )
+        l_self_modules_down1_modules_1_modules_double_conv_modules_4_eps = None
+        input_19 = torch.nn.functional.batch_norm(
+            input_18,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_,
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_,
+            False,
+            item_12,
+            item_13,
+        )
+        input_18 = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_
+        ) = (
+            l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_
+        ) = item_12 = item_13 = None
+        input_20 = torch.nn.functional.silu(input_19, inplace=True)
+        input_19 = None
+        input_21 = torch.nn.functional.max_pool2d(
+            input_20, 2, 2, 0, 1, ceil_mode=False, return_indices=False
+        )
+        input_22 = torch.conv2d(
+            input_21,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_21 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_
+        ) = None
+        item_14 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum.item()
+        )
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum = None
+        item_15 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps.item()
+        )
+        l_self_modules_down2_modules_1_modules_double_conv_modules_1_eps = None
+        input_23 = torch.nn.functional.batch_norm(
+            input_22,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_,
+            False,
+            item_14,
+            item_15,
+        )
+        input_22 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_
+        ) = item_14 = item_15 = None
+        input_24 = torch.nn.functional.silu(input_23, inplace=True)
+        input_23 = None
+        input_25 = torch.conv2d(
+            input_24,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_24 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_
+        ) = None
+        item_16 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum.item()
+        )
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum = None
+        item_17 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps.item()
+        )
+        l_self_modules_down2_modules_1_modules_double_conv_modules_4_eps = None
+        input_26 = torch.nn.functional.batch_norm(
+            input_25,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_,
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_,
+            False,
+            item_16,
+            item_17,
+        )
+        input_25 = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_
+        ) = (
+            l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_
+        ) = item_16 = item_17 = None
+        input_27 = torch.nn.functional.silu(input_26, inplace=True)
+        input_26 = None
+        item_18 = l_self_modules_up1_scale_factor.item()
+        l_self_modules_up1_scale_factor = None
+        x_1 = torch.nn.functional.interpolate(
+            input_27, None, item_18, "bilinear", True, recompute_scale_factor=None
+        )
+        input_27 = item_18 = None
+        x_2 = torch.cat([x_1, input_20], dim=1)
+        x_1 = input_20 = None
+        input_28 = torch.conv2d(
+            x_2,
+            l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_,
+            l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        x_2 = (
+            l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_
+        ) = l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = (
+            None
+        )
+        item_19 = l_self_modules_conv_up1_modules_double_conv_modules_1_momentum.item()
+        l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = None
+        item_20 = l_self_modules_conv_up1_modules_double_conv_modules_1_eps.item()
+        l_self_modules_conv_up1_modules_double_conv_modules_1_eps = None
+        input_29 = torch.nn.functional.batch_norm(
+            input_28,
+            l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_,
+            l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_,
+            l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_,
+            l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_,
+            False,
+            item_19,
+            item_20,
+        )
+        input_28 = (
+            l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_
+        ) = (
+            l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_
+        ) = l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = (
+            l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_
+        ) = item_19 = item_20 = None
+        input_30 = torch.nn.functional.silu(input_29, inplace=True)
+        input_29 = None
+        input_31 = torch.conv2d(
+            input_30,
+            l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_,
+            l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_30 = (
+            l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_
+        ) = l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = (
+            None
+        )
+        item_21 = l_self_modules_conv_up1_modules_double_conv_modules_4_momentum.item()
+        l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = None
+        item_22 = l_self_modules_conv_up1_modules_double_conv_modules_4_eps.item()
+        l_self_modules_conv_up1_modules_double_conv_modules_4_eps = None
+        input_32 = torch.nn.functional.batch_norm(
+            input_31,
+            l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_,
+            l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_,
+            l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_,
+            l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_,
+            False,
+            item_21,
+            item_22,
+        )
+        input_31 = (
+            l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_
+        ) = (
+            l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_
+        ) = l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = (
+            l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_
+        ) = item_21 = item_22 = None
+        input_33 = torch.nn.functional.silu(input_32, inplace=True)
+        input_32 = None
+        item_23 = l_self_modules_up2_scale_factor.item()
+        l_self_modules_up2_scale_factor = None
+        x_3 = torch.nn.functional.interpolate(
+            input_33, None, item_23, "bilinear", True, recompute_scale_factor=None
+        )
+        input_33 = item_23 = None
+        x_4 = torch.cat([x_3, input_13], dim=1)
+        x_3 = input_13 = None
+        input_34 = torch.conv2d(
+            x_4,
+            l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_,
+            l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        x_4 = (
+            l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_
+        ) = l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = (
+            None
+        )
+        item_24 = l_self_modules_conv_up2_modules_double_conv_modules_1_momentum.item()
+        l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = None
+        item_25 = l_self_modules_conv_up2_modules_double_conv_modules_1_eps.item()
+        l_self_modules_conv_up2_modules_double_conv_modules_1_eps = None
+        input_35 = torch.nn.functional.batch_norm(
+            input_34,
+            l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_,
+            l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_,
+            l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_,
+            l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_,
+            False,
+            item_24,
+            item_25,
+        )
+        input_34 = (
+            l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_
+        ) = (
+            l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_
+        ) = l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = (
+            l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_
+        ) = item_24 = item_25 = None
+        input_36 = torch.nn.functional.silu(input_35, inplace=True)
+        input_35 = None
+        input_37 = torch.conv2d(
+            input_36,
+            l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_,
+            l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_,
+            (1, 1),
+            (1, 1),
+            (1, 1),
+            1,
+        )
+        input_36 = (
+            l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_
+        ) = l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = (
+            None
+        )
+        item_26 = l_self_modules_conv_up2_modules_double_conv_modules_4_momentum.item()
+        l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = None
+        item_27 = l_self_modules_conv_up2_modules_double_conv_modules_4_eps.item()
+        l_self_modules_conv_up2_modules_double_conv_modules_4_eps = None
+        input_38 = torch.nn.functional.batch_norm(
+            input_37,
+            l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_,
+            l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_,
+            l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_,
+            l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_,
+            False,
+            item_26,
+            item_27,
+        )
+        input_37 = (
+            l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_
+        ) = (
+            l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_
+        ) = l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = (
+            l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_
+        ) = item_26 = item_27 = None
+        input_39 = torch.nn.functional.silu(input_38, inplace=True)
+        input_38 = None
+        fluc_out = torch.conv2d(
+            input_39,
+            l_self_modules_out_conv_parameters_weight_,
+            l_self_modules_out_conv_parameters_bias_,
+            (1, 1),
+            (0, 0),
+            (1, 1),
+            1,
+        )
+        input_39 = l_self_modules_out_conv_parameters_weight_ = (
+            l_self_modules_out_conv_parameters_bias_
+        ) = None
+        add_2 = fluc_out + base_flow
+        fluc_out = base_flow = None
         return (add_2,)
-        
\ No newline at end of file
diff --git a/samples/pigu_hybrid/weight_meta.py b/samples/pigu_hybrid/weight_meta.py
index 676c00349..08fcc32e3 100644
--- a/samples/pigu_hybrid/weight_meta.py
+++ b/samples/pigu_hybrid/weight_meta.py
@@ -1,1061 +1,1231 @@
 class Program_weight_tensor_meta_s10:
-	name = "s10"
-	shape = []
-	dtype = "torch.int64"
-	device = "cpu"
-	mean = None
-	std = None
-	data = [4]
+    name = "s10"
+    shape = []
+    dtype = "torch.int64"
+    device = "cpu"
+    mean = None
+    std = None
+    data = [4]
+
 
 class Program_weight_tensor_meta_L_sensor_vals_:
-	name = "L_sensor_vals_"
-	shape = [1, 65, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.023
-	std = 0.933
-	data = None
+    name = "L_sensor_vals_"
+    shape = [1, 65, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.023
+    std = 0.933
+    data = None
+
 
 class Program_weight_tensor_meta_L_sensor_coords_:
-	name = "L_sensor_coords_"
-	shape = [1, 65, 2]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.012
-	std = 0.964
-	data = None
+    name = "L_sensor_coords_"
+    shape = [1, 65, 2]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.012
+    std = 0.964
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_"
-	shape = [64, 5]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.013
-	std = 0.249
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_weight_"
+    shape = [64, 5]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.013
+    std = 0.249
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.003
-	std = 0.258
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_0_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.003
+    std = 0.258
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_1_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_"
-	shape = [64, 64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.002
-	std = 0.072
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_weight_"
+    shape = [64, 64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.002
+    std = 0.072
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_:
-	name = "L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.002
-	std = 0.067
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_mlp1_modules_3_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.002
+    std = 0.067
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_:
-	name = "L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_"
-	shape = [192]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_"
+    shape = [192]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_:
-	name = "L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_"
-	shape = [192, 64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.001
-	std = 0.088
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_"
+    shape = [192, 64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.001
+    std = 0.088
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_:
-	name = "L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_"
-	shape = [64, 64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.072
-	data = None
+    name = (
+        "L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_"
+    )
+    shape = [64, 64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.072
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_:
-	name = "L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = (
+        "L_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_"
+    )
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_attn_dropout:
-	name = "L_self_modules_sensor_encoder_modules_attn_dropout"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000000]
+    name = "L_self_modules_sensor_encoder_modules_attn_dropout"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000000]
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_norm_parameters_weight_:
-	name = "L_self_modules_sensor_encoder_modules_norm_parameters_weight_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_norm_parameters_weight_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_norm_parameters_bias_:
-	name = "L_self_modules_sensor_encoder_modules_norm_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_sensor_encoder_modules_norm_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_sensor_encoder_modules_norm_eps:
-	name = "L_self_modules_sensor_encoder_modules_norm_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_sensor_encoder_modules_norm_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_:
-	name = "L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_"
-	shape = [64, 2]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.072
-	std = 0.392
-	data = None
+    name = "L_self_modules_projector_modules_query_mlp_modules_0_parameters_weight_"
+    shape = [64, 2]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.072
+    std = 0.392
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_:
-	name = "L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.033
-	std = 0.412
-	data = None
+    name = "L_self_modules_projector_modules_query_mlp_modules_0_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.033
+    std = 0.412
+    data = None
+
 
 class Program_weight_tensor_meta_L_grid_coords_:
-	name = "L_grid_coords_"
-	shape = [1, 128, 256, 2]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.008
-	std = 1.002
-	data = None
+    name = "L_grid_coords_"
+    shape = [1, 128, 256, 2]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.008
+    std = 1.002
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_:
-	name = "L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_"
-	shape = [128, 64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.001
-	std = 0.072
-	data = None
+    name = "L_self_modules_projector_modules_query_mlp_modules_2_parameters_weight_"
+    shape = [128, 64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.001
+    std = 0.072
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_:
-	name = "L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.006
-	std = 0.075
-	data = None
+    name = "L_self_modules_projector_modules_query_mlp_modules_2_parameters_bias_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.006
+    std = 0.075
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_transition_x:
-	name = "L_self_modules_projector_transition_x"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.250
-	std = 0.000
-	data = [0.250000]
+    name = "L_self_modules_projector_transition_x"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.250
+    std = 0.000
+    data = [0.250000]
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_:
-	name = "L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_"
-	shape = [32, 2]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.071
-	std = 0.406
-	data = None
+    name = "L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_"
+    shape = [32, 2]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.071
+    std = 0.406
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_:
-	name = "L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_"
-	shape = [32]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.044
-	std = 0.406
-	data = [-0.109570, -0.500612, 0.499090, -0.595455, -0.314464, 0.311158, 0.427396, -0.394249, 0.155079, 0.525865, 0.152357, 0.517181, 0.421748, -0.172479, 0.703390, -0.336485, -0.533521, 0.561451, -0.191133, -0.205178, -0.142233, -0.455833, -0.541973, -0.376394, -0.008022, 0.528279, 0.360097, 0.309068, 0.316133, -0.304521, 0.450108, 0.365386]
+    name = "L_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_"
+    shape = [32]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.044
+    std = 0.406
+    data = [
+        -0.109570,
+        -0.500612,
+        0.499090,
+        -0.595455,
+        -0.314464,
+        0.311158,
+        0.427396,
+        -0.394249,
+        0.155079,
+        0.525865,
+        0.152357,
+        0.517181,
+        0.421748,
+        -0.172479,
+        0.703390,
+        -0.336485,
+        -0.533521,
+        0.561451,
+        -0.191133,
+        -0.205178,
+        -0.142233,
+        -0.455833,
+        -0.541973,
+        -0.376394,
+        -0.008022,
+        0.528279,
+        0.360097,
+        0.309068,
+        0.316133,
+        -0.304521,
+        0.450108,
+        0.365386,
+    ]
+
 
 class Program_weight_tensor_meta_G_np_pi:
-	name = "G_np_pi"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 3.142
-	std = 0.000
-	data = [3.141593]
+    name = "G_np_pi"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 3.142
+    std = 0.000
+    data = [3.141593]
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_sensor_embedding_buffers_B_high_:
-	name = "L_self_modules_projector_modules_sensor_embedding_buffers_B_high_"
-	shape = [2, 32]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.810
-	std = 11.562
-	data = None
+    name = "L_self_modules_projector_modules_sensor_embedding_buffers_B_high_"
+    shape = [2, 32]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.810
+    std = 11.562
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_smooth_parameters_weight_:
-	name = "L_self_modules_projector_modules_key_proj_smooth_parameters_weight_"
-	shape = [64, 32]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.004
-	std = 0.101
-	data = None
+    name = "L_self_modules_projector_modules_key_proj_smooth_parameters_weight_"
+    shape = [64, 32]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.004
+    std = 0.101
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_smooth_parameters_bias_:
-	name = "L_self_modules_projector_modules_key_proj_smooth_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.008
-	std = 0.097
-	data = None
+    name = "L_self_modules_projector_modules_key_proj_smooth_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.008
+    std = 0.097
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_detail_parameters_weight_:
-	name = "L_self_modules_projector_modules_key_proj_detail_parameters_weight_"
-	shape = [64, 64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.001
-	std = 0.072
-	data = None
+    name = "L_self_modules_projector_modules_key_proj_detail_parameters_weight_"
+    shape = [64, 64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.001
+    std = 0.072
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_modules_key_proj_detail_parameters_bias_:
-	name = "L_self_modules_projector_modules_key_proj_detail_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.075
-	data = None
+    name = "L_self_modules_projector_modules_key_proj_detail_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.075
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_projector_scale:
-	name = "L_self_modules_projector_scale"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.125
-	std = 0.000
-	data = [0.125000]
+    name = "L_self_modules_projector_scale"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.125
+    std = 0.000
+    data = [0.125000]
+
 
 class Program_weight_tensor_meta_L_base_flow_:
-	name = "L_base_flow_"
-	shape = [1, 3, 128, 256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.002
-	std = 0.998
-	data = None
+    name = "L_base_flow_"
+    shape = [1, 3, 128, 256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.002
+    std = 0.998
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_"
-	shape = [64, 133, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.017
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_"
+    shape = [64, 133, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.001
-	std = 0.016
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.001
+    std = 0.016
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_momentum:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_1_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_in_conv_modules_double_conv_modules_1_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_1_eps:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_1_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_in_conv_modules_double_conv_modules_1_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_"
-	shape = [64, 64, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.024
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_3_parameters_weight_"
+    shape = [64, 64, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.024
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.001
-	std = 0.022
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_3_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.001
+    std = 0.022
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_momentum:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_4_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_in_conv_modules_double_conv_modules_4_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_in_conv_modules_double_conv_modules_4_eps:
-	name = "L_self_modules_in_conv_modules_double_conv_modules_4_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_in_conv_modules_double_conv_modules_4_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_"
-	shape = [128, 64, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.024
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_"
+    shape = [128, 64, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.024
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.001
-	std = 0.024
-	data = None
+    name = (
+        "L_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_"
+    )
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.001
+    std = 0.024
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = (
+        "L_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_"
+    )
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_1_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_"
-	shape = [128, 128, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.017
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_"
+    shape = [128, 128, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.017
-	data = None
+    name = (
+        "L_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_"
+    )
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = (
+        "L_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_"
+    )
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps:
-	name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_down1_modules_1_modules_double_conv_modules_4_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_"
-	shape = [256, 128, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.017
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_"
+    shape = [256, 128, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.001
-	std = 0.017
-	data = None
+    name = (
+        "L_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_"
+    )
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.001
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_"
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_"
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_"
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = (
+        "L_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_"
+    )
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_1_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_"
-	shape = [256, 256, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.012
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_"
+    shape = [256, 256, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.012
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.002
-	std = 0.012
-	data = None
+    name = (
+        "L_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_"
+    )
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.002
+    std = 0.012
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_"
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_"
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_"
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_"
-	shape = [256]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = (
+        "L_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_"
+    )
+    shape = [256]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps:
-	name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_down2_modules_1_modules_double_conv_modules_4_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_up1_scale_factor:
-	name = "L_self_modules_up1_scale_factor"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 2.000
-	std = 0.000
-	data = [2.000000]
+    name = "L_self_modules_up1_scale_factor"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 2.000
+    std = 0.000
+    data = [2.000000]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_"
-	shape = [128, 384, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.010
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_"
+    shape = [128, 384, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.010
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.001
-	std = 0.011
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.001
+    std = 0.011
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_momentum:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_1_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_1_eps:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_1_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_1_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_"
-	shape = [128, 128, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.017
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_"
+    shape = [128, 128, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.001
-	std = 0.017
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.001
+    std = 0.017
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_momentum:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_4_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_"
-	shape = [128]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_"
+    shape = [128]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up1_modules_double_conv_modules_4_eps:
-	name = "L_self_modules_conv_up1_modules_double_conv_modules_4_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_conv_up1_modules_double_conv_modules_4_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_up2_scale_factor:
-	name = "L_self_modules_up2_scale_factor"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 2.000
-	std = 0.000
-	data = [2.000000]
+    name = "L_self_modules_up2_scale_factor"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 2.000
+    std = 0.000
+    data = [2.000000]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_"
-	shape = [64, 192, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.000
-	std = 0.014
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_"
+    shape = [64, 192, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.000
+    std = 0.014
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = -0.001
-	std = 0.014
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = -0.001
+    std = 0.014
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_momentum:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_1_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_1_eps:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_1_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_1_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_"
-	shape = [64, 64, 3, 3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.024
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_"
+    shape = [64, 64, 3, 3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.024
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.022
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.022
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_momentum:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_momentum"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.100
-	std = 0.000
-	data = [0.100000]
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_4_momentum"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.100
+    std = 0.000
+    data = [0.100000]
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 1.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 1.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_"
-	shape = [64]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = None
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_"
+    shape = [64]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_conv_up2_modules_double_conv_modules_4_eps:
-	name = "L_self_modules_conv_up2_modules_double_conv_modules_4_eps"
-	shape = []
-	dtype = "torch.float64"
-	device = "cpu"
-	mean = 0.000
-	std = 0.000
-	data = [0.000010]
+    name = "L_self_modules_conv_up2_modules_double_conv_modules_4_eps"
+    shape = []
+    dtype = "torch.float64"
+    device = "cpu"
+    mean = 0.000
+    std = 0.000
+    data = [0.000010]
+
 
 class Program_weight_tensor_meta_L_self_modules_out_conv_parameters_weight_:
-	name = "L_self_modules_out_conv_parameters_weight_"
-	shape = [3, 64, 1, 1]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.001
-	std = 0.073
-	data = None
+    name = "L_self_modules_out_conv_parameters_weight_"
+    shape = [3, 64, 1, 1]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.001
+    std = 0.073
+    data = None
+
 
 class Program_weight_tensor_meta_L_self_modules_out_conv_parameters_bias_:
-	name = "L_self_modules_out_conv_parameters_bias_"
-	shape = [3]
-	dtype = "torch.float32"
-	device = "cpu"
-	mean = 0.019
-	std = 0.047
-	data = [0.064352, -0.028943, 0.022260]
+    name = "L_self_modules_out_conv_parameters_bias_"
+    shape = [3]
+    dtype = "torch.float32"
+    device = "cpu"
+    mean = 0.019
+    std = 0.047
+    data = [0.064352, -0.028943, 0.022260]

From 2c6e07a9b4796bef9306e1030fdcac847d70b86b Mon Sep 17 00:00:00 2001
From: inaniloquentee <3051000145@qq.com>
Date: Sun, 15 Feb 2026 11:28:24 +0800
Subject: [PATCH 3/3] style: format code with black 23.1.0 matching CI

---
 samples/pigu_hybrid/model.py | 155 ++++++++++++++---------------------
 1 file changed, 62 insertions(+), 93 deletions(-)

diff --git a/samples/pigu_hybrid/model.py b/samples/pigu_hybrid/model.py
index 2f52bca76..392377529 100644
--- a/samples/pigu_hybrid/model.py
+++ b/samples/pigu_hybrid/model.py
@@ -2,7 +2,6 @@
 
 
 class GraphModule(torch.nn.Module):
-
     def forward(
         self,
         s10: torch.SymInt,
@@ -424,9 +423,9 @@ def forward(
         )
         input_1 = (
             l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_weight_
-        ) = l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_ = (
-            item
-        ) = None
+        ) = (
+            l_self_modules_sensor_encoder_modules_mlp1_modules_1_parameters_bias_
+        ) = item = None
         input_3 = torch.nn.functional.silu(input_2, inplace=False)
         input_2 = None
         input_4 = torch.nn.functional.linear(
@@ -463,11 +462,11 @@ def forward(
         )
         query = (
             l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_weight_
-        ) = l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_ = (
-            item_1
         ) = (
-            l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_
+            l_self_modules_sensor_encoder_modules_attn_parameters_in_proj_bias_
         ) = (
+            item_1
+        ) = l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_weight_ = (
             l_self_modules_sensor_encoder_modules_attn_modules_out_proj_parameters_bias_
         ) = None
         attn_output = multi_head_attention_forward[0]
@@ -485,9 +484,9 @@ def forward(
             l_self_modules_sensor_encoder_modules_norm_parameters_bias_,
             item_2,
         )
-        add = l_self_modules_sensor_encoder_modules_norm_parameters_weight_ = (
-            l_self_modules_sensor_encoder_modules_norm_parameters_bias_
-        ) = item_2 = None
+        add = (
+            l_self_modules_sensor_encoder_modules_norm_parameters_weight_
+        ) = l_self_modules_sensor_encoder_modules_norm_parameters_bias_ = item_2 = None
         global_context = torch.mean(h, dim=1)
         input_5 = torch.nn.functional.linear(
             l_grid_coords_,
@@ -526,9 +525,7 @@ def forward(
             l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_,
             l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_,
         )
-        l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = (
-            l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_
-        ) = None
+        l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_weight_ = l_self_modules_projector_modules_sensor_embedding_modules_raw_proj_parameters_bias_ = (None)
         feat_smooth = torch.nn.functional.silu(linear_4)
         linear_4 = None
         item_4 = g_np_pi.item()
@@ -603,9 +600,9 @@ def forward(
             (1, 1),
             1,
         )
-        x = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_ = (
-            l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_
-        ) = None
+        x = (
+            l_self_modules_in_conv_modules_double_conv_modules_0_parameters_weight_
+        ) = l_self_modules_in_conv_modules_double_conv_modules_0_parameters_bias_ = None
         item_6 = l_self_modules_in_conv_modules_double_conv_modules_1_momentum.item()
         l_self_modules_in_conv_modules_double_conv_modules_1_momentum = None
         item_7 = l_self_modules_in_conv_modules_double_conv_modules_1_eps.item()
@@ -624,7 +621,9 @@ def forward(
             l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_mean_
         ) = (
             l_self_modules_in_conv_modules_double_conv_modules_1_buffers_running_var_
-        ) = l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_ = (
+        ) = (
+            l_self_modules_in_conv_modules_double_conv_modules_1_parameters_weight_
+        ) = (
             l_self_modules_in_conv_modules_double_conv_modules_1_parameters_bias_
         ) = item_6 = item_7 = None
         input_10 = torch.nn.functional.silu(input_9, inplace=True)
@@ -659,7 +658,9 @@ def forward(
             l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_mean_
         ) = (
             l_self_modules_in_conv_modules_double_conv_modules_4_buffers_running_var_
-        ) = l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_ = (
+        ) = (
+            l_self_modules_in_conv_modules_double_conv_modules_4_parameters_weight_
+        ) = (
             l_self_modules_in_conv_modules_double_conv_modules_4_parameters_bias_
         ) = item_8 = item_9 = None
         input_13 = torch.nn.functional.silu(input_12, inplace=True)
@@ -676,11 +677,7 @@ def forward(
             (1, 1),
             1,
         )
-        input_14 = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_
-        ) = None
+        input_14 = l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_0_parameters_bias_ = (None)
         item_10 = (
             l_self_modules_down1_modules_1_modules_double_conv_modules_1_momentum.item()
         )
@@ -699,15 +696,9 @@ def forward(
             item_10,
             item_11,
         )
-        input_15 = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_
-        ) = item_10 = item_11 = None
+        input_15 = l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_1_parameters_bias_ = (item_10) = (
+            item_11
+        ) = None
         input_17 = torch.nn.functional.silu(input_16, inplace=True)
         input_16 = None
         input_18 = torch.conv2d(
@@ -719,11 +710,7 @@ def forward(
             (1, 1),
             1,
         )
-        input_17 = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_
-        ) = None
+        input_17 = l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_3_parameters_bias_ = (None)
         item_12 = (
             l_self_modules_down1_modules_1_modules_double_conv_modules_4_momentum.item()
         )
@@ -742,15 +729,9 @@ def forward(
             item_12,
             item_13,
         )
-        input_18 = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_
-        ) = (
-            l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_
-        ) = item_12 = item_13 = None
+        input_18 = l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_down1_modules_1_modules_double_conv_modules_4_parameters_bias_ = (item_12) = (
+            item_13
+        ) = None
         input_20 = torch.nn.functional.silu(input_19, inplace=True)
         input_19 = None
         input_21 = torch.nn.functional.max_pool2d(
@@ -765,11 +746,7 @@ def forward(
             (1, 1),
             1,
         )
-        input_21 = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_
-        ) = None
+        input_21 = l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_0_parameters_bias_ = (None)
         item_14 = (
             l_self_modules_down2_modules_1_modules_double_conv_modules_1_momentum.item()
         )
@@ -788,15 +765,9 @@ def forward(
             item_14,
             item_15,
         )
-        input_22 = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_
-        ) = item_14 = item_15 = None
+        input_22 = l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_mean_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_buffers_running_var_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_1_parameters_bias_ = (item_14) = (
+            item_15
+        ) = None
         input_24 = torch.nn.functional.silu(input_23, inplace=True)
         input_23 = None
         input_25 = torch.conv2d(
@@ -808,11 +779,7 @@ def forward(
             (1, 1),
             1,
         )
-        input_24 = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_
-        ) = None
+        input_24 = l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_3_parameters_bias_ = (None)
         item_16 = (
             l_self_modules_down2_modules_1_modules_double_conv_modules_4_momentum.item()
         )
@@ -831,15 +798,9 @@ def forward(
             item_16,
             item_17,
         )
-        input_25 = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_
-        ) = (
-            l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_
-        ) = item_16 = item_17 = None
+        input_25 = l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_mean_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_buffers_running_var_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_weight_ = l_self_modules_down2_modules_1_modules_double_conv_modules_4_parameters_bias_ = (item_16) = (
+            item_17
+        ) = None
         input_27 = torch.nn.functional.silu(input_26, inplace=True)
         input_26 = None
         item_18 = l_self_modules_up1_scale_factor.item()
@@ -861,9 +822,9 @@ def forward(
         )
         x_2 = (
             l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_weight_
-        ) = l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_ = (
-            None
-        )
+        ) = (
+            l_self_modules_conv_up1_modules_double_conv_modules_0_parameters_bias_
+        ) = None
         item_19 = l_self_modules_conv_up1_modules_double_conv_modules_1_momentum.item()
         l_self_modules_conv_up1_modules_double_conv_modules_1_momentum = None
         item_20 = l_self_modules_conv_up1_modules_double_conv_modules_1_eps.item()
@@ -882,7 +843,9 @@ def forward(
             l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_mean_
         ) = (
             l_self_modules_conv_up1_modules_double_conv_modules_1_buffers_running_var_
-        ) = l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_ = (
+        ) = (
+            l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_weight_
+        ) = (
             l_self_modules_conv_up1_modules_double_conv_modules_1_parameters_bias_
         ) = item_19 = item_20 = None
         input_30 = torch.nn.functional.silu(input_29, inplace=True)
@@ -898,9 +861,9 @@ def forward(
         )
         input_30 = (
             l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_weight_
-        ) = l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_ = (
-            None
-        )
+        ) = (
+            l_self_modules_conv_up1_modules_double_conv_modules_3_parameters_bias_
+        ) = None
         item_21 = l_self_modules_conv_up1_modules_double_conv_modules_4_momentum.item()
         l_self_modules_conv_up1_modules_double_conv_modules_4_momentum = None
         item_22 = l_self_modules_conv_up1_modules_double_conv_modules_4_eps.item()
@@ -919,7 +882,9 @@ def forward(
             l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_mean_
         ) = (
             l_self_modules_conv_up1_modules_double_conv_modules_4_buffers_running_var_
-        ) = l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_ = (
+        ) = (
+            l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_weight_
+        ) = (
             l_self_modules_conv_up1_modules_double_conv_modules_4_parameters_bias_
         ) = item_21 = item_22 = None
         input_33 = torch.nn.functional.silu(input_32, inplace=True)
@@ -943,9 +908,9 @@ def forward(
         )
         x_4 = (
             l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_weight_
-        ) = l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_ = (
-            None
-        )
+        ) = (
+            l_self_modules_conv_up2_modules_double_conv_modules_0_parameters_bias_
+        ) = None
         item_24 = l_self_modules_conv_up2_modules_double_conv_modules_1_momentum.item()
         l_self_modules_conv_up2_modules_double_conv_modules_1_momentum = None
         item_25 = l_self_modules_conv_up2_modules_double_conv_modules_1_eps.item()
@@ -964,7 +929,9 @@ def forward(
             l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_mean_
         ) = (
             l_self_modules_conv_up2_modules_double_conv_modules_1_buffers_running_var_
-        ) = l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_ = (
+        ) = (
+            l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_weight_
+        ) = (
             l_self_modules_conv_up2_modules_double_conv_modules_1_parameters_bias_
         ) = item_24 = item_25 = None
         input_36 = torch.nn.functional.silu(input_35, inplace=True)
@@ -980,9 +947,9 @@ def forward(
         )
         input_36 = (
             l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_weight_
-        ) = l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_ = (
-            None
-        )
+        ) = (
+            l_self_modules_conv_up2_modules_double_conv_modules_3_parameters_bias_
+        ) = None
         item_26 = l_self_modules_conv_up2_modules_double_conv_modules_4_momentum.item()
         l_self_modules_conv_up2_modules_double_conv_modules_4_momentum = None
         item_27 = l_self_modules_conv_up2_modules_double_conv_modules_4_eps.item()
@@ -1001,7 +968,9 @@ def forward(
             l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_mean_
         ) = (
             l_self_modules_conv_up2_modules_double_conv_modules_4_buffers_running_var_
-        ) = l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_ = (
+        ) = (
+            l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_weight_
+        ) = (
             l_self_modules_conv_up2_modules_double_conv_modules_4_parameters_bias_
         ) = item_26 = item_27 = None
         input_39 = torch.nn.functional.silu(input_38, inplace=True)
@@ -1015,9 +984,9 @@ def forward(
             (1, 1),
             1,
         )
-        input_39 = l_self_modules_out_conv_parameters_weight_ = (
-            l_self_modules_out_conv_parameters_bias_
-        ) = None
+        input_39 = (
+            l_self_modules_out_conv_parameters_weight_
+        ) = l_self_modules_out_conv_parameters_bias_ = None
         add_2 = fluc_out + base_flow
         fluc_out = base_flow = None
         return (add_2,)