WARNING:tensorflow:Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them. __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== Input (InputLayer) (None, 3, None, None 0 __________________________________________________________________________________________________ conv1 (Conv2D) (None, 64, None, Non 9408 Input[0][0] __________________________________________________________________________________________________ bn_conv1 (BatchNormalization) (None, 64, None, Non 256 conv1[0][0] __________________________________________________________________________________________________ activation_2 (Activation) (None, 64, None, Non 0 bn_conv1[0][0] __________________________________________________________________________________________________ block_1a_conv_1 (Conv2D) (None, 64, None, Non 36864 activation_2[0][0] __________________________________________________________________________________________________ block_1a_bn_1 (BatchNormalizati (None, 64, None, Non 256 block_1a_conv_1[0][0] __________________________________________________________________________________________________ block_1a_relu_1 (Activation) (None, 64, None, Non 0 block_1a_bn_1[0][0] __________________________________________________________________________________________________ block_1a_conv_2 (Conv2D) (None, 64, None, Non 36864 block_1a_relu_1[0][0] __________________________________________________________________________________________________ block_1a_conv_shortcut (Conv2D) (None, 64, None, Non 4096 activation_2[0][0] __________________________________________________________________________________________________ block_1a_bn_2 (BatchNormalizati (None, 64, None, Non 256 block_1a_conv_2[0][0] __________________________________________________________________________________________________ block_1a_bn_shortcut (BatchNorm (None, 64, None, Non 256 block_1a_conv_shortcut[0][0] __________________________________________________________________________________________________ add_9 (Add) (None, 64, None, Non 0 block_1a_bn_2[0][0] block_1a_bn_shortcut[0][0] __________________________________________________________________________________________________ block_1a_relu (Activation) (None, 64, None, Non 0 add_9[0][0] __________________________________________________________________________________________________ block_1b_conv_1 (Conv2D) (None, 64, None, Non 36864 block_1a_relu[0][0] __________________________________________________________________________________________________ block_1b_bn_1 (BatchNormalizati (None, 64, None, Non 256 block_1b_conv_1[0][0] __________________________________________________________________________________________________ block_1b_relu_1 (Activation) (None, 64, None, Non 0 block_1b_bn_1[0][0] __________________________________________________________________________________________________ block_1b_conv_2 (Conv2D) (None, 64, None, Non 36864 block_1b_relu_1[0][0] __________________________________________________________________________________________________ block_1b_conv_shortcut (Conv2D) (None, 64, None, Non 4096 block_1a_relu[0][0] __________________________________________________________________________________________________ block_1b_bn_2 (BatchNormalizati (None, 64, None, Non 256 block_1b_conv_2[0][0] __________________________________________________________________________________________________ block_1b_bn_shortcut (BatchNorm (None, 64, None, Non 256 block_1b_conv_shortcut[0][0] __________________________________________________________________________________________________ add_10 (Add) (None, 64, None, Non 0 block_1b_bn_2[0][0] block_1b_bn_shortcut[0][0] __________________________________________________________________________________________________ block_1b_relu (Activation) (None, 64, None, Non 0 add_10[0][0] __________________________________________________________________________________________________ block_2a_conv_1 (Conv2D) (None, 128, None, No 73728 block_1b_relu[0][0] __________________________________________________________________________________________________ block_2a_bn_1 (BatchNormalizati (None, 128, None, No 512 block_2a_conv_1[0][0] __________________________________________________________________________________________________ block_2a_relu_1 (Activation) (None, 128, None, No 0 block_2a_bn_1[0][0] __________________________________________________________________________________________________ block_2a_conv_2 (Conv2D) (None, 128, None, No 147456 block_2a_relu_1[0][0] __________________________________________________________________________________________________ block_2a_conv_shortcut (Conv2D) (None, 128, None, No 8192 block_1b_relu[0][0] __________________________________________________________________________________________________ block_2a_bn_2 (BatchNormalizati (None, 128, None, No 512 block_2a_conv_2[0][0] __________________________________________________________________________________________________ block_2a_bn_shortcut (BatchNorm (None, 128, None, No 512 block_2a_conv_shortcut[0][0] __________________________________________________________________________________________________ add_11 (Add) (None, 128, None, No 0 block_2a_bn_2[0][0] block_2a_bn_shortcut[0][0] __________________________________________________________________________________________________ block_2a_relu (Activation) (None, 128, None, No 0 add_11[0][0] __________________________________________________________________________________________________ block_2b_conv_1 (Conv2D) (None, 128, None, No 147456 block_2a_relu[0][0] __________________________________________________________________________________________________ block_2b_bn_1 (BatchNormalizati (None, 128, None, No 512 block_2b_conv_1[0][0] __________________________________________________________________________________________________ block_2b_relu_1 (Activation) (None, 128, None, No 0 block_2b_bn_1[0][0] __________________________________________________________________________________________________ block_2b_conv_2 (Conv2D) (None, 128, None, No 147456 block_2b_relu_1[0][0] __________________________________________________________________________________________________ block_2b_conv_shortcut (Conv2D) (None, 128, None, No 16384 block_2a_relu[0][0] __________________________________________________________________________________________________ block_2b_bn_2 (BatchNormalizati (None, 128, None, No 512 block_2b_conv_2[0][0] __________________________________________________________________________________________________ block_2b_bn_shortcut (BatchNorm (None, 128, None, No 512 block_2b_conv_shortcut[0][0] __________________________________________________________________________________________________ add_12 (Add) (None, 128, None, No 0 block_2b_bn_2[0][0] block_2b_bn_shortcut[0][0] __________________________________________________________________________________________________ block_2b_relu (Activation) (None, 128, None, No 0 add_12[0][0] __________________________________________________________________________________________________ block_3a_conv_1 (Conv2D) (None, 256, None, No 294912 block_2b_relu[0][0] __________________________________________________________________________________________________ block_3a_bn_1 (BatchNormalizati (None, 256, None, No 1024 block_3a_conv_1[0][0] __________________________________________________________________________________________________ block_3a_relu_1 (Activation) (None, 256, None, No 0 block_3a_bn_1[0][0] __________________________________________________________________________________________________ block_3a_conv_2 (Conv2D) (None, 256, None, No 589824 block_3a_relu_1[0][0] __________________________________________________________________________________________________ block_3a_conv_shortcut (Conv2D) (None, 256, None, No 32768 block_2b_relu[0][0] __________________________________________________________________________________________________ block_3a_bn_2 (BatchNormalizati (None, 256, None, No 1024 block_3a_conv_2[0][0] __________________________________________________________________________________________________ block_3a_bn_shortcut (BatchNorm (None, 256, None, No 1024 block_3a_conv_shortcut[0][0] __________________________________________________________________________________________________ add_13 (Add) (None, 256, None, No 0 block_3a_bn_2[0][0] block_3a_bn_shortcut[0][0] __________________________________________________________________________________________________ block_3a_relu (Activation) (None, 256, None, No 0 add_13[0][0] __________________________________________________________________________________________________ block_3b_conv_1 (Conv2D) (None, 256, None, No 589824 block_3a_relu[0][0] __________________________________________________________________________________________________ block_3b_bn_1 (BatchNormalizati (None, 256, None, No 1024 block_3b_conv_1[0][0] __________________________________________________________________________________________________ block_3b_relu_1 (Activation) (None, 256, None, No 0 block_3b_bn_1[0][0] __________________________________________________________________________________________________ block_3b_conv_2 (Conv2D) (None, 256, None, No 589824 block_3b_relu_1[0][0] __________________________________________________________________________________________________ block_3b_conv_shortcut (Conv2D) (None, 256, None, No 65536 block_3a_relu[0][0] __________________________________________________________________________________________________ block_3b_bn_2 (BatchNormalizati (None, 256, None, No 1024 block_3b_conv_2[0][0] __________________________________________________________________________________________________ block_3b_bn_shortcut (BatchNorm (None, 256, None, No 1024 block_3b_conv_shortcut[0][0] __________________________________________________________________________________________________ add_14 (Add) (None, 256, None, No 0 block_3b_bn_2[0][0] block_3b_bn_shortcut[0][0] __________________________________________________________________________________________________ block_3b_relu (Activation) (None, 256, None, No 0 add_14[0][0] __________________________________________________________________________________________________ block_4a_conv_1 (Conv2D) (None, 512, None, No 1179648 block_3b_relu[0][0] __________________________________________________________________________________________________ block_4a_bn_1 (BatchNormalizati (None, 512, None, No 2048 block_4a_conv_1[0][0] __________________________________________________________________________________________________ block_4a_relu_1 (Activation) (None, 512, None, No 0 block_4a_bn_1[0][0] __________________________________________________________________________________________________ block_4a_conv_2 (Conv2D) (None, 512, None, No 2359296 block_4a_relu_1[0][0] __________________________________________________________________________________________________ block_4a_conv_shortcut (Conv2D) (None, 512, None, No 131072 block_3b_relu[0][0] __________________________________________________________________________________________________ block_4a_bn_2 (BatchNormalizati (None, 512, None, No 2048 block_4a_conv_2[0][0] __________________________________________________________________________________________________ block_4a_bn_shortcut (BatchNorm (None, 512, None, No 2048 block_4a_conv_shortcut[0][0] __________________________________________________________________________________________________ add_15 (Add) (None, 512, None, No 0 block_4a_bn_2[0][0] block_4a_bn_shortcut[0][0] __________________________________________________________________________________________________ block_4a_relu (Activation) (None, 512, None, No 0 add_15[0][0] __________________________________________________________________________________________________ block_4b_conv_1 (Conv2D) (None, 512, None, No 2359296 block_4a_relu[0][0] __________________________________________________________________________________________________ block_4b_bn_1 (BatchNormalizati (None, 512, None, No 2048 block_4b_conv_1[0][0] __________________________________________________________________________________________________ block_4b_relu_1 (Activation) (None, 512, None, No 0 block_4b_bn_1[0][0] __________________________________________________________________________________________________ block_4b_conv_2 (Conv2D) (None, 512, None, No 2359296 block_4b_relu_1[0][0] __________________________________________________________________________________________________ block_4b_conv_shortcut (Conv2D) (None, 512, None, No 262144 block_4a_relu[0][0] __________________________________________________________________________________________________ block_4b_bn_2 (BatchNormalizati (None, 512, None, No 2048 block_4b_conv_2[0][0] __________________________________________________________________________________________________ block_4b_bn_shortcut (BatchNorm (None, 512, None, No 2048 block_4b_conv_shortcut[0][0] __________________________________________________________________________________________________ add_16 (Add) (None, 512, None, No 0 block_4b_bn_2[0][0] block_4b_bn_shortcut[0][0] __________________________________________________________________________________________________ block_4b_relu (Activation) (None, 512, None, No 0 add_16[0][0] __________________________________________________________________________________________________ yolo_expand_conv1 (Conv2D) (None, 512, None, No 2359296 block_4b_relu[0][0] __________________________________________________________________________________________________ yolo_expand_conv1_bn (BatchNorm (None, 512, None, No 2048 yolo_expand_conv1[0][0] __________________________________________________________________________________________________ yolo_expand_conv1_lrelu (LeakyR (None, 512, None, No 0 yolo_expand_conv1_bn[0][0] __________________________________________________________________________________________________ yolo_conv1_1 (Conv2D) (None, 256, None, No 131072 yolo_expand_conv1_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv1_1_bn (BatchNormaliza (None, 256, None, No 1024 yolo_conv1_1[0][0] __________________________________________________________________________________________________ yolo_conv1_1_lrelu (LeakyReLU) (None, 256, None, No 0 yolo_conv1_1_bn[0][0] __________________________________________________________________________________________________ yolo_conv1_2 (Conv2D) (None, 512, None, No 1179648 yolo_conv1_1_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv1_2_bn (BatchNormaliza (None, 512, None, No 2048 yolo_conv1_2[0][0] __________________________________________________________________________________________________ yolo_conv1_2_lrelu (LeakyReLU) (None, 512, None, No 0 yolo_conv1_2_bn[0][0] __________________________________________________________________________________________________ yolo_conv1_3 (Conv2D) (None, 256, None, No 131072 yolo_conv1_2_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv1_3_bn (BatchNormaliza (None, 256, None, No 1024 yolo_conv1_3[0][0] __________________________________________________________________________________________________ yolo_conv1_3_lrelu (LeakyReLU) (None, 256, None, No 0 yolo_conv1_3_bn[0][0] __________________________________________________________________________________________________ yolo_conv1_4 (Conv2D) (None, 512, None, No 1179648 yolo_conv1_3_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv1_4_bn (BatchNormaliza (None, 512, None, No 2048 yolo_conv1_4[0][0] __________________________________________________________________________________________________ yolo_conv1_4_lrelu (LeakyReLU) (None, 512, None, No 0 yolo_conv1_4_bn[0][0] __________________________________________________________________________________________________ yolo_conv1_5 (Conv2D) (None, 256, None, No 131072 yolo_conv1_4_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv1_5_bn (BatchNormaliza (None, 256, None, No 1024 yolo_conv1_5[0][0] __________________________________________________________________________________________________ yolo_conv1_5_lrelu (LeakyReLU) (None, 256, None, No 0 yolo_conv1_5_bn[0][0] __________________________________________________________________________________________________ yolo_conv2 (Conv2D) (None, 128, None, No 32768 yolo_conv1_5_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv2_bn (BatchNormalizati (None, 128, None, No 512 yolo_conv2[0][0] __________________________________________________________________________________________________ yolo_conv2_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv2_bn[0][0] __________________________________________________________________________________________________ upsample0 (UpSampling2D) (None, 128, None, No 0 yolo_conv2_lrelu[0][0] __________________________________________________________________________________________________ concatenate_3 (Concatenate) (None, 384, None, No 0 upsample0[0][0] block_3b_relu[0][0] __________________________________________________________________________________________________ yolo_conv3_1 (Conv2D) (None, 128, None, No 49152 concatenate_3[0][0] __________________________________________________________________________________________________ yolo_conv3_1_bn (BatchNormaliza (None, 128, None, No 512 yolo_conv3_1[0][0] __________________________________________________________________________________________________ yolo_conv3_1_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv3_1_bn[0][0] __________________________________________________________________________________________________ yolo_conv3_2 (Conv2D) (None, 256, None, No 294912 yolo_conv3_1_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv3_2_bn (BatchNormaliza (None, 256, None, No 1024 yolo_conv3_2[0][0] __________________________________________________________________________________________________ yolo_conv3_2_lrelu (LeakyReLU) (None, 256, None, No 0 yolo_conv3_2_bn[0][0] __________________________________________________________________________________________________ yolo_conv3_3 (Conv2D) (None, 128, None, No 32768 yolo_conv3_2_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv3_3_bn (BatchNormaliza (None, 128, None, No 512 yolo_conv3_3[0][0] __________________________________________________________________________________________________ yolo_conv3_3_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv3_3_bn[0][0] __________________________________________________________________________________________________ yolo_conv3_4 (Conv2D) (None, 256, None, No 294912 yolo_conv3_3_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv3_4_bn (BatchNormaliza (None, 256, None, No 1024 yolo_conv3_4[0][0] __________________________________________________________________________________________________ yolo_conv3_4_lrelu (LeakyReLU) (None, 256, None, No 0 yolo_conv3_4_bn[0][0] __________________________________________________________________________________________________ yolo_conv3_5 (Conv2D) (None, 128, None, No 32768 yolo_conv3_4_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv3_5_bn (BatchNormaliza (None, 128, None, No 512 yolo_conv3_5[0][0] __________________________________________________________________________________________________ yolo_conv3_5_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv3_5_bn[0][0] __________________________________________________________________________________________________ yolo_conv4 (Conv2D) (None, 64, None, Non 8192 yolo_conv3_5_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv4_bn (BatchNormalizati (None, 64, None, Non 256 yolo_conv4[0][0] __________________________________________________________________________________________________ yolo_conv4_lrelu (LeakyReLU) (None, 64, None, Non 0 yolo_conv4_bn[0][0] __________________________________________________________________________________________________ upsample1 (UpSampling2D) (None, 64, None, Non 0 yolo_conv4_lrelu[0][0] __________________________________________________________________________________________________ concatenate_4 (Concatenate) (None, 192, None, No 0 upsample1[0][0] block_2b_relu[0][0] __________________________________________________________________________________________________ yolo_conv5_1 (Conv2D) (None, 64, None, Non 12288 concatenate_4[0][0] __________________________________________________________________________________________________ yolo_conv5_1_bn (BatchNormaliza (None, 64, None, Non 256 yolo_conv5_1[0][0] __________________________________________________________________________________________________ yolo_conv5_1_lrelu (LeakyReLU) (None, 64, None, Non 0 yolo_conv5_1_bn[0][0] __________________________________________________________________________________________________ yolo_conv5_2 (Conv2D) (None, 128, None, No 73728 yolo_conv5_1_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv5_2_bn (BatchNormaliza (None, 128, None, No 512 yolo_conv5_2[0][0] __________________________________________________________________________________________________ yolo_conv5_2_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv5_2_bn[0][0] __________________________________________________________________________________________________ yolo_conv5_3 (Conv2D) (None, 64, None, Non 8192 yolo_conv5_2_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv5_3_bn (BatchNormaliza (None, 64, None, Non 256 yolo_conv5_3[0][0] __________________________________________________________________________________________________ yolo_conv5_3_lrelu (LeakyReLU) (None, 64, None, Non 0 yolo_conv5_3_bn[0][0] __________________________________________________________________________________________________ yolo_conv5_4 (Conv2D) (None, 128, None, No 73728 yolo_conv5_3_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv5_4_bn (BatchNormaliza (None, 128, None, No 512 yolo_conv5_4[0][0] __________________________________________________________________________________________________ yolo_conv5_4_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv5_4_bn[0][0] __________________________________________________________________________________________________ yolo_conv5_5 (Conv2D) (None, 64, None, Non 8192 yolo_conv5_4_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv5_5_bn (BatchNormaliza (None, 64, None, Non 256 yolo_conv5_5[0][0] __________________________________________________________________________________________________ yolo_conv5_5_lrelu (LeakyReLU) (None, 64, None, Non 0 yolo_conv5_5_bn[0][0] __________________________________________________________________________________________________ yolo_conv1_6 (Conv2D) (None, 512, None, No 1179648 yolo_conv1_5_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv3_6 (Conv2D) (None, 256, None, No 294912 yolo_conv3_5_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv5_6 (Conv2D) (None, 128, None, No 73728 yolo_conv5_5_lrelu[0][0] __________________________________________________________________________________________________ yolo_conv1_6_bn (BatchNormaliza (None, 512, None, No 2048 yolo_conv1_6[0][0] __________________________________________________________________________________________________ yolo_conv3_6_bn (BatchNormaliza (None, 256, None, No 1024 yolo_conv3_6[0][0] __________________________________________________________________________________________________ yolo_conv5_6_bn (BatchNormaliza (None, 128, None, No 512 yolo_conv5_6[0][0] __________________________________________________________________________________________________ yolo_conv1_6_lrelu (LeakyReLU) (None, 512, None, No 0 yolo_conv1_6_bn[0][0] __________________________________________________________________________________________________ yolo_conv3_6_lrelu (LeakyReLU) (None, 256, None, No 0 yolo_conv3_6_bn[0][0] __________________________________________________________________________________________________ yolo_conv5_6_lrelu (LeakyReLU) (None, 128, None, No 0 yolo_conv5_6_bn[0][0] __________________________________________________________________________________________________ conv_big_object (Conv2D) (None, 33, None, Non 16929 yolo_conv1_6_lrelu[0][0] __________________________________________________________________________________________________ conv_mid_object (Conv2D) (None, 33, None, Non 8481 yolo_conv3_6_lrelu[0][0] __________________________________________________________________________________________________ conv_sm_object (Conv2D) (None, 33, None, Non 4257 yolo_conv5_6_lrelu[0][0] __________________________________________________________________________________________________ bg_permute (Permute) (None, None, None, 3 0 conv_big_object[0][0] __________________________________________________________________________________________________ md_permute (Permute) (None, None, None, 3 0 conv_mid_object[0][0] __________________________________________________________________________________________________ sm_permute (Permute) (None, None, None, 3 0 conv_sm_object[0][0] __________________________________________________________________________________________________ bg_anchor (YOLOAnchorBox) (None, None, 6) 0 conv_big_object[0][0] __________________________________________________________________________________________________ bg_reshape (Reshape) (None, None, 11) 0 bg_permute[0][0] __________________________________________________________________________________________________ md_anchor (YOLOAnchorBox) (None, None, 6) 0 conv_mid_object[0][0] __________________________________________________________________________________________________ md_reshape (Reshape) (None, None, 11) 0 md_permute[0][0] __________________________________________________________________________________________________ sm_anchor (YOLOAnchorBox) (None, None, 6) 0 conv_sm_object[0][0] __________________________________________________________________________________________________ sm_reshape (Reshape) (None, None, 11) 0 sm_permute[0][0] __________________________________________________________________________________________________ encoded_bg (Concatenate) (None, None, 17) 0 bg_anchor[0][0] bg_reshape[0][0] __________________________________________________________________________________________________ encoded_md (Concatenate) (None, None, 17) 0 md_anchor[0][0] md_reshape[0][0] __________________________________________________________________________________________________ encoded_sm (Concatenate) (None, None, 17) 0 sm_anchor[0][0] sm_reshape[0][0] __________________________________________________________________________________________________/usr/local/lib/python3.6/dist-packages/requests/__init__.py:91: RequestsDependencyWarning: urllib3 (1.26.5) or chardet (3.0.4) doesn't match a supported version! RequestsDependencyWarning) Using TensorFlow backend. 2022-06-08 15:31:45,680 [INFO] modulus.pruning.pruning: Exploring graph for retainable indices 2022-06-08 15:31:47,147 [INFO] modulus.pruning.pruning: Pruning model and appending pruned nodes to new graph encoded_detections (Concatenate (None, None, 17) 0 encoded_bg[0][0] encoded_md[0][0] encoded_sm[0][0] ================================================================================================== Total params: 19,172,771 Trainable params: 19,151,651 Non-trainable params: 21,120 __________________________________________________________________________________________________ Traceback (most recent call last): File "/root/.cache/bazel/_bazel_root/b770f990bb7b9e2db5771981fb3a38b4/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/iva/build_wheel.runfiles/ai_infra/iva/detectnet_v2/scripts/prune.py", line 16, in File "/root/.cache/bazel/_bazel_root/b770f990bb7b9e2db5771981fb3a38b4/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/iva/build_wheel.runfiles/ai_infra/iva/common/magnet_prune.py", line 266, in main File "/root/.cache/bazel/_bazel_root/b770f990bb7b9e2db5771981fb3a38b4/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/iva/build_wheel.runfiles/ai_infra/iva/common/magnet_prune.py", line 227, in run_pruning File "/root/.cache/bazel/_bazel_root/b770f990bb7b9e2db5771981fb3a38b4/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/core/build_wheel.runfiles/ai_infra/moduluspy/modulus/pruning/pruning.py", line 1613, in prune File "/root/.cache/bazel/_bazel_root/b770f990bb7b9e2db5771981fb3a38b4/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/core/build_wheel.runfiles/ai_infra/moduluspy/modulus/pruning/pruning.py", line 1289, in prune File "/root/.cache/bazel/_bazel_root/b770f990bb7b9e2db5771981fb3a38b4/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/core/build_wheel.runfiles/ai_infra/moduluspy/modulus/pruning/pruning.py", line 659, in _prune_explored_concat_layer TypeError: 'NoneType' object cannot be interpreted as an integer