From 62936fc7d32e5330bb2bb969099118c2fcfa6d65 Mon Sep 17 00:00:00 2001 From: Avi Lumelsky Date: Tue, 23 May 2023 15:16:38 +0300 Subject: [PATCH] Added CoreML export notebook & method for Apple devices. --- .../training/models/__init__.py | 3 +- .../training/models/conversion.py | 126 +- tests/unit_tests/export_coreml_test.py | 56 + tests/unit_tests/export_onnx_test.py | 2 + tutorials/SG_export_models_to_coreml.ipynb | 3545 +++++++++++++++++ utils_script/create_sub_coco.py | 8 +- 6 files changed, 3727 insertions(+), 13 deletions(-) create mode 100644 tests/unit_tests/export_coreml_test.py create mode 100644 tutorials/SG_export_models_to_coreml.ipynb diff --git a/src/super_gradients/training/models/__init__.py b/src/super_gradients/training/models/__init__.py index cf0f3d9205..3c8ad8e5af 100755 --- a/src/super_gradients/training/models/__init__.py +++ b/src/super_gradients/training/models/__init__.py @@ -123,7 +123,7 @@ import super_gradients.training.models.user_models as user_models from super_gradients.training.models.model_factory import get, get_model_name from super_gradients.training.models.arch_params_factory import get_arch_params -from super_gradients.training.models.conversion import convert_to_onnx, convert_from_config +from super_gradients.training.models.conversion import convert_to_coreml, convert_to_onnx, convert_from_config from super_gradients.common.object_names import Models @@ -280,6 +280,7 @@ "get", "get_model_name", "get_arch_params", + "convert_to_coreml", "convert_to_onnx", "convert_from_config", "ARCHITECTURES", diff --git a/src/super_gradients/training/models/conversion.py b/src/super_gradients/training/models/conversion.py index 8e5e514009..30d249dc79 100644 --- a/src/super_gradients/training/models/conversion.py +++ b/src/super_gradients/training/models/conversion.py @@ -1,26 +1,34 @@ +import os +import pathlib from pathlib import Path import hydra +import numpy as np +import onnx import torch from omegaconf import DictConfig -import numpy as np +from onnxsim import simplify from torch.nn import Identity from super_gradients.common.abstractions.abstract_logger import get_logger from super_gradients.common.decorators.factory_decorator import resolve_param +from super_gradients.common.environment.cfg_utils import load_experiment_cfg +from super_gradients.common.environment.checkpoints_dir_utils import get_checkpoints_dir_path from super_gradients.common.factories.transforms_factory import TransformsFactory from super_gradients.training import models -from super_gradients.common.environment.checkpoints_dir_utils import get_checkpoints_dir_path -from super_gradients.common.environment.cfg_utils import load_experiment_cfg from super_gradients.training.utils.sg_trainer_utils import parse_args -import os -import pathlib - -from onnxsim import simplify -import onnx logger = get_logger(__name__) +ct = None + +try: + import coremltools as coreml_tools + + ct = coreml_tools +except (ImportError, ModuleNotFoundError): + pass + class ConvertableCompletePipelineModel(torch.nn.Module): """ @@ -48,6 +56,108 @@ def forward(self, x): return self.post_process(self.model(self.pre_process(x))) +@resolve_param("pre_process", TransformsFactory()) +@resolve_param("post_process", TransformsFactory()) +def convert_to_coreml( + model: torch.nn.Module, + out_path: str, + input_size: tuple = None, + pre_process: torch.nn.Module = None, + post_process: torch.nn.Module = None, + prep_model_for_conversion_kwargs=None, + export_as_ml_program=False, + torch_trace_kwargs=None, +): + """ + Exports a given SG model to CoreML mlprogram or package. + + :param model: torch.nn.Module, model to export to ONNX. + :param out_path: str, destination path for the .onnx file. + :param input_size: Input shape without batch dimensions ([C,H,W]). Batch size assumed to be 1. + :param pre_process: torch.nn.Module, preprocessing pipeline, will be resolved by TransformsFactory() + :param post_process: torch.nn.Module, postprocessing pipeline, will be resolved by TransformsFactory() + :param prep_model_for_conversion_kwargs: dict, for SgModules- args to be passed to model.prep_model_for_conversion + prior to torch.onnx.export call. Supported keys are: + - input_size - Shape of inputs with batch dimension, [C,H,W] for image inputs. + When true, the simplified model will be saved in out_path (default=True). + :param export_as_ml_program: Whether to convert to the new program format (better) or legacy coreml proto file + (Supports more iOS versions and devices, but this format will be deprecated at some point). + :param torch_trace_kwargs: kwargs for torch.jit.trace + :return: Path + """ + if ct is None: + raise ImportError( + '"coremltools" is required for CoreML export, but is not installed. Please install CoreML Tools using:\n' + ' "python3 -m pip install coremltools" and try again (Tested with version 6.3.0);' + ) + + logger.debug("Building model...") + logger.debug(model) + logger.debug("Model child nodes:") + logger.debug(next(model.named_children())) + + if not os.path.isdir(pathlib.Path(out_path).parent.resolve()): + raise FileNotFoundError(f"Could not find destination directory {out_path} for the ONNX file.") + torch_trace_kwargs = torch_trace_kwargs or dict() + prep_model_for_conversion_kwargs = prep_model_for_conversion_kwargs or dict() + + if input_size is not None: + input_size = (1, *input_size) + logger.warning( + f"input_shape is deprecated and will be removed in the next major release." + f"Use the convert_to_coreml(..., prep_model_for_conversion_kwargs(input_size={input_size})) instead" + ) + prep_model_for_conversion_kwargs["input_size"] = input_size + + if "input_size" not in prep_model_for_conversion_kwargs: + raise KeyError("input_size must be provided in prep_model_for_conversion_kwargs") + + input_size = prep_model_for_conversion_kwargs["input_size"] + + # TODO: support more than 1 input when prep_for_conversoin will support it. + example_inputs = [torch.Tensor(np.zeros(input_size))] + + if not out_path.endswith(".mlpackage") and not out_path.endswith(".mlmodel"): + out_path += ".mlpackage" if export_as_ml_program else ".mlmodel" + + complete_model = ConvertableCompletePipelineModel(model, pre_process, post_process, **prep_model_for_conversion_kwargs) + + # Set the model in evaluation mode. + complete_model.eval() + + logger.info("Creating torch jit trace...") + traced_model = torch.jit.trace(complete_model, example_inputs, **torch_trace_kwargs) + logger.info("Tracing the model with the provided inputs...") + out = traced_model(*example_inputs) # using * because example_inputs is a list + logger.info(f"Inferred output shapes: {[o.shape for o in out]}") + if export_as_ml_program: + coreml_model = ct.convert( + traced_model, convert_to="mlprogram", inputs=[ct.ImageType(name=f"x_{i + 1}", shape=_.shape) for i, _ in enumerate(example_inputs)] + ) + else: + coreml_model = ct.convert(traced_model, inputs=[ct.ImageType(name=f"x_{i + 1}", shape=_.shape) for i, _ in enumerate(example_inputs)]) + + spec = coreml_model.get_spec() + logger.debug(spec.description) + + # Changing the input names: + # In CoreML, the input name is compiled into classes (named keyword argument in predict). + # We want to re-use the same names among different models to make research easier. + # We normalize the inputs names to be x_1, x_2, etc. + for i, _input in enumerate(spec.description.input): + new_input_name = "x_" + str(i + 1) + logger.info(f"Renaming input {_input.name} to {new_input_name}") + ct.utils.rename_feature(spec, _input.name, new_input_name) + + # Re-Initializing the model with the new spec + coreml_model = ct.models.MLModel(spec, weights_dir=coreml_model.weights_dir) + + # Saving the model + coreml_model.save(out_path) + logger.info(f"CoreML model successfully save to {os.path.abspath(out_path)}") + return out_path + + @resolve_param("pre_process", TransformsFactory()) @resolve_param("post_process", TransformsFactory()) def convert_to_onnx( diff --git a/tests/unit_tests/export_coreml_test.py b/tests/unit_tests/export_coreml_test.py new file mode 100644 index 0000000000..7b39b37ab0 --- /dev/null +++ b/tests/unit_tests/export_coreml_test.py @@ -0,0 +1,56 @@ +import os +import tempfile +import unittest + +from torchvision.transforms import Compose, Normalize, Resize + +from super_gradients.common.object_names import Models +from super_gradients.training import models +from super_gradients.training.transforms import Standardize + + +class TestModelsCoreMLExport(unittest.TestCase): + def test_models_onnx_export_with_explicit_input_size(self): + pretrained_model = models.get(Models.RESNET18, num_classes=1000, pretrained_weights="imagenet") + preprocess = Compose([Resize(224), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) + with tempfile.TemporaryDirectory() as tmpdirname: + out_path = os.path.join(tmpdirname, "resnet18.mlmodel") + models.convert_to_coreml(model=pretrained_model, out_path=out_path, input_size=(3, 256, 256), pre_process=preprocess) + self.assertTrue(os.path.isfile(out_path)) + + def test_models_onnx_export_without_explicit_input_size_raises_error(self): + pretrained_model = models.get(Models.RESNET18, num_classes=1000, pretrained_weights="imagenet") + preprocess = Compose([Resize(224), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) + with self.assertRaises(KeyError): + models.convert_to_coreml(model=pretrained_model, out_path="some-output-path.coreml", pre_process=preprocess) + + def test_models_coreml_export(self, **export_kwargs): + pretrained_model = models.get(Models.YOLO_NAS_S, num_classes=1000, pretrained_weights="coco") + + # Just for the sake of testing, not really COCO preprocessing + preprocess = Compose([Resize(224), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) + with tempfile.TemporaryDirectory() as tmpdirname: + out_path = os.path.join(tmpdirname, "yolo_nas_s") + model_path = models.convert_to_coreml( + model=pretrained_model, + out_path=out_path, + pre_process=preprocess, + prep_model_for_conversion_kwargs=dict(input_size=(1, 3, 640, 640)), + **export_kwargs, + ) + + if export_kwargs.get("export_as_ml_program"): + # Expecting a directory + self.assertTrue(os.path.isdir(model_path)) + self.assertTrue(model_path.endswith(".mlpackage")) + else: + # Expecting a single file + self.assertTrue(os.path.isfile(model_path)) + self.assertTrue(model_path.endswith(".mlmodel")) + + def test_models_coreml_export_as_mlprogram(self): + self.test_models_coreml_export(export_as_ml_program=True) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/unit_tests/export_onnx_test.py b/tests/unit_tests/export_onnx_test.py index 7608aeab20..0e4d256842 100644 --- a/tests/unit_tests/export_onnx_test.py +++ b/tests/unit_tests/export_onnx_test.py @@ -15,6 +15,7 @@ def test_models_onnx_export_with_deprecated_input_shape(self): with tempfile.TemporaryDirectory() as tmpdirname: out_path = os.path.join(tmpdirname, "resnet18.onnx") models.convert_to_onnx(model=pretrained_model, out_path=out_path, input_shape=(3, 256, 256), pre_process=preprocess) + self.assertTrue(os.path.exists(out_path)) def test_models_onnx_export(self): pretrained_model = models.get(Models.RESNET18, num_classes=1000, pretrained_weights="imagenet") @@ -24,6 +25,7 @@ def test_models_onnx_export(self): models.convert_to_onnx( model=pretrained_model, out_path=out_path, pre_process=preprocess, prep_model_for_conversion_kwargs=dict(input_size=(1, 3, 640, 640)) ) + self.assertTrue(os.path.exists(out_path)) if __name__ == "__main__": diff --git a/tutorials/SG_export_models_to_coreml.ipynb b/tutorials/SG_export_models_to_coreml.ipynb new file mode 100644 index 0000000000..59c49bc49d --- /dev/null +++ b/tutorials/SG_export_models_to_coreml.ipynb @@ -0,0 +1,3545 @@ +{ + "cells": [ + { + "cell_type": "markdown", + "metadata": { + "id": "Gd1erdPjNwTo" + }, + "source": [ + "# Convert SG Models to CoreML" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "KrBvHOjq669c" + }, + "source": [ + "Let's install the minimal dependencies for getting started.
\n", + "Those will be 'super-gradients' and 'coremltools'.
\n", + "If you are using Colab, please restart the kernel after they are installed and only then continue." + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "yMeIBtysNwTr", + "outputId": "65e73080-917f-44b6-9006-e7933b26cca0" + }, + "outputs": [], + "source": [ + "!python3 -m pip install super-gradients coremltools" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "En9ynj-NQBHz" + }, + "source": [ + "#### In Google Colab, Restart the notebook, and then continue;" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "tIryGpopNwTt" + }, + "source": [ + "## Load any SG model" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "mC6z8r4J4Ruu" + }, + "source": [ + "For this tutorial we will use the YOLO-NAS-S model.
\n", + "We will load the pretrained version of Deci's YOLO NAS S model and convert the graph representation from PyTorch to CoreML.
\n", + "We also must define an example input - It is required when exporting torch models.
" + ] + }, + { + "cell_type": "code", + "execution_count": 1, + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "O9iQMb5aNwTu", + "is_executing": true, + "outputId": "dafef7a9-9005-475a-f616-afa4e1749835" + }, + "outputs": [ + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:23:23] INFO - crash_tips_setup.py - Crash tips is enabled. You can set your environment variable to CRASH_HANDLER=FALSE to disable it\n" + ] + }, + { + "name": "stdout", + "output_type": "stream", + "text": [ + "The console stream is logged into /home/oem/sg_logs/console.log\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:23:25] WARNING - __init__.py - Torch version 2.0.1+cu117 has not been tested with coremltools. You may run into unexpected errors. Torch 2.0.0 is the most recent version that has been tested.\n", + "[2023-05-18 14:23:25] WARNING - __init__.py - Failed to import pytorch_quantization\n", + "[2023-05-18 14:23:25] WARNING - calibrator.py - Failed to import pytorch_quantization\n", + "[2023-05-18 14:23:25] WARNING - export.py - Failed to import pytorch_quantization\n", + "[2023-05-18 14:23:25] WARNING - selective_quantization_utils.py - Failed to import pytorch_quantization\n", + "[2023-05-18 14:23:25] INFO - checkpoint_utils.py - License Notification: YOLO-NAS pre-trained weights are subjected to the specific license terms and conditions detailed in \n", + "https://github.com/Deci-AI/super-gradients/blob/master/LICENSE.YOLONAS.md\n", + "By downloading the pre-trained weight files you agree to comply with these terms.\n" + ] + }, + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Loaded successfully\n" + ] + } + ], + "source": [ + "from super_gradients.common.object_names import Models\n", + "from super_gradients.training import models\n", + "\n", + "\n", + "sg_torch_model = models.get(Models.YOLO_NAS_S, pretrained_weights=\"coco\")\n", + "print('Loaded successfully')" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "UW_ykyb0bg-g" + }, + "source": [ + "## Convert to CoreML" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "IvkD2MWF5XNr" + }, + "source": [ + "After the model loaded successfully, we will define a target path to store the compiled model into.
\n", + "In this case, our file is named \"deci-yolo-nas-s.mlmodel\" in the current working directory ('/content' in google colab)." + ] + }, + { + "cell_type": "code", + "execution_count": 6, + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "e5ArIe1cNwTx", + "is_executing": true, + "outputId": "f4689314-1ba1-4d0c-c973-356d97e0bca1" + }, + "outputs": [ + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:19] INFO - conversion.py - Creating torch jit trace...\n", + "[2023-05-18 14:24:21] INFO - conversion.py - Tracing the model with the provided inputs...\n", + "[2023-05-18 14:24:22] INFO - conversion.py - Inferred output shapes: [torch.Size([1, 8400, 4]), torch.Size([1, 8400, 80])]\n", + "[2023-05-18 14:24:23] WARNING - torchir_passes.py - Tuple detected at graph output. This will be flattened in the converted model.\n", + "[2023-05-18 14:24:23] INFO - converter.py - Converting graph.\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stem.conv.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stem.conv.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.downsample.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.downsample.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.0.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.0.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.0.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.0.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.1.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.1.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.1.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.bottlenecks.1.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage1.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.downsample.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.downsample.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.0.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.0.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.0.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.0.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.1.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.1.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.1.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.1.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.2.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.2.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.2.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.2.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.bottlenecks.2.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage2.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.downsample.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.downsample.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv1.conv.weight' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.0.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.0.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.0.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.0.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.1.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.1.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.1.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.1.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.2.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.2.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.2.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.2.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.2.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.3.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.3.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.3.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.3.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.3.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.4.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.4.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.4.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.4.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.bottlenecks.4.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage3.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.downsample.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.downsample.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.0.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.0.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.0.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.0.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.1.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.1.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.1.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.bottlenecks.1.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.stage4.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv1.conv.weight' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.backbone.context_module.cv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_skip2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.downsample.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.downsample.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.downsample.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.downsample.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.downsample.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.conv.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.conv.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.conv.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.conv.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.conv.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.upsample.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.upsample.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_after_concat.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_after_concat.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_after_concat.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_after_concat.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.reduce_after_concat.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.0.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.0.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.0.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.0.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.1.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.1.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.1.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.bottlenecks.1.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck1.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_skip2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.downsample.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.downsample.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.downsample.bn.running_mean' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.downsample.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.downsample.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.conv.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.conv.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.conv.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.conv.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.conv.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.upsample.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.upsample.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_after_concat.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_after_concat.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_after_concat.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_after_concat.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.reduce_after_concat.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.0.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.0.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.0.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.0.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.1.cv1.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.1.cv1.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.1.cv2.rbr_reparam.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.bottlenecks.1.cv2.rbr_reparam.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck2.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.conv.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.conv.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.conv.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.conv.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.conv.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.0.cv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.bottlenecks.1.cv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv2.conv.weight' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck3.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.conv.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.conv.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.conv.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.conv.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.conv.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.0.cv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.alpha' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv1.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv1.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv1.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv1.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv1.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.bottlenecks.1.cv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv2.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv2.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv2.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv2.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv2.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv3.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv3.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv3.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv3.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.neck.neck4.blocks.conv3.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.proj_conv' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.stem.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.stem.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.stem.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.stem.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.stem.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_convs.0.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_convs.0.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_convs.0.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_convs.0.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_convs.0.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_pred.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.cls_pred.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_convs.0.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_convs.0.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_convs.0.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_convs.0.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_convs.0.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_pred.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head1.reg_pred.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.stem.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.stem.seq.bn.running_var' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.stem.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.stem.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.stem.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_convs.0.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_convs.0.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_convs.0.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_convs.0.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_convs.0.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_pred.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.cls_pred.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_convs.0.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_convs.0.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_convs.0.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_convs.0.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_convs.0.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_pred.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head2.reg_pred.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.stem.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.stem.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.stem.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.stem.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.stem.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_convs.0.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_convs.0.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_convs.0.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_convs.0.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_convs.0.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_pred.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.cls_pred.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_convs.0.seq.conv.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_convs.0.seq.bn.running_var' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_convs.0.seq.bn.running_mean' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_convs.0.seq.bn.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_convs.0.seq.bn.weight' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_pred.bias' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'model.heads.head3.reg_pred.weight' of type const\n", + "Converting PyTorch Frontend ==> MIL Ops: 0%| | 0/919 [00:00 MIL Ops: 12%|█████████████████▌ | 109/919 [00:00<00:00, 1089.06 ops/s][2023-05-18 14:24:23] INFO - ops.py - Converting op input.49 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.49' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.49_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.49_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 204 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '204' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.9 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.9' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 209 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '209' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 213 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '213' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 214 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '214' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 215 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '215' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 216 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '216' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.51 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.51' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.51_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.51_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.53 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.53' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 222 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '222' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 223 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '223' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 224 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '224' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 225 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '225' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.55 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.55' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.55_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.55_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 227 : relu_\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '227' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.11 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.11' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 232 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '232' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 236 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '236' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 237 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '237' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 238 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '238' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 239 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '239' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.57 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.57' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.57_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.57_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.59 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.59' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 245 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '245' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 246 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '246' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 247 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '247' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 248 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '248' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.61 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.61' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.61_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.61_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 250 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '250' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 251 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '251' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 255 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '255' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 256 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '256' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 257 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '257' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 258 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '258' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.63 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.63' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.63_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.63_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.65 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.65' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x_2.3 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2.3' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 266 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.67 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.67' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.67_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 271 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '271' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 272 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '272' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 273 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '273' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 274 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '274' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.69 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.69' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.69_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.69_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.71 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.71' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.73 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.73' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 287 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '287' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 288 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '288' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 289 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '289' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 290 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '290' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.75 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.75' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.75_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.75_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.77 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.77' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 300 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '300' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 301 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '301' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 302 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '302' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 303 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '303' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.79 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.79' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.79_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.79_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.81 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.81' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.13 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.13' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 319 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '319' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 323 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '323' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 324 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '324' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 325 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '325' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 326 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '326' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.83 : _convolution\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.83' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.83_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.83_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.85 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.85' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 332 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '332' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 333 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '333' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 334 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '334' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 335 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '335' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.87 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.87' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.87_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.87_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 337 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '337' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.15 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.15' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 342 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '342' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 346 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '346' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 347 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '347' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 348 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '348' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 349 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '349' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.89 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.89' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.89_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.89_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.91 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.91' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 355 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '355' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 356 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '356' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 357 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '357' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 358 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '358' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.93 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.93' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.93_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.93_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 360 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '360' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.17 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.17' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 365 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '365' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 369 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '369' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 370 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '370' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 371 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '371' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 372 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '372' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.95 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.95' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.95_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.95_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.97 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.97' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 378 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '378' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 379 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '379' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 380 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '380' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 381 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '381' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.99 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.99' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.99_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.99_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 383 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '383' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.19 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.19' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 388 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '388' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 392 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '392' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 393 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '393' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 394 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '394' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 395 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '395' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.101 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.101' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.101_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.101_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.103 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.103' of type relu\n", + "Converting PyTorch Frontend ==> MIL Ops: 24%|███████████████████████████████████ | 218/919 [00:00<00:00, 1052.93 ops/s][2023-05-18 14:24:23] INFO - ops.py - Converting op 401 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '401' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 402 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '402' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 403 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '403' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 404 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '404' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.105 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.105' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.105_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.105_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 406 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '406' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.21 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.21' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 411 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '411' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 415 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '415' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 416 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '416' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 417 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '417' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 418 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '418' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.107 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.107' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.107_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.107_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.109 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.109' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 424 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '424' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 425 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '425' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 426 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '426' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 427 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '427' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.111 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.111' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.111_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.111_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 429 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '429' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 430 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '430' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 434 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '434' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 435 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '435' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 436 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '436' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 437 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '437' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.113 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.113' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.113_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.113_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.115 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.115' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x_2.5 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2.5' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 445 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.117 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.117' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.117_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 450 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '450' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 451 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '451' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 452 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '452' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 453 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '453' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.119 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.119' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.119_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.119_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.121 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.121' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.123 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.123' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 466 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '466' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 467 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '467' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 468 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '468' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 469 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '469' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.125 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.125' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.125_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.125_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.127 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.127' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 479 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '479' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 480 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '480' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 481 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '481' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 482 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '482' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.129 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.129' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.129_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.129_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.131 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.131' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.23 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.23' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 495 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '495' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 499 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '499' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 500 : listconstruct\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '500' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 501 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '501' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 502 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '502' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.133 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.133' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.133_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.133_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.135 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.135' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 508 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '508' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 509 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '509' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 510 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '510' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 511 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '511' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.137 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.137' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.137_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.137_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 513 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '513' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.25 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.25' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 518 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '518' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 522 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '522' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 523 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '523' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 524 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '524' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 525 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '525' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.139 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.139' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.139_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.139_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.141 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.141' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 531 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '531' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 532 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '532' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 533 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '533' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 534 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '534' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.143 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.143' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.143_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.143_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 536 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '536' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 537 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '537' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 541 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '541' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 542 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '542' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 543 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '543' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 544 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '544' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.145 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.145' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.145_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.145_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.147 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.147' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x_2.7 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2.7' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 552 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.149 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.149' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.149_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 557 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '557' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 558 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '558' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 559 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '559' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 560 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '560' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.151 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.151' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.151_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.151_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.153 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.153' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.155 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.155' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 573 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '573' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 574 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '574' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 575 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '575' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 576 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '576' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.157 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.157' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.157_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.157_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.159 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.159' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.161 : relu_\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.161' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 584 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '584' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 585 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '585' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 586 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '586' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 587 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '587' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 588 : max_pool2d\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '588' of type max_pool\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '588_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '588_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '588_ceil_mode_0' of type const\n", + "Converting PyTorch Frontend ==> MIL Ops: 35%|████████████████████████████████████████████████████▏ | 324/919 [00:00<00:00, 1029.36 ops/s][2023-05-18 14:24:23] INFO - ops.py - Converting op 589 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '589' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 590 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '590' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 591 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '591' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 592 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '592' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 593 : max_pool2d\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '593' of type max_pool\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '593_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '593_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '593_ceil_mode_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 594 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '594' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 595 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '595' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 596 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '596' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 597 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '597' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 598 : max_pool2d\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '598' of type max_pool\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '598_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '598_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '598_ceil_mode_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 599 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.163 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.163' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.163_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 604 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '604' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 605 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '605' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 606 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '606' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 607 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '607' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.165 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.165' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.165_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.165_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.167 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.167' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.183 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.183' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 615 : tupleconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 616 : tupleunpack\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 635 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '635' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 636 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '636' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 637 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '637' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 638 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '638' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.169 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.169' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.169_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.169_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.171 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.171' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op skip_x1.1 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'skip_x1.1' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 649 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '649' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 650 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '650' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 651 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '651' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 652 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '652' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.173 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.173' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.173_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.173_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.175 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.175' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.177 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.177' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 663 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '663' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 664 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '664' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 665 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '665' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 666 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '666' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.179 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.179' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.179_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.179_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.181 : batch_norm\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.181' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 673 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '673' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 677 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '677' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 678 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '678' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 679 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '679' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 680 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '680' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.185 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.185' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.185_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.185_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.187 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.187' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.189 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.189' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 690 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '690' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 691 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '691' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 692 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '692' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 693 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '693' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.27 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.27' of type conv_transpose\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.27_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.27_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 695 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.191 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.191' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.191_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 700 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '700' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 701 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '701' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 702 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '702' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 703 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '703' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.193 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.193' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.193_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.193_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.195 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.195' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.197 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.197' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 718 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '718' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 719 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '719' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 720 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '720' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 721 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '721' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.199 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.199' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.199_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.199_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.201 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.201' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.29 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.29' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 734 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '734' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 738 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '738' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 739 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '739' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 740 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '740' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 741 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '741' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.203 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.203' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.203_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.203_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.205 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.205' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 747 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '747' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 748 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '748' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 749 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '749' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 750 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '750' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.207 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.207' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.207_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.207_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 752 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '752' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.31 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.31' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 757 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '757' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 761 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '761' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 762 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '762' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 763 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '763' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 764 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '764' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.209 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.209' of type conv\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.209_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.209_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.211 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.211' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 770 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '770' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 771 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '771' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 772 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '772' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 773 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '773' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.213 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.213' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.213_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.213_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 775 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '775' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 776 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '776' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 780 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '780' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 781 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '781' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 782 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '782' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 783 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '783' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.215 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.215' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.215_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.215_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.217 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.217' of type batch_norm\n", + "Converting PyTorch Frontend ==> MIL Ops: 47%|████████████████████████████████████████████████████████████████████▉ | 428/919 [00:00<00:00, 1024.11 ops/s][2023-05-18 14:24:23] INFO - ops.py - Converting op x_2.9 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2.9' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 791 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.219 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.219' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.219_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 796 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '796' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 797 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '797' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 798 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '798' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 799 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '799' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.221 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.221' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.221_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.221_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.223 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.223' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.239 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.239' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 807 : tupleconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 808 : tupleunpack\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 820 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '820' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 821 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '821' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 822 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '822' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 823 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '823' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.225 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.225' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.225_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.225_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.227 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.227' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op skip_x1 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'skip_x1' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 834 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '834' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 835 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '835' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 836 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '836' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 837 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '837' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.229 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.229' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.229_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.229_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.231 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.231' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.233 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.233' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 848 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '848' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 849 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '849' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 850 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '850' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 851 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '851' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.235 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.235' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.235_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.235_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.237 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.237' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 858 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '858' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 862 : listconstruct\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '862' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 863 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '863' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 864 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '864' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 865 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '865' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.241 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.241' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.241_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.241_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.243 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.243' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.245 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.245' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 875 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '875' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 876 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '876' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 877 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '877' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 878 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '878' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.33 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.33' of type conv_transpose\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.33_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.33_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 880 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.247 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.247' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.247_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 885 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '885' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 886 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '886' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 887 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '887' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 888 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '888' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.249 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.249' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.249_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.249_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.251 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.251' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.253 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.253' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 903 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '903' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 904 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '904' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 905 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '905' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 906 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '906' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.255 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.255' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.255_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.255_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.257 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.257' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.35 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.35' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 919 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '919' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 923 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '923' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 924 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '924' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 925 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '925' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 926 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '926' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.259 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.259' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.259_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.259_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.261 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.261' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 932 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '932' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 933 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '933' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 934 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '934' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 935 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '935' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.263 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.263' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.263_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.263_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 937 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '937' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.37 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.37' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 942 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '942' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 946 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '946' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 947 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '947' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 948 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '948' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 949 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '949' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.265 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.265' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.265_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.265_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.267 : relu_\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.267' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 955 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '955' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 956 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '956' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 957 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '957' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 958 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '958' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.269 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.269' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.269_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.269_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 960 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '960' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 961 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '961' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 965 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '965' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 966 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '966' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 967 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '967' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 968 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '968' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.271 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.271' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.271_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.271_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.273 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.273' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x_2.11 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2.11' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 976 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.275 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.275' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.275_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 981 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '981' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 982 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '982' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 983 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '983' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 984 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '984' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.277 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.277' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.277_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.277_pad_0' of type const\n", + "Converting PyTorch Frontend ==> MIL Ops: 58%|██████████████████████████████████████████████████████████████████████████████████████ | 531/919 [00:00<00:00, 972.29 ops/s][2023-05-18 14:24:23] INFO - ops.py - Converting op input.279 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.279' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.281 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.281' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 992 : tupleconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 993 : tupleunpack\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1000 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1000' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1001 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1001' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1002 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1002' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1003 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1003' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.283 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.283' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.283_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.283_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.285 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.285' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.39 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.39' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1011 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.287 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.287' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.287_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1020 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1020' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1021 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1021' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1022 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1022' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1023 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1023' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.289 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.289' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.289_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.289_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.291 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.291' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.41 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.41' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1036 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1036' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1040 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1040' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1041 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1041' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1042 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1042' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1043 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1043' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.293 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.293' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.293_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.293_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.295 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.295' of type batch_norm\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.297 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.297' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1054 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1054' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1055 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1055' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1056 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1056' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1057 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1057' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.299 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.299' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.299_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.299_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.301 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.301' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1064 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1064' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.43 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.43' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1069 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1069' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1073 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1073' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1074 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1074' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1075 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1075' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1076 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1076' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.303 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.303' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.303_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.303_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.305 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.305' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.307 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.307' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1087 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1087' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1088 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1088' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1089 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1089' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1090 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1090' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.309 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.309' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.309_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.309_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.311 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.311' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1097 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1097' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1098 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1098' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1102 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1102' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1103 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1103' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1104 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1104' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1105 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1105' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.313 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.313' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.313_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.313_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.315 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.315' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x_2.13 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2.13' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1113 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.317 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.317' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.317_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1118 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1118' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1119 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1119' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1120 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1120' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1121 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1121' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.319 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.319' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.319_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.319_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.321 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.321' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.323 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.323' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1134 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1134' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1135 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1135' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1136 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1136' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1137 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1137' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.325 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.325' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.325_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.325_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.327 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.327' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.45 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.45' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1145 : listconstruct\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.329 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.329' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.329_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1154 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1154' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1155 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1155' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1156 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1156' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1157 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1157' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.331 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.331' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.331_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.331_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.333 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.333' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x.47 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x.47' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1170 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1170' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1174 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1174' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1175 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1175' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1176 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1176' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1177 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1177' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.335 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.335' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.335_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.335_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.337 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.337' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.339 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.339' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1188 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1188' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1189 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1189' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1190 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1190' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1191 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1191' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.341 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.341' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.341_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.341_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.343 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.343' of type batch_norm\n", + "Converting PyTorch Frontend ==> MIL Ops: 68%|█████████████████████████████████████████████████████████████████████████████████████████████████████▉ | 629/919 [00:00<00:00, 914.80 ops/s][2023-05-18 14:24:23] INFO - ops.py - Converting op 1198 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1198' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1203 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1203' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1207 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1207' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1208 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1208' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1209 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1209' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1210 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1210' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.345 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.345' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.345_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.345_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.347 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.347' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.349 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.349' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1221 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1221' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1222 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1222' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1223 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1223' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1224 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1224' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.351 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.351' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.351_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.351_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.353 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.353' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1231 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1231' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1232 : add\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1232' of type add\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1236 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1236' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1237 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1237' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1238 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1238' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1239 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1239' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.355 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.355' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.355_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.355_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.357 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.357' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op x_2 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'x_2' of type relu\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1247 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.359 : cat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.359' of type concat\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.359_interleave_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1252 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1252' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1253 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1253' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1254 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1254' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1255 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1255' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.361 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.361' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.361_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.361_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.363 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.363' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op feat : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'feat' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1263 : tupleconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1264 : tupleunpack\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1287 : size\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1287_shape' of type shape\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'const_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1288 : size\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1288_shape' of type shape\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'const_1' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op h.1 : numtotensor\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'h.1' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1290 : size\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1290_shape' of type shape\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'const_2' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op w.1 : numtotensor\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'w.1' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op height_mul_width.1 : mul\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'height_mul_width.1' of type mul\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1293 : int\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1293' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1294 : int\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1294' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1304 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1304' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1305 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1305' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1306 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1306' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1307 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1307' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.365 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.365' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.365_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.365_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.367 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.367' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.369 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.369' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1320 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1320' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1321 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1321' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1322 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1322' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1323 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1323' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.371 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.371' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.371_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.371_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.373 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.373' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.375 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.375' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1333 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1333' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1334 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1334' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1335 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1335' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1336 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1336' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op cls_logit.1 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cls_logit.1' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cls_logit.1_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cls_logit.1_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1343 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1343' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1344 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1344' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1345 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1345' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1346 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1346' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.377 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.377' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.377_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.377_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.379 : batch_norm\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.379' of type batch_norm\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.381 : relu_\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.381' of type relu\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1356 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1356' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1357 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1357' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1358 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1358' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1359 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1359' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op reg_distri.1 : _convolution\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'reg_distri.1' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'reg_distri.1_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'reg_distri.1_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1361 : tupleconstruct\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1362 : tupleunpack\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1364 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1364' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1365 : reshape\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cast_0' of type cast\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cast_0_dtype_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1365' of type reshape\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1366 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1366' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op input.383 : permute\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'input.383' of type transpose\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1368 : softmax\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1368' of type softmax\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1369 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1369' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1370 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1370' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1371 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1371' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1372 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1372' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1373 : _convolution\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1373' of type conv\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1373_pad_type_0' of type const\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1373_pad_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1374 : squeeze\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1374' of type squeeze\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1374_axes_0' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1375 : listconstruct\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op '1375' of type const\n", + "[2023-05-18 14:24:23] INFO - ops.py - Converting op 1376 : reshape\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cast_1' of type cast\n", + "[2023-05-18 14:24:23] INFO - builder.py - Adding op 'cast_1_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1376' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1377 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1377_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_3' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1378 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1378_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_4' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op h.3 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'h.3' of type const\n", + "Converting PyTorch Frontend ==> MIL Ops: 79%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████ | 722/919 [00:00<00:00, 878.71 ops/s][2023-05-18 14:24:24] INFO - ops.py - Converting op 1380 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1380_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_5' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op w.3 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'w.3' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op height_mul_width.3 : mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'height_mul_width.3' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1383 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1383' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1384 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1384' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1394 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1394' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1395 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1395' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1396 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1396' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1397 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1397' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.385 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.385' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.385_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.385_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.387 : batch_norm\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.387' of type batch_norm\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.389 : relu_\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.389' of type relu\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1410 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1410' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1411 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1411' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1412 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1412' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1413 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1413' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.391 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.391' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.391_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.391_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.393 : batch_norm\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.393' of type batch_norm\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.395 : relu_\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.395' of type relu\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1423 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1423' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1424 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1424' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1425 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1425' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1426 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1426' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op cls_logit.3 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_logit.3' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_logit.3_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_logit.3_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1433 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1433' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1434 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1434' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1435 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1435' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1436 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1436' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.397 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.397' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.397_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.397_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.399 : batch_norm\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.399' of type batch_norm\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.401 : relu_\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.401' of type relu\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1446 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1446' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1447 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1447' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1448 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1448' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1449 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1449' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op reg_distri.3 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_distri.3' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_distri.3_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_distri.3_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1451 : tupleconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1452 : tupleunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1454 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1454' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1455 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_2' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_2_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1455' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1456 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1456' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.403 : permute\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.403' of type transpose\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1458 : softmax\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1458' of type softmax\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1459 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1459' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1460 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1460' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1461 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1461' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1462 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1462' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1463 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1463' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1463_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1463_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1464 : squeeze\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1464' of type squeeze\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1464_axes_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1465 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1465' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1466 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_3' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_3_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1466' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1467 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1467_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_6' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1468 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1468_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_7' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op h.5 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'h.5' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1470 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1470_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_8' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op w.5 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'w.5' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op height_mul_width : mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'height_mul_width' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1473 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1473' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1474 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1474' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1484 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1484' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1485 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1485' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1486 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1486' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1487 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1487' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.405 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.405' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.405_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.405_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.407 : batch_norm\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.407' of type batch_norm\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.409 : relu_\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.409' of type relu\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1500 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1500' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1501 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1501' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1502 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1502' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1503 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1503' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.411 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.411' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.411_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.411_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.413 : batch_norm\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.413' of type batch_norm\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.415 : relu_\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.415' of type relu\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1513 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1513' of type const\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1514 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1514' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1515 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1515' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1516 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1516' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op cls_logit : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_logit' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_logit_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_logit_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1523 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1523' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1524 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1524' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1525 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1525' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1526 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1526' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.417 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.417' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.417_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.417_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.419 : batch_norm\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.419' of type batch_norm\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input.421 : relu_\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input.421' of type relu\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1536 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1536' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1537 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1537' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1538 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1538' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1539 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1539' of type const\n", + "Converting PyTorch Frontend ==> MIL Ops: 88%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▍ | 811/919 [00:00<00:00, 874.78 ops/s][2023-05-18 14:24:24] INFO - ops.py - Converting op reg_distri : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_distri' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_distri_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_distri_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1541 : tupleconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1542 : tupleunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1544 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1544' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1545 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_4' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_4_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1545' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1546 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1546' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op input : permute\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'input' of type transpose\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1548 : softmax\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1548' of type softmax\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1549 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1549' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1550 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1550' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1551 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1551' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1552 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1552' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1553 : _convolution\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1553' of type conv\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1553_pad_type_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1553_pad_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op reg_dist_reduced : squeeze\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_dist_reduced' of type squeeze\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'reg_dist_reduced_axes_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1555 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1555' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1556 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_5' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_5_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1556' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1557 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op cls_score_list.1 : cat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_score_list.1' of type concat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_score_list.1_interleave_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1559 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1559' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op cls_score_list : permute\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cls_score_list' of type transpose\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1561 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1562 : cat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1562' of type concat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1562_interleave_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1563 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1563_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_9' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op h.7 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'h.7' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1565 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1565_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_10' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op w.7 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'w.7' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1567 : arange\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1567' of type range_1d\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1567_start_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1567_step_0' of type const\n", + "[2023-05-18 14:24:24] WARNING - type_tensor.py - Saving value type of int64 into a builtin type of int32, might lose precision!\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1568 : constant\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1568' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1569 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1567_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1569' of type add\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1570 : arange\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1570' of type range_1d\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1570_start_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1570_step_0' of type const\n", + "[2023-05-18 14:24:24] WARNING - type_tensor.py - Saving value type of int64 into a builtin type of int32, might lose precision!\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1571 : constant\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1571' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1572 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1570_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1572' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1573 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1574 : meshgrid\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_view_0' of type reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_view_0_shape_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_expand_0' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_expand_0_reps_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_view_1' of type reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_view_1_shape_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_expand_1' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1574_expand_1_reps_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op shift_y.1 : listunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1577 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1578 : stack\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1578' of type stack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op anchor_point.1 : to\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'anchor_point.1' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1580 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1580' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1581 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_6' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_6_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1581' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1582 : mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1582' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1583 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1583' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1584 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1584' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1585 : full\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1585' of type fill\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1585_value_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1586 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1586_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_11' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op h.9 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'h.9' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1588 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1588_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_12' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op w.9 : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'w.9' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1590 : arange\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1590' of type range_1d\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1590_start_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1590_step_0' of type const\n", + "[2023-05-18 14:24:24] WARNING - type_tensor.py - Saving value type of int64 into a builtin type of int32, might lose precision!\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1591 : constant\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1591' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1592 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1590_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1592' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1593 : arange\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1593' of type range_1d\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1593_start_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1593_step_0' of type const\n", + "[2023-05-18 14:24:24] WARNING - type_tensor.py - Saving value type of int64 into a builtin type of int32, might lose precision!\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1594 : constant\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1594' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1595 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1593_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1595' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1596 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1597 : meshgrid\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_view_0' of type reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_view_0_shape_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_expand_0' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_expand_0_reps_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_view_1' of type reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_view_1_shape_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_expand_1' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1597_expand_1_reps_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op shift_y.3 : listunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1600 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1601 : stack\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1601' of type stack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op anchor_point.3 : to\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'anchor_point.3' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1603 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1603' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1604 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_7' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_7_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1604' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1605 : mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1605' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1606 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1606' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1607 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1607' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1608 : full\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1608' of type fill\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1608_value_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1609 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1609_shape' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_13' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op h : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'h' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1611 : size\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1611_shape' of type shape\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_14' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op w : numtotensor\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'w' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1613 : arange\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1613' of type range_1d\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1613_start_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1613_step_0' of type const\n", + "[2023-05-18 14:24:24] WARNING - type_tensor.py - Saving value type of int64 into a builtin type of int32, might lose precision!\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1614 : constant\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1614' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1615 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1613_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1615' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1616 : arange\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1616' of type range_1d\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1616_start_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1616_step_0' of type const\n", + "[2023-05-18 14:24:24] WARNING - type_tensor.py - Saving value type of int64 into a builtin type of int32, might lose precision!\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1617 : constant\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1617' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1618 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1616_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1618' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1619 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1620 : meshgrid\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_view_0' of type reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_view_0_shape_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_expand_0' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_expand_0_reps_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_view_1' of type reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_view_1_shape_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_expand_1' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1620_expand_1_reps_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op shift_y : listunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1623 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1624 : stack\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1624' of type stack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op anchor_point : to\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'anchor_point' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1626 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1626' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1627 : reshape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_8' of type cast\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'cast_8_dtype_0' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1627' of type reshape\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1628 : mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1628' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1629 : int\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1629' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1630 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1630' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1631 : full\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1631' of type fill\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1631_value_0' of type const\n", + "Converting PyTorch Frontend ==> MIL Ops: 98%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▊ | 899/919 [00:01<00:00, 717.35 ops/s][2023-05-18 14:24:24] INFO - ops.py - Converting op 1632 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op anchor_points : cat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'anchor_points' of type concat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'anchor_points_interleave_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1634 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op stride_tensor.1 : cat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'stride_tensor.1' of type concat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'stride_tensor.1_interleave_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op points : to\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'points' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op stride_tensor : to\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'stride_tensor' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1638 : sigmoid\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1638' of type sigmoid\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1639 : split\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'shape_0' of type shape\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_15' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'floor_div_0' of type floor_div\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'mod_0' of type mod\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_16' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'mul_0' of type mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'mul_1' of type mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'tile_0' of type tile\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1639' of type split\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1639_axis_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op lt : listunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1642 : neg\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_17' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'const_17_promoted' of type const\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1642' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op x1y1 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'x1y1' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op x2y2 : add\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op 'x2y2' of type add\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1645 : listconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1646 : cat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1646' of type concat\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1646_interleave_0' of type const\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1647 : mul\n", + "[2023-05-18 14:24:24] INFO - builder.py - Adding op '1647' of type mul\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1648 : tupleconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1649 : tupleunpack\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1651 : tupleconstruct\n", + "[2023-05-18 14:24:24] INFO - ops.py - Converting op 1652 : tupleunpack\n", + "Converting PyTorch Frontend ==> MIL Ops: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▋| 917/919 [00:01<00:00, 857.86 ops/s]\n", + "Running MIL frontend_pytorch pipeline: 0%| | 0/5 [00:00 It can definitely fit into an iOS app or OS X Desktop application." + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "chMlXWPNNwTx" + }, + "source": [ + "## Visualize the CoreML model with Netron" + ] + }, + { + "cell_type": "markdown", + "metadata": { + "id": "CtXXUyL35ZQf" + }, + "source": [ + "If the notebook runs on localhost, you can visualize the exported CoreML model to Netron.
\n", + "If you use Google Colab, you can download the model in the directory explorer on the left, and upload it to https://netron.app/ instead of self-installing. " + ] + }, + { + "cell_type": "code", + "execution_count": 8, + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/" + }, + "id": "iU5hZ91TNwTy", + "is_executing": true, + "outputId": "c9544434-ce23-404f-fb04-62e0d5055a03" + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Collecting netron\n", + " Downloading netron-6.8.9-py3-none-any.whl (1.6 MB)\n", + "\u001B[2K \u001B[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001B[0m \u001B[32m1.6/1.6 MB\u001B[0m \u001B[31m4.8 MB/s\u001B[0m eta \u001B[36m0:00:00\u001B[0m00:01\u001B[0m00:01\u001B[0m0m\n", + "\u001B[?25hInstalling collected packages: netron\n", + "Successfully installed netron-6.8.9\n" + ] + } + ], + "source": [ + "!python3 -m pip install netron" + ] + }, + { + "cell_type": "code", + "execution_count": 9, + "metadata": { + "colab": { + "base_uri": "https://localhost:8080/", + "height": 1000 + }, + "id": "XynPAOQbNwTy", + "is_executing": true, + "outputId": "f20d4bcd-9026-4473-c161-04605e056521" + }, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Serving 'deci-yolo-nas-s.mlmodel' at http://localhost:8080\n" + ] + }, + { + "data": { + "text/html": [ + "\n", + " \n", + " " + ], + "text/plain": [ + "" + ] + }, + "metadata": {}, + "output_type": "display_data" + } + ], + "source": [ + "import IPython\n", + "import threading\n", + "import time\n", + "import os\n", + "\n", + "\n", + "def display_netron(path):\n", + " os.system(f'netron {path}')\n", + "\n", + "\n", + "thread = threading.Thread(target=display_netron, args=(output_path,))\n", + "thread.start()\n", + "\n", + "time.sleep(1)\n", + "display(IPython.display.IFrame(f\"http://localhost:8080\", width=1000, height=1000))" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": { + "id": "ulv8jfPiNwTz", + "is_executing": true + }, + "outputs": [], + "source": [] + } + ], + "metadata": { + "colab": { + "provenance": [] + }, + "kernelspec": { + "display_name": "Python 3 (ipykernel)", + "language": "python", + "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.8.10" + } + }, + "nbformat": 4, + "nbformat_minor": 1 +} diff --git a/utils_script/create_sub_coco.py b/utils_script/create_sub_coco.py index ee6654d6d3..312b66b218 100644 --- a/utils_script/create_sub_coco.py +++ b/utils_script/create_sub_coco.py @@ -34,7 +34,7 @@ def _copy_to_new_dir(mode: str, n_images: int, input_data_dir: Path, dest_data_d image_ids = {instance["id"] for instance in instances["images"]} annotation_image_ids = {instance["image_id"] for instance in instances["annotations"]} - kept_image_ids = list(image_ids & annotation_image_ids)[: n_images] # Make sure that the ids taken include both image and annotation + kept_image_ids = list(image_ids & annotation_image_ids)[:n_images] # Make sure that the ids taken include both image and annotation kept_annotations = [image for image in instances["annotations"] if image["image_id"] in kept_image_ids] kept_images = [image for image in instances["images"] if image["id"] in kept_image_ids] @@ -54,8 +54,8 @@ def _copy_to_new_dir(mode: str, n_images: int, input_data_dir: Path, dest_data_d if __name__ == "__main__": - parser = argparse.ArgumentParser(description='Extract a sub set of Coco into specified dir') - parser.add_argument('--input_data_dir', help='Where the full coco dataset is stored', default="/data/coco") - parser.add_argument('--dest_data_dir', help='Where the resulting data should be stored', required=True) + parser = argparse.ArgumentParser(description="Extract a sub set of Coco into specified dir") + parser.add_argument("--input_data_dir", help="Where the full coco dataset is stored", default="/data/coco") + parser.add_argument("--dest_data_dir", help="Where the resulting data should be stored", required=True) args = parser.parse_args() copy_train_val_to_new_dir(args.input_data_dir, args.dest_data_dir)