diff --git a/graph_net/test/vision_model_video_test.py b/graph_net/test/vision_model_video_test.py new file mode 100644 index 000000000..168c41a89 --- /dev/null +++ b/graph_net/test/vision_model_video_test.py @@ -0,0 +1,68 @@ +import argparse +import os +import torch +import torchvision +import graph_net +import importlib + + +def load_video_model(model_name): + """ + 动态加载视频模型,兼容不同版本的 torchvision + """ + video_module = importlib.import_module("torchvision.models.video") + if hasattr(video_module, model_name): + model_builder = getattr(video_module, model_name) + if isinstance(model_builder, type): + try: + weights_enum = getattr( + torchvision.models.video, f"{model_name.upper()}_WEIGHTS", None + ) + if weights_enum and hasattr(weights_enum, "DEFAULT"): + return model_builder(weights=weights_enum.DEFAULT) + else: + return model_builder(pretrained=True) + except Exception: + print("Initialize randomly") + return model_builder() + else: + # 如果是函数,直接调用 + return model_builder(pretrained=True) + + raise ValueError(f"Not found: {model_name}") + + +def extract_visio_graph(model_name, model_path): + model = load_video_model(model_path) + model.eval() + + device = torch.device("cuda" if torch.cuda.is_available() else "cpu") + model.to(device) + + batch_size, channels, frames, height, width = 1, 3, 16, 224, 224 + random_input = torch.rand(batch_size, channels, frames, height, width) + mean = torch.tensor([0.43216, 0.394666, 0.37645]).view(1, channels, 1, 1, 1) + std = torch.tensor([0.22803, 0.22145, 0.216989]).view(1, channels, 1, 1, 1) + normalized_input = (random_input - mean) / std + + normalized_input = normalized_input.to(device) + + model = graph_net.torch.extract(name=model_name, dynamic=True)(model) + + print("Running inference...") + print("Input shape:", normalized_input.shape) + + +if __name__ == "__main__": + workspace_default = os.environ.get("GRAPH_NET_EXTRACT_WORKSPACE", "../../workspace") + + parser = argparse.ArgumentParser() + parser.add_argument("--model_name", type=str, default="mc3_18") + parser.add_argument("--model_path", type=str, default="mc3_18") + parser.add_argument("--workspace", type=str, default=workspace_default) + parser.add_argument("--dynamic", type=bool, default=True) + args = parser.parse_args() + + os.environ["GRAPH_NET_EXTRACT_WORKSPACE"] = args.workspace + + extract_visio_graph(args.model_name, args.model_path) diff --git a/samples/torchvision/mc3_18/graph_net.json b/samples/torchvision/mc3_18/graph_net.json new file mode 100644 index 000000000..00b343099 --- /dev/null +++ b/samples/torchvision/mc3_18/graph_net.json @@ -0,0 +1,7 @@ +{ + "framework": "torch", + "num_devices_required": 1, + "num_nodes_required": 1, + "dynamic": true, + "model_name": "mc3_18" +} \ No newline at end of file diff --git a/samples/torchvision/mc3_18/input_meta.py b/samples/torchvision/mc3_18/input_meta.py new file mode 100644 index 000000000..e69de29bb diff --git a/samples/torchvision/mc3_18/input_tensor_constraints.py b/samples/torchvision/mc3_18/input_tensor_constraints.py new file mode 100644 index 000000000..e69de29bb diff --git a/samples/torchvision/mc3_18/model.py b/samples/torchvision/mc3_18/model.py new file mode 100644 index 000000000..3763a473c --- /dev/null +++ b/samples/torchvision/mc3_18/model.py @@ -0,0 +1,978 @@ +import torch + + +class GraphModule(torch.nn.Module): + def forward( + self, + L_self_modules_stem_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + s1: torch.SymInt, + s2: torch.SymInt, + L_x_: torch.Tensor, + L_self_modules_stem_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_stem_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_stem_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_stem_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_: torch.Tensor, + L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_: torch.Tensor, + L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_: torch.nn.parameter.Parameter, + L_self_modules_fc_parameters_weight_: torch.nn.parameter.Parameter, + L_self_modules_fc_parameters_bias_: torch.nn.parameter.Parameter, + ): + l_self_modules_stem_modules_0_parameters_weight_ = ( + L_self_modules_stem_modules_0_parameters_weight_ + ) + l_x_ = L_x_ + l_self_modules_stem_modules_1_buffers_running_mean_ = ( + L_self_modules_stem_modules_1_buffers_running_mean_ + ) + l_self_modules_stem_modules_1_buffers_running_var_ = ( + L_self_modules_stem_modules_1_buffers_running_var_ + ) + l_self_modules_stem_modules_1_parameters_weight_ = ( + L_self_modules_stem_modules_1_parameters_weight_ + ) + l_self_modules_stem_modules_1_parameters_bias_ = ( + L_self_modules_stem_modules_1_parameters_bias_ + ) + l_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_ = L_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_ + l_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_ = L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_ + l_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_ = L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_ + l_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_ = L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_ + l_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_ = L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_ + l_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_ = L_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_ + l_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_ = L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_ + l_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_ = L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_ + l_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_ = L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_ + l_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_ = L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_ + l_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_ = L_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_ + l_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_ = L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_ + l_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_ = L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_ + l_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_ = L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_ + l_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_ = L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_ + l_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + L_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_ + ) + l_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_ = L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_ + l_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_ = ( + L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) + l_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_ = ( + L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_ + ) + l_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_ = ( + L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_ + ) + l_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_ = ( + L_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_ + ) + l_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_ = L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_ + l_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_ = ( + L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) + l_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_ = ( + L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_ + ) + l_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_ = ( + L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_ + ) + l_self_modules_fc_parameters_weight_ = L_self_modules_fc_parameters_weight_ + l_self_modules_fc_parameters_bias_ = L_self_modules_fc_parameters_bias_ + input_1 = torch.conv3d( + l_x_, + l_self_modules_stem_modules_0_parameters_weight_, + None, + (1, 2, 2), + (1, 3, 3), + (1, 1, 1), + 1, + ) + l_x_ = l_self_modules_stem_modules_0_parameters_weight_ = None + input_2 = torch.nn.functional.batch_norm( + input_1, + l_self_modules_stem_modules_1_buffers_running_mean_, + l_self_modules_stem_modules_1_buffers_running_var_, + l_self_modules_stem_modules_1_parameters_weight_, + l_self_modules_stem_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_1 = ( + l_self_modules_stem_modules_1_buffers_running_mean_ + ) = ( + l_self_modules_stem_modules_1_buffers_running_var_ + ) = ( + l_self_modules_stem_modules_1_parameters_weight_ + ) = l_self_modules_stem_modules_1_parameters_bias_ = None + input_3 = torch.nn.functional.relu(input_2, inplace=True) + input_2 = None + input_4 = torch.conv3d( + input_3, + l_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_, + None, + (1, 1, 1), + (1, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_5 = torch.nn.functional.batch_norm( + input_4, + l_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_4 = l_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_ + ) = None + input_6 = torch.nn.functional.relu(input_5, inplace=True) + input_5 = None + input_7 = torch.conv3d( + input_6, + l_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (1, 1, 1), + (1, 1, 1), + 1, + ) + input_6 = ( + l_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_ + ) = None + input_8 = torch.nn.functional.batch_norm( + input_7, + l_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_7 = l_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_ + ) = None + input_8 += input_3 + out = input_8 + input_8 = input_3 = None + out_1 = torch.nn.functional.relu(out, inplace=True) + out = None + input_9 = torch.conv3d( + out_1, + l_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_, + None, + (1, 1, 1), + (1, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_10 = torch.nn.functional.batch_norm( + input_9, + l_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_9 = l_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_ + ) = None + input_11 = torch.nn.functional.relu(input_10, inplace=True) + input_10 = None + input_12 = torch.conv3d( + input_11, + l_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (1, 1, 1), + (1, 1, 1), + 1, + ) + input_11 = ( + l_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_ + ) = None + input_13 = torch.nn.functional.batch_norm( + input_12, + l_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_12 = l_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_ + ) = None + input_13 += out_1 + out_2 = input_13 + input_13 = out_1 = None + out_3 = torch.nn.functional.relu(out_2, inplace=True) + out_2 = None + input_14 = torch.conv3d( + out_3, + l_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_, + None, + (1, 2, 2), + (0, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_15 = torch.nn.functional.batch_norm( + input_14, + l_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_14 = l_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_ + ) = None + input_16 = torch.nn.functional.relu(input_15, inplace=True) + input_15 = None + input_17 = torch.conv3d( + input_16, + l_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + input_16 = ( + l_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_ + ) = None + input_18 = torch.nn.functional.batch_norm( + input_17, + l_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_17 = l_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_ + ) = None + input_19 = torch.conv3d( + out_3, + l_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_, + None, + (1, 2, 2), + (0, 0, 0), + (1, 1, 1), + 1, + ) + out_3 = l_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_ = (None) + input_20 = torch.nn.functional.batch_norm( + input_19, + l_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_, + l_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_, + l_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_, + l_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_19 = l_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_ = l_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_ = l_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_ = l_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_ = (None) + input_18 += input_20 + out_4 = input_18 + input_18 = input_20 = None + out_5 = torch.nn.functional.relu(out_4, inplace=True) + out_4 = None + input_21 = torch.conv3d( + out_5, + l_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_22 = torch.nn.functional.batch_norm( + input_21, + l_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_21 = l_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_ + ) = None + input_23 = torch.nn.functional.relu(input_22, inplace=True) + input_22 = None + input_24 = torch.conv3d( + input_23, + l_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + input_23 = ( + l_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_ + ) = None + input_25 = torch.nn.functional.batch_norm( + input_24, + l_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_24 = l_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_ + ) = None + input_25 += out_5 + out_6 = input_25 + input_25 = out_5 = None + out_7 = torch.nn.functional.relu(out_6, inplace=True) + out_6 = None + input_26 = torch.conv3d( + out_7, + l_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_, + None, + (1, 2, 2), + (0, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_27 = torch.nn.functional.batch_norm( + input_26, + l_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_26 = l_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_ + ) = None + input_28 = torch.nn.functional.relu(input_27, inplace=True) + input_27 = None + input_29 = torch.conv3d( + input_28, + l_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + input_28 = ( + l_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_ + ) = None + input_30 = torch.nn.functional.batch_norm( + input_29, + l_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_29 = l_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_ + ) = None + input_31 = torch.conv3d( + out_7, + l_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_, + None, + (1, 2, 2), + (0, 0, 0), + (1, 1, 1), + 1, + ) + out_7 = l_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_ = (None) + input_32 = torch.nn.functional.batch_norm( + input_31, + l_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_, + l_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_, + l_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_, + l_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_31 = l_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_ = l_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_ = l_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_ = l_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_ = (None) + input_30 += input_32 + out_8 = input_30 + input_30 = input_32 = None + out_9 = torch.nn.functional.relu(out_8, inplace=True) + out_8 = None + input_33 = torch.conv3d( + out_9, + l_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_34 = torch.nn.functional.batch_norm( + input_33, + l_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_33 = l_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_ + ) = None + input_35 = torch.nn.functional.relu(input_34, inplace=True) + input_34 = None + input_36 = torch.conv3d( + input_35, + l_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + input_35 = ( + l_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_ + ) = None + input_37 = torch.nn.functional.batch_norm( + input_36, + l_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_36 = l_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_ + ) = None + input_37 += out_9 + out_10 = input_37 + input_37 = out_9 = None + out_11 = torch.nn.functional.relu(out_10, inplace=True) + out_10 = None + input_38 = torch.conv3d( + out_11, + l_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_, + None, + (1, 2, 2), + (0, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_39 = torch.nn.functional.batch_norm( + input_38, + l_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_38 = l_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_ + ) = None + input_40 = torch.nn.functional.relu(input_39, inplace=True) + input_39 = None + input_41 = torch.conv3d( + input_40, + l_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + input_40 = ( + l_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_ + ) = None + input_42 = torch.nn.functional.batch_norm( + input_41, + l_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_41 = l_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_ + ) = None + input_43 = torch.conv3d( + out_11, + l_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_, + None, + (1, 2, 2), + (0, 0, 0), + (1, 1, 1), + 1, + ) + out_11 = l_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_ = (None) + input_44 = torch.nn.functional.batch_norm( + input_43, + l_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_, + l_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_, + l_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_, + l_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_43 = l_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_ = l_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_ = l_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_ = l_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_ = (None) + input_42 += input_44 + out_12 = input_42 + input_42 = input_44 = None + out_13 = torch.nn.functional.relu(out_12, inplace=True) + out_12 = None + input_45 = torch.conv3d( + out_13, + l_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + l_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_ = ( + None + ) + input_46 = torch.nn.functional.batch_norm( + input_45, + l_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_, + l_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_, + l_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_, + l_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_45 = l_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_ = ( + l_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_ + ) = None + input_47 = torch.nn.functional.relu(input_46, inplace=True) + input_46 = None + input_48 = torch.conv3d( + input_47, + l_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_, + None, + (1, 1, 1), + (0, 1, 1), + (1, 1, 1), + 1, + ) + input_47 = ( + l_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_ + ) = None + input_49 = torch.nn.functional.batch_norm( + input_48, + l_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_, + l_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_, + l_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_, + l_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_, + False, + 0.1, + 1e-05, + ) + input_48 = l_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_ = ( + l_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_ + ) = ( + l_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_ + ) = ( + l_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_ + ) = None + input_49 += out_13 + out_14 = input_49 + input_49 = out_13 = None + out_15 = torch.nn.functional.relu(out_14, inplace=True) + out_14 = None + x = torch.nn.functional.adaptive_avg_pool3d(out_15, (1, 1, 1)) + out_15 = None + x_1 = x.flatten(1) + x = None + x_2 = torch.nn.functional.linear( + x_1, + l_self_modules_fc_parameters_weight_, + l_self_modules_fc_parameters_bias_, + ) + x_1 = ( + l_self_modules_fc_parameters_weight_ + ) = l_self_modules_fc_parameters_bias_ = None + return (x_2,) diff --git a/samples/torchvision/mc3_18/weight_meta.py b/samples/torchvision/mc3_18/weight_meta.py new file mode 100644 index 000000000..4c2c34a44 --- /dev/null +++ b/samples/torchvision/mc3_18/weight_meta.py @@ -0,0 +1,1118 @@ +class Program_weight_tensor_meta_L_self_modules_stem_modules_0_parameters_weight_: + name = "L_self_modules_stem_modules_0_parameters_weight_" + shape = [64, 3, 3, 7, 7] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.000 + std = 0.070 + data = None + + +class Program_weight_tensor_meta_s1: + name = "s1" + shape = [] + dtype = "torch.int64" + device = "cpu" + mean = None + std = None + data = [4] + + +class Program_weight_tensor_meta_s2: + name = "s2" + shape = [] + dtype = "torch.int64" + device = "cpu" + mean = None + std = None + data = [4] + + +class Program_weight_tensor_meta_L_x_: + name = "L_x_" + shape = [1, 3, 16, 224, 224] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.447 + std = 1.305 + data = None + + +class Program_weight_tensor_meta_L_self_modules_stem_modules_1_buffers_running_mean_: + name = "L_self_modules_stem_modules_1_buffers_running_mean_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.011 + std = 0.161 + data = None + + +class Program_weight_tensor_meta_L_self_modules_stem_modules_1_buffers_running_var_: + name = "L_self_modules_stem_modules_1_buffers_running_var_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 9.455 + std = 14.731 + data = None + + +class Program_weight_tensor_meta_L_self_modules_stem_modules_1_parameters_weight_: + name = "L_self_modules_stem_modules_1_parameters_weight_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.221 + std = 0.174 + data = None + + +class Program_weight_tensor_meta_L_self_modules_stem_modules_1_parameters_bias_: + name = "L_self_modules_stem_modules_1_parameters_bias_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.066 + std = 0.285 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer1_modules_0_modules_conv1_modules_0_parameters_weight_" + shape = [64, 64, 3, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.001 + std = 0.032 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.612 + std = 0.753 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer1_modules_0_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.484 + std = 0.289 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_weight_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.339 + std = 0.140 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer1_modules_0_modules_conv1_modules_1_parameters_bias_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.046 + std = 0.193 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer1_modules_0_modules_conv2_modules_0_parameters_weight_" + shape = [64, 64, 3, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.033 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.350 + std = 0.430 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer1_modules_0_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.440 + std = 0.284 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_weight_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.433 + std = 0.227 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer1_modules_0_modules_conv2_modules_1_parameters_bias_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.017 + std = 0.187 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer1_modules_1_modules_conv1_modules_0_parameters_weight_" + shape = [64, 64, 3, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.033 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.632 + std = 0.857 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer1_modules_1_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 1.503 + std = 1.115 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_weight_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.416 + std = 0.105 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer1_modules_1_modules_conv1_modules_1_parameters_bias_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.185 + std = 0.141 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer1_modules_1_modules_conv2_modules_0_parameters_weight_" + shape = [64, 64, 3, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.001 + std = 0.031 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.118 + std = 0.214 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer1_modules_1_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.166 + std = 0.145 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_weight_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.348 + std = 0.205 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer1_modules_1_modules_conv2_modules_1_parameters_bias_" + shape = [64] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.066 + std = 0.352 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer2_modules_0_modules_conv1_modules_0_parameters_weight_" + shape = [128, 64, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.003 + std = 0.045 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.493 + std = 0.526 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer2_modules_0_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.594 + std = 0.226 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_weight_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.369 + std = 0.071 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer2_modules_0_modules_conv1_modules_1_parameters_bias_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.101 + std = 0.120 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer2_modules_0_modules_conv2_modules_0_parameters_weight_" + shape = [128, 128, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.038 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.296 + std = 0.290 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer2_modules_0_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.134 + std = 0.050 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_weight_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.403 + std = 0.106 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer2_modules_0_modules_conv2_modules_1_parameters_bias_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.020 + std = 0.088 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_: + name = "L_self_modules_layer2_modules_0_modules_downsample_modules_0_parameters_weight_" + shape = [128, 64, 1, 1, 1] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.078 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_: + name = "L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_mean_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.033 + std = 0.216 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_: + name = "L_self_modules_layer2_modules_0_modules_downsample_modules_1_buffers_running_var_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.069 + std = 0.052 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_: + name = "L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_weight_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.207 + std = 0.104 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_: + name = ( + "L_self_modules_layer2_modules_0_modules_downsample_modules_1_parameters_bias_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.020 + std = 0.088 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer2_modules_1_modules_conv1_modules_0_parameters_weight_" + shape = [128, 128, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.003 + std = 0.037 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.561 + std = 0.408 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer2_modules_1_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.434 + std = 0.161 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_weight_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.375 + std = 0.049 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer2_modules_1_modules_conv1_modules_1_parameters_bias_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.276 + std = 0.095 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer2_modules_1_modules_conv2_modules_0_parameters_weight_" + shape = [128, 128, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.033 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.103 + std = 0.148 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer2_modules_1_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.053 + std = 0.027 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_weight_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.334 + std = 0.154 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer2_modules_1_modules_conv2_modules_1_parameters_bias_" + shape = [128] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.158 + std = 0.167 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer3_modules_0_modules_conv1_modules_0_parameters_weight_" + shape = [256, 128, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.032 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.618 + std = 0.426 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer3_modules_0_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.428 + std = 0.112 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_weight_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.367 + std = 0.036 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer3_modules_0_modules_conv1_modules_1_parameters_bias_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.199 + std = 0.090 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer3_modules_0_modules_conv2_modules_0_parameters_weight_" + shape = [256, 256, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.028 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.292 + std = 0.179 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer3_modules_0_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.128 + std = 0.029 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_weight_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.425 + std = 0.068 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer3_modules_0_modules_conv2_modules_1_parameters_bias_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.111 + std = 0.079 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_: + name = "L_self_modules_layer3_modules_0_modules_downsample_modules_0_parameters_weight_" + shape = [256, 128, 1, 1, 1] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.003 + std = 0.047 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_: + name = "L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_mean_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.081 + std = 0.143 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_: + name = "L_self_modules_layer3_modules_0_modules_downsample_modules_1_buffers_running_var_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.045 + std = 0.024 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_: + name = "L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_weight_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.166 + std = 0.064 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_: + name = ( + "L_self_modules_layer3_modules_0_modules_downsample_modules_1_parameters_bias_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.111 + std = 0.079 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer3_modules_1_modules_conv1_modules_0_parameters_weight_" + shape = [256, 256, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.003 + std = 0.025 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.594 + std = 0.343 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer3_modules_1_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.274 + std = 0.060 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_weight_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.341 + std = 0.039 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer3_modules_1_modules_conv1_modules_1_parameters_bias_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.319 + std = 0.103 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer3_modules_1_modules_conv2_modules_0_parameters_weight_" + shape = [256, 256, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.003 + std = 0.024 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.161 + std = 0.119 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer3_modules_1_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.033 + std = 0.012 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_weight_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.313 + std = 0.096 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer3_modules_1_modules_conv2_modules_1_parameters_bias_" + shape = [256] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.205 + std = 0.119 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer4_modules_0_modules_conv1_modules_0_parameters_weight_" + shape = [512, 256, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.024 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.481 + std = 0.232 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer4_modules_0_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.289 + std = 0.041 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_weight_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.351 + std = 0.031 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer4_modules_0_modules_conv1_modules_1_parameters_bias_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.316 + std = 0.075 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer4_modules_0_modules_conv2_modules_0_parameters_weight_" + shape = [512, 512, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.002 + std = 0.022 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.214 + std = 0.084 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer4_modules_0_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.052 + std = 0.007 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_weight_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.538 + std = 0.038 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer4_modules_0_modules_conv2_modules_1_parameters_bias_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.258 + std = 0.057 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_: + name = "L_self_modules_layer4_modules_0_modules_downsample_modules_0_parameters_weight_" + shape = [512, 256, 1, 1, 1] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.001 + std = 0.032 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_: + name = "L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_mean_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.047 + std = 0.067 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_: + name = "L_self_modules_layer4_modules_0_modules_downsample_modules_1_buffers_running_var_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.023 + std = 0.007 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_: + name = "L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_weight_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.190 + std = 0.038 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_: + name = ( + "L_self_modules_layer4_modules_0_modules_downsample_modules_1_parameters_bias_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.258 + std = 0.057 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_: + name = "L_self_modules_layer4_modules_1_modules_conv1_modules_0_parameters_weight_" + shape = [512, 512, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.003 + std = 0.021 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_mean_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.736 + std = 0.131 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer4_modules_1_modules_conv1_modules_1_buffers_running_var_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.199 + std = 0.022 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_: + name = "L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_weight_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.362 + std = 0.038 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_: + name = "L_self_modules_layer4_modules_1_modules_conv1_modules_1_parameters_bias_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.379 + std = 0.087 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_: + name = "L_self_modules_layer4_modules_1_modules_conv2_modules_0_parameters_weight_" + shape = [512, 512, 1, 3, 3] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.000 + std = 0.015 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_: + name = ( + "L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_mean_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = -0.006 + std = 0.017 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_: + name = ( + "L_self_modules_layer4_modules_1_modules_conv2_modules_1_buffers_running_var_" + ) + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.017 + std = 0.002 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_: + name = "L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_weight_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 1.787 + std = 0.133 + data = None + + +class Program_weight_tensor_meta_L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_: + name = "L_self_modules_layer4_modules_1_modules_conv2_modules_1_parameters_bias_" + shape = [512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.134 + std = 0.062 + data = None + + +class Program_weight_tensor_meta_L_self_modules_fc_parameters_weight_: + name = "L_self_modules_fc_parameters_weight_" + shape = [400, 512] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.000 + std = 0.110 + data = None + + +class Program_weight_tensor_meta_L_self_modules_fc_parameters_bias_: + name = "L_self_modules_fc_parameters_bias_" + shape = [400] + dtype = "torch.float32" + device = "cuda:0" + mean = 0.000 + std = 0.048 + data = None