junmcenroe / YOLOv5-CoreML-Converter

YOLOv5 CoreML Coverter using export.py with train mode
1 stars 0 forks source link

index out of range when adding detect & NMS layers #1

Open Mercury-ML opened 1 year ago

Mercury-ML commented 1 year ago

Thank you for posting this CoreML converter for YOLOv5!

I'm running into an error at the last step:

Traceback (most recent call last): File "/content/YOLOv5CoreMLConverter.py", line 331, in main() File "/content/YOLOv5CoreMLConverter.py", line 322, in main addExportLayerToCoreml(builder,featureMapDimensions,strides,anchorGrid,numberOfClassLabels,imgSize) File "/content/YOLOv5CoreMLConverter.py", line 99, in addExportLayerToCoreml featureMapDimensions[i], featureMapDimensions[i]).numpy() IndexError: list index out of range

My files were trained at 1280 using YOLOv5m . There are only 4 labels.
here's the command I used in Colab: !python /content/YOLOv5CoreMLConverter.py -input /content/yolov5/v418e-best.mlmodel -size 1280 -label /content/counter.txt -output /content -name v4e18-iOS.mlmodel

I created a text file, "counter.txt" , with the 4 labels in order.

Any advice or thoughts on how to overcome this would be greatly appreciated!

junmcenroe commented 1 year ago

Dear @Mercury-ML

I checked with 1280 image size using YOLOv5m of coco-dataset. As you see, I can transform to CoreML model. One possibility of failure is that original export.py is not support of train mode export. So we need to exchange from original export.py to my export.py. Can you try again after replace export.py?

1280 yolov5m  coreml
Mercury-ML commented 1 year ago

Thank you for looking into this! I see you were able to convert the model using 1280 & Yolov5m without issue Again, any advice or help is greatly appreciated!

I'm pretty sure I was using your export.py because I was using your notebook which switches it out (saving the old one as export_org.py)

but just to be sure I ran Yolov5-coreML-junmcenroe.ipynb again, changing out the model, yaml & txt files with my own, and had the same results.

Could there be a problem with the yaml or txt file? (see below) I am unfamiliar with train mode export, does it require an image for tracing by chance?

BTW, this model was trained on 9/7/22 using Ultralytics most current repo

filename: counter.yaml names: 0: label_a 1: label_b 2: label_c 3: label_d nc: 4 train: /content/yolov5/custom_dataset/train val: /content/yolov5/custom_dataset/val

filename: counter.txt label_a label_b label_c label_d

Here's the last cell command: !python /content/YOLOv5CoreMLConverter.py -input /content/yolov5/v418e-best.mlmodel -size 1280 -label /content/counter.txt -output /content -name v4e18-out.mlmodel

Output: WARNING:root:TensorFlow version 2.9.2 has not been tested with coremltools. You may run into unexpected errors. TensorFlow 2.8.0 is the most recent version that has been tested. 1280 ['label_a', 'label_b', 'label_c', 'label_d'] /usr/local/lib/python3.7/dist-packages/torch/functional.py:478: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:2894.) return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined] Traceback (most recent call last): File "/content/YOLOv5CoreMLConverter.py", line 331, in main() File "/content/YOLOv5CoreMLConverter.py", line 322, in main addExportLayerToCoreml(builder,featureMapDimensions,strides,anchorGrid,numberOfClassLabels,imgSize) File "/content/YOLOv5CoreMLConverter.py", line 99, in addExportLayerToCoreml featureMapDimensions[i], featureMapDimensions[i]).numpy() IndexError: list index out of range

Here's the export cell: !python export.py --data /content/counter.yaml --img-size 1280 --weights v418e-best.pt --train --include coreml

Output:

export: data=/content/counter.yaml, weights=['v418e-best.pt'], imgsz=[1280], batch_size=1, device=cpu, half=False, inplace=False, train=True, keras=False, optimize=False, int8=False, dynamic=False, simplify=False, opset=12, verbose=False, workspace=4, nms=False, agnostic_nms=False, topk_per_class=100, topk_all=100, iou_thres=0.45, conf_thres=0.25, include=['coreml'] YOLOv5 🚀 v6.2-224-g82a5585 Python-3.7.15 torch-1.12.1+cu113 CPU

Fusing layers... Model summary: 378 layers, 35266236 parameters, 0 gradients

PyTorch: starting from v418e-best.pt with output shape (tensor([[[[[ 6.20487e-02, 4.27787e-01, 1.55988e-02, ..., -3.49689e+00, 9.28358e-02, -3.07012e+00], [ 1.95959e-02, 7.82394e-01, 6.16673e-01, ..., -3.89255e+00, 1.06197e-01, -3.44949e+00], [-3.32826e-02, 1.15501e+00, 1.15216e+00, ..., -3.89549e+00, 8.43974e-02, -3.60715e+00], ..., [-1.52949e-01, 1.28054e+00, 1.01302e+00, ..., -3.57405e+00, 4.46051e-01, -3.19782e+00], [-2.94692e-01, 1.08079e+00, 3.79013e-01, ..., -3.71270e+00, 4.39387e-01, -3.08765e+00], [ 1.49061e-01, 6.79476e-01, 1.53083e-01, ..., -3.36413e+00, 1.63019e-01, -3.14890e+00]],

      [[ 2.92739e-01,  8.10136e-01, -4.34245e-01,  ..., -3.67982e+00,  5.43849e-01, -2.74237e+00],
       [-1.21620e-01,  9.68975e-01,  2.43908e-01,  ..., -4.21346e+00,  2.04524e-01, -3.31969e+00],
       [-1.08092e-01,  6.97694e-01,  1.07222e+00,  ..., -4.42781e+00,  2.01753e-01, -3.56928e+00],
       ...,
       [-2.50997e-01,  8.48509e-01,  9.95461e-01,  ..., -4.16896e+00,  9.32239e-01, -3.24170e+00],
       [-3.23021e-01,  8.45299e-01,  2.21338e-01,  ..., -4.18204e+00,  7.31738e-01, -3.25197e+00],
       [-2.95500e-01,  5.80674e-01, -2.09653e-01,  ..., -3.64702e+00,  6.96068e-01, -3.04307e+00]],

      [[ 8.48902e-01,  3.67776e-01, -4.46961e-01,  ..., -3.70949e+00,  3.03622e-01, -3.07046e+00],
       [-3.49372e-02,  4.42261e-01,  2.42750e-01,  ..., -3.99236e+00, -3.02734e-01, -3.38781e+00],
       [-6.88530e-02,  4.77085e-01,  9.94476e-01,  ..., -4.22451e+00,  1.37349e+00, -3.25259e+00],
       ...,
       [-3.30604e-01,  7.85979e-01,  9.86488e-01,  ..., -4.02654e+00,  1.28310e+00, -3.25001e+00],
       [-5.61993e-01,  6.43333e-01,  1.93963e-01,  ..., -4.01174e+00,  5.54016e-01, -3.07534e+00],
       [-7.46767e-01,  4.80447e-01, -2.46715e-01,  ..., -3.74045e+00,  1.00889e+00, -3.23637e+00]],

      ...,

      [[ 1.08931e+00,  1.65222e-01, -3.21380e-01,  ..., -4.27011e+00, -3.90158e-01, -3.51180e+00],
       [ 1.40210e+00,  3.63183e-01,  5.82896e-01,  ..., -4.46745e+00, -1.45812e+00, -4.05723e+00],
       [ 4.07554e-01,  6.95774e-01,  1.00816e+00,  ..., -4.17873e+00, -2.77745e-01, -3.54851e+00],
       ...,
       [-7.30196e-01,  6.42292e-01,  9.01875e-01,  ..., -3.84916e+00,  3.97733e-01, -3.18358e+00],
       [-1.53193e+00,  4.18769e-01,  2.52889e-01,  ..., -3.96808e+00, -2.12899e-02, -3.05928e+00],
       [-8.53474e-01,  2.53407e-01, -3.49374e-01,  ..., -3.80941e+00,  6.96987e-01, -2.95405e+00]],

      [[ 8.51006e-01, -6.41843e-01, -2.44410e-01,  ..., -4.49477e+00, -5.26410e-01, -3.82796e+00],
       [ 8.66500e-01, -6.36076e-01,  6.62065e-01,  ..., -5.03880e+00, -1.13748e+00, -4.37412e+00],
       [ 5.32875e-01, -6.81887e-01,  1.30016e+00,  ..., -4.68717e+00, -9.56320e-01, -4.49729e+00],
       ...,
       [-7.68527e-01, -4.56323e-01,  1.06113e+00,  ..., -4.17305e+00,  4.20666e-03, -3.64451e+00],
       [-1.12423e+00, -3.66707e-01,  2.88761e-01,  ..., -4.37047e+00,  1.83992e-01, -3.56685e+00],
       [-3.22494e-01, -1.96670e-01, -1.74715e-01,  ..., -3.84181e+00,  7.15309e-01, -3.21890e+00]],

      [[ 2.43895e-01, -1.86846e-01,  2.39477e-01,  ..., -4.17392e+00, -2.60129e-01, -3.34733e+00],
       [ 1.68658e-01, -3.10196e-01,  7.81725e-01,  ..., -4.37419e+00, -3.83016e-01, -3.43009e+00],
       [ 2.92276e-01, -6.41545e-01,  1.31066e+00,  ..., -4.37250e+00, -2.40360e-01, -3.54472e+00],
       ...,
       [-4.65235e-01, -6.40174e-01,  1.09701e+00,  ..., -3.82548e+00,  3.30268e-01, -3.00235e+00],
       [-4.28108e-01, -3.10022e-01,  4.42717e-01,  ..., -3.85850e+00,  5.84293e-01, -3.02188e+00],
       [-5.86649e-02, -1.09578e-01,  2.65147e-01,  ..., -3.56154e+00,  6.94315e-01, -2.82383e+00]]],

     [[[ 7.05352e-01,  7.15968e-01, -6.20185e-01,  ..., -2.96427e+00, -2.39073e-01, -3.54185e+00],
       [ 7.48658e-01,  1.05156e+00, -2.63135e-01,  ..., -3.16006e+00, -1.61075e-01, -3.85064e+00],
       [ 6.99726e-01,  1.41370e+00,  5.42487e-02,  ..., -3.28000e+00, -4.73633e-02, -4.09248e+00],
       ...,
       [ 5.28535e-01,  1.50172e+00,  2.74436e-02,  ..., -3.06680e+00,  4.30992e-01, -3.70652e+00],
       [ 3.01384e-01,  1.37055e+00, -2.84867e-01,  ..., -3.03448e+00,  1.17064e-01, -3.55309e+00],
       [ 6.53940e-01,  8.92880e-01, -4.99064e-01,  ..., -2.91456e+00, -2.57191e-01, -3.05836e+00]],

      [[ 1.09804e+00,  1.19418e+00, -7.93662e-01,  ..., -3.03429e+00,  2.18680e-01, -3.87114e+00],
       [ 6.17988e-01,  1.20162e+00, -4.12941e-01,  ..., -3.40888e+00,  5.38087e-02, -4.16654e+00],
       [ 6.78966e-01,  9.17347e-01,  1.10520e-01,  ..., -3.75104e+00,  1.67183e-02, -4.50597e+00],
       ...,
       [ 5.02068e-01,  1.01397e+00,  1.27148e-01,  ..., -3.66743e+00,  7.59172e-01, -4.21355e+00],
       [ 2.36514e-01,  1.14816e+00, -3.48637e-01,  ..., -3.53674e+00,  4.76555e-01, -4.01915e+00],
       [ 3.17621e-01,  8.55384e-01, -6.52552e-01,  ..., -3.16293e+00,  4.91513e-01, -3.18006e+00]],

      [[ 1.63095e+00,  7.55307e-01, -7.82267e-01,  ..., -3.08968e+00,  2.66980e-01, -4.27130e+00],
       [ 6.57192e-01,  6.30595e-01, -4.69117e-01,  ..., -3.24956e+00, -2.83322e-01, -4.19190e+00],
       [ 6.00875e-01,  6.68081e-01,  2.34662e-02,  ..., -3.64290e+00,  1.39966e+00, -4.05325e+00],
       ...,
       [ 2.46257e-01,  9.56955e-01,  5.68475e-02,  ..., -3.53256e+00,  1.34607e+00, -4.26868e+00],
       [-8.43206e-02,  9.01067e-01, -3.76236e-01,  ..., -3.30907e+00,  5.33804e-01, -3.99811e+00],
       [-2.01578e-01,  7.19403e-01, -6.84374e-01,  ..., -3.27669e+00,  9.89166e-01, -3.39439e+00]],

      ...,

      [[ 1.89239e+00,  5.32558e-01, -8.03622e-01,  ..., -3.70901e+00, -2.63768e-01, -4.90592e+00],
       [ 2.10102e+00,  5.60999e-01, -3.38210e-01,  ..., -3.82669e+00, -1.33593e+00, -4.80707e+00],
       [ 1.07152e+00,  9.68595e-01,  1.16317e-02,  ..., -3.76487e+00, -2.65820e-01, -4.86057e+00],
       ...,
       [-1.64288e-01,  8.35032e-01,  4.41563e-02,  ..., -3.35977e+00,  4.00373e-01, -4.49254e+00],
       [-1.01286e+00,  6.68350e-01, -3.02878e-01,  ..., -3.23940e+00,  9.35662e-03, -4.16807e+00],
       [-2.54015e-01,  5.35437e-01, -6.66487e-01,  ..., -3.13227e+00,  6.69246e-01, -3.19355e+00]],

      [[ 1.60159e+00, -3.03052e-01, -8.62578e-01,  ..., -4.04286e+00, -5.08027e-01, -4.78706e+00],
       [ 1.52669e+00, -4.37726e-01, -3.38704e-01,  ..., -4.29678e+00, -1.09478e+00, -5.05759e+00],
       [ 1.10682e+00, -3.83707e-01,  3.77943e-02,  ..., -4.21484e+00, -1.04087e+00, -5.57551e+00],
       ...,
       [-1.74661e-01, -2.61812e-01,  9.53742e-02,  ..., -3.60856e+00, -1.02744e-01, -4.65282e+00],
       [-6.56383e-01, -1.50988e-01, -3.62031e-01,  ..., -3.68028e+00,  9.76665e-02, -4.37281e+00],
       [ 2.06850e-01,  7.22807e-02, -6.47881e-01,  ..., -3.30797e+00,  5.98690e-01, -3.48412e+00]],

      [[ 1.09488e+00,  1.39421e-01, -6.45828e-01,  ..., -3.81872e+00, -4.60960e-01, -3.83854e+00],
       [ 1.17405e+00, -1.29379e-02, -2.84782e-01,  ..., -3.66883e+00, -6.70052e-01, -3.83158e+00],
       [ 1.23229e+00, -3.47522e-01,  8.04565e-02,  ..., -3.77373e+00, -3.87711e-01, -3.99897e+00],
       ...,
       [ 4.43989e-01, -3.70635e-01,  3.24055e-02,  ..., -3.14862e+00,  1.49033e-01, -3.50545e+00],
       [ 2.74589e-01, -4.04820e-02, -2.97102e-01,  ..., -3.25055e+00,  2.42565e-01, -3.50773e+00],
       [ 5.84659e-01,  2.15219e-01, -4.29588e-01,  ..., -3.08757e+00,  3.31898e-01, -3.06972e+00]]],

     [[[ 2.67107e-02,  2.32866e-01, -5.26019e-01,  ..., -3.38957e+00,  3.65709e-01, -2.94427e+00],
       [-3.57120e-02,  6.26330e-01, -9.75466e-02,  ..., -3.86060e+00,  5.73262e-01, -3.09997e+00],
       [-2.12069e-01,  9.42022e-01,  2.26782e-01,  ..., -3.97525e+00,  8.67922e-01, -3.15382e+00],
       ...,
       [-3.25388e-01,  1.05728e+00,  1.90597e-01,  ..., -3.61662e+00,  1.19689e+00, -2.80127e+00],
       [-4.55426e-01,  8.72752e-01, -1.60052e-01,  ..., -3.77633e+00,  1.16193e+00, -2.63695e+00],
       [-7.55174e-03,  4.46073e-01, -4.12855e-01,  ..., -3.45057e+00,  6.97115e-01, -2.64083e+00]],

      [[ 3.21929e-01,  4.75396e-01, -7.43494e-01,  ..., -3.61493e+00,  6.33388e-01, -2.88857e+00],
       [-1.20457e-01,  7.01147e-01, -2.92082e-01,  ..., -4.15060e+00,  3.76301e-01, -3.31954e+00],
       [-1.72071e-01,  4.09101e-01,  2.42079e-01,  ..., -4.55984e+00,  5.21147e-01, -3.53969e+00],
       ...,
       [-3.12563e-01,  5.11613e-01,  3.04444e-01,  ..., -4.29983e+00,  1.16176e+00, -3.29997e+00],
       [-4.16546e-01,  7.28955e-01, -2.18408e-01,  ..., -4.22808e+00,  1.04065e+00, -3.11587e+00],
       [-3.17837e-01,  4.45777e-01, -5.81599e-01,  ..., -3.80764e+00,  1.05276e+00, -2.86848e+00]],

      [[ 8.98433e-01,  7.38285e-02, -7.28172e-01,  ..., -3.68707e+00,  4.03784e-01, -3.34099e+00],
       [-5.33982e-02,  2.02856e-01, -3.78546e-01,  ..., -4.15503e+00, -1.07386e-01, -3.44610e+00],
       [-1.22078e-01,  2.36349e-01,  1.67758e-01,  ..., -4.43521e+00,  1.73949e+00, -3.22920e+00],
       ...,
       [-4.20589e-01,  5.32807e-01,  2.19597e-01,  ..., -4.16159e+00,  1.63571e+00, -3.36029e+00],
       [-6.94332e-01,  4.83238e-01, -2.83524e-01,  ..., -4.15329e+00,  8.27144e-01, -3.13215e+00],
       [-7.72860e-01,  3.26781e-01, -6.29171e-01,  ..., -3.98372e+00,  1.30890e+00, -3.16482e+00]],

      ...,

      [[ 1.13428e+00, -1.31974e-01, -7.36413e-01,  ..., -4.15174e+00, -3.21477e-01, -4.05682e+00],
       [ 1.36166e+00,  1.11309e-01, -2.45227e-01,  ..., -4.56060e+00, -1.24407e+00, -4.30207e+00],
       [ 2.13163e-01,  3.41231e-01,  1.56777e-01,  ..., -4.33827e+00,  3.19035e-01, -3.96584e+00],
       ...,
       [-9.46124e-01,  3.20451e-01,  2.03142e-01,  ..., -3.88168e+00,  8.69617e-01, -3.57022e+00],
       [-1.71473e+00,  1.95298e-01, -1.88578e-01,  ..., -4.06059e+00,  3.94255e-01, -3.30045e+00],
       [-9.04832e-01,  1.15867e-01, -6.48700e-01,  ..., -3.89831e+00,  1.18411e+00, -2.98430e+00]],

      [[ 8.97992e-01, -9.30271e-01, -7.91983e-01,  ..., -4.51709e+00, -3.21029e-01, -4.08529e+00],
       [ 8.19465e-01, -9.15246e-01, -1.78276e-01,  ..., -4.94203e+00, -8.33296e-01, -4.59281e+00],
       [ 3.32781e-01, -1.01579e+00,  1.46948e-01,  ..., -4.81243e+00, -4.07483e-01, -4.88881e+00],
       ...,
       [-9.53886e-01, -8.41034e-01,  1.96383e-01,  ..., -4.24471e+00,  5.64938e-01, -3.94794e+00],
       [-1.27540e+00, -6.35046e-01, -2.56933e-01,  ..., -4.33520e+00,  6.20664e-01, -3.64897e+00],
       [-4.29614e-01, -4.07294e-01, -6.13312e-01,  ..., -3.91657e+00,  1.25307e+00, -3.21925e+00]],

      [[ 2.11534e-01, -4.78199e-01, -4.86134e-01,  ..., -4.42227e+00,  1.23875e-01, -3.29661e+00],
       [ 1.98836e-01, -6.87620e-01, -7.02214e-02,  ..., -4.64169e+00,  1.06352e-01, -3.24309e+00],
       [ 1.85524e-01, -1.08004e+00,  2.66038e-01,  ..., -4.61787e+00,  6.17470e-01, -3.24275e+00],
       ...,
       [-5.14543e-01, -1.08121e+00,  1.81634e-01,  ..., -3.97458e+00,  1.09796e+00, -2.75107e+00],
       [-5.20248e-01, -6.63695e-01, -1.80569e-01,  ..., -3.96915e+00,  1.20263e+00, -2.70895e+00],
       [-1.60012e-01, -3.60713e-01, -3.59314e-01,  ..., -3.71223e+00,  1.33209e+00, -2.52636e+00]]]]]), tensor([[[[[ 9.70841e-01,  7.19438e-02, -6.67875e-01,  ..., -7.87527e-01,  1.57784e+00, -1.92265e+00],
       [ 2.71422e-01,  7.21558e-01, -2.92675e-01,  ..., -9.89362e-01,  2.10278e+00, -2.17297e+00],
       [ 4.17649e-01,  6.41992e-01, -4.29088e-01,  ..., -4.12658e-01,  3.03501e+00, -1.87410e+00],
       ...,
       [ 4.21385e-02,  7.37713e-01, -3.51685e-01,  ..., -1.00444e+00,  3.19984e+00, -2.38292e+00],
       [ 5.05699e-01,  7.41057e-01, -2.45010e-01,  ..., -1.06096e+00,  2.43416e+00, -2.52577e+00],
       [-4.27970e-01,  3.06949e-01, -5.56618e-01,  ..., -1.28560e+00,  1.91862e+00, -2.37633e+00]],

      [[ 1.71279e+00, -3.55899e-01, -6.93878e-01,  ..., -9.00350e-01,  1.22501e+00, -2.47050e+00],
       [ 2.33447e-01, -3.74554e-01, -3.67988e-01,  ..., -1.61795e+00,  3.00187e+00, -2.55920e+00],
       [ 3.35846e-01, -3.12608e-01, -4.24376e-01,  ..., -1.57393e+00,  4.34744e+00, -2.66226e+00],
       ...,
       [-6.87682e-02, -2.38888e-01, -3.76776e-01,  ..., -2.17305e+00,  5.45215e+00, -2.88471e+00],
       [ 2.74303e-01, -3.76762e-01, -3.65020e-01,  ..., -1.88891e+00,  3.37676e+00, -3.32852e+00],
       [-1.05529e+00, -4.27080e-01, -5.84608e-01,  ..., -1.52763e+00,  1.91901e+00, -3.04346e+00]],

      [[ 1.85388e+00, -1.15976e-01, -8.03341e-01,  ..., -1.01021e+00,  1.26348e+00, -2.28847e+00],
       [ 3.57420e-01, -1.52608e-01, -4.77569e-01,  ..., -1.90254e+00,  4.32209e+00, -2.28918e+00],
       [ 4.83754e-01, -3.22993e-02, -5.31423e-01,  ..., -1.65904e+00,  4.81163e+00, -2.32564e+00],
       ...,
       [-3.25348e-03,  1.74739e-03, -4.87120e-01,  ..., -2.42548e+00,  6.00003e+00, -2.99377e+00],
       [ 3.09184e-01, -2.35279e-01, -3.98909e-01,  ..., -2.08611e+00,  5.44138e+00, -3.28648e+00],
       [-1.03749e+00, -4.04033e-01, -6.82549e-01,  ..., -1.53027e+00,  3.25867e+00, -2.64824e+00]],

      ...,

      [[ 2.06780e+00, -1.60763e-02, -7.72666e-01,  ..., -6.06710e-01,  1.60126e+00, -2.11023e+00],
       [ 5.59734e-01, -3.29757e-01, -4.32278e-01,  ..., -1.53214e+00,  5.09874e+00, -2.48182e+00],
       [ 6.46871e-01, -3.26275e-01, -4.88194e-01,  ..., -1.24110e+00,  4.29960e+00, -1.87582e+00],
       ...,
       [-3.64673e-01, -5.71122e-01, -4.19243e-01,  ..., -1.63448e+00,  3.99876e+00, -2.51915e+00],
       [-2.45810e-01, -4.65842e-02, -3.18997e-01,  ..., -1.64750e+00,  4.98972e+00, -3.22685e+00],
       [-1.18354e+00, -7.89089e-02, -6.57263e-01,  ..., -1.36611e+00,  3.48685e+00, -2.63512e+00]],

      [[ 2.12040e+00,  3.31226e-01, -6.44945e-01,  ..., -2.32001e+00, -4.55853e-01, -3.68612e+00],
       [ 3.67884e-01,  4.07311e-01, -2.97259e-01,  ..., -2.71750e+00,  1.31858e+00, -3.57216e+00],
       [ 4.10665e-01,  2.97937e-01, -3.82516e-01,  ..., -2.52612e+00,  1.97704e+00, -3.48992e+00],
       ...,
       [-2.64573e-01,  1.75337e-01, -3.17003e-01,  ..., -2.80365e+00,  3.36300e+00, -3.39216e+00],
       [ 1.17382e-01,  3.02737e-01, -2.83651e-01,  ..., -2.50814e+00,  2.99950e+00, -3.84078e+00],
       [-1.43937e+00,  9.55693e-02, -5.32340e-01,  ..., -2.39746e+00,  1.72113e+00, -3.58612e+00]],

      [[ 1.58736e+00, -7.60968e-01, -6.42212e-01,  ..., -8.75003e-01,  1.09893e+00, -2.69200e+00],
       [ 3.36146e-01, -1.80979e+00, -3.11036e-01,  ..., -2.29406e+00,  6.58327e-01, -3.80732e+00],
       [ 5.11697e-01, -1.86441e+00, -4.59026e-01,  ..., -1.41919e+00,  1.39803e+00, -3.21920e+00],
       ...,
       [ 1.07562e-01, -2.04132e+00, -3.70506e-01,  ..., -1.77860e+00,  2.13449e+00, -3.17919e+00],
       [ 4.66863e-01, -2.11350e+00, -3.46415e-01,  ..., -1.99330e+00,  2.00441e+00, -3.72452e+00],
       [-4.27420e-01, -1.19782e+00, -4.46980e-01,  ..., -1.03339e+00,  2.35215e+00, -2.77179e+00]]],

     [[[ 8.50652e-01, -9.30609e-02, -6.08074e-01,  ...,  1.95263e-01,  3.21971e-01, -1.94982e+00],
       [ 1.46718e-01,  5.03695e-01, -1.83246e-01,  ...,  1.59229e-01,  7.09638e-01, -2.17222e+00],
       [ 2.43548e-01,  3.51498e-01, -3.53942e-01,  ...,  9.60465e-01,  1.91433e+00, -1.81923e+00],
       ...,
       [-1.23076e-01,  4.94902e-01, -2.64858e-01,  ...,  1.45296e-01,  2.14129e+00, -2.39854e+00],
       [ 3.91850e-01,  5.29182e-01, -1.45324e-01,  ...,  5.27190e-02,  1.25552e+00, -2.24738e+00],
       [-5.51307e-01,  1.39537e-01, -4.81835e-01,  ..., -2.57817e-01,  6.91436e-01, -2.46776e+00]],

      [[ 1.55451e+00, -5.42548e-01, -6.41237e-01,  ..., -4.36566e-02,  1.53778e-01, -2.57851e+00],
       [ 3.21213e-02, -4.34971e-01, -2.87706e-01,  ..., -7.00950e-01,  1.88336e+00, -2.50216e+00],
       [ 7.36439e-02, -4.17065e-01, -3.44193e-01,  ..., -6.58743e-01,  3.61488e+00, -2.85515e+00],
       ...,
       [-3.28604e-01, -3.03848e-01, -2.90781e-01,  ..., -1.35442e+00,  4.69573e+00, -3.16105e+00],
       [ 1.34242e-01, -4.26568e-01, -2.95411e-01,  ..., -1.16126e+00,  2.56355e+00, -2.98944e+00],
       [-1.23323e+00, -5.14015e-01, -5.38291e-01,  ..., -9.66430e-01,  9.72652e-01, -3.16833e+00]],

      [[ 1.75746e+00, -3.01109e-01, -7.51035e-01,  ..., -3.64162e-01,  2.37350e-01, -2.78251e+00],
       [ 2.33151e-01, -2.39705e-01, -4.05763e-01,  ..., -1.18894e+00,  3.24991e+00, -2.67708e+00],
       [ 2.66195e-01, -2.30876e-01, -4.65671e-01,  ..., -9.37791e-01,  4.04630e+00, -2.86755e+00],
       ...,
       [-2.16339e-01, -1.33094e-01, -4.16025e-01,  ..., -1.63040e+00,  5.15063e+00, -3.50953e+00],
       [ 2.05508e-01, -3.45914e-01, -3.36002e-01,  ..., -1.39491e+00,  4.46838e+00, -3.36696e+00],
       [-1.16869e+00, -5.18955e-01, -6.38617e-01,  ..., -1.10517e+00,  2.30993e+00, -3.09845e+00]],

      ...,

      [[ 1.90883e+00, -2.51731e-01, -7.03335e-01,  ...,  9.89175e-03,  5.44337e-01, -2.53338e+00],
       [ 4.14448e-01, -5.22669e-01, -3.42766e-01,  ..., -1.02695e+00,  4.03619e+00, -2.82957e+00],
       [ 4.25188e-01, -6.21581e-01, -4.05114e-01,  ..., -9.31155e-01,  3.67293e+00, -2.25965e+00],
       ...,
       [-5.62532e-01, -7.78986e-01, -3.34942e-01,  ..., -1.12310e+00,  3.35633e+00, -2.89228e+00],
       [-3.68794e-01, -2.62228e-01, -2.41165e-01,  ..., -1.14537e+00,  4.15212e+00, -3.20383e+00],
       [-1.34852e+00, -2.28808e-01, -6.03447e-01,  ..., -9.83183e-01,  2.64425e+00, -3.12834e+00]],

      [[ 1.99314e+00,  6.19191e-02, -5.55858e-01,  ..., -1.36176e+00, -1.20211e+00, -3.40548e+00],
       [ 2.02662e-01,  2.30033e-01, -2.06585e-01,  ..., -1.63672e+00,  4.13913e-01, -3.42455e+00],
       [ 2.17388e-01,  5.07504e-03, -3.04658e-01,  ..., -1.28278e+00,  1.44055e+00, -3.22833e+00],
       ...,
       [-4.30703e-01, -1.04956e-02, -2.32259e-01,  ..., -1.56128e+00,  2.74039e+00, -3.26959e+00],
       [ 1.25432e-02,  1.06412e-01, -2.06399e-01,  ..., -1.41040e+00,  2.22542e+00, -3.34279e+00],
       [-1.54301e+00, -7.24666e-02, -4.68721e-01,  ..., -1.68307e+00,  1.02883e+00, -3.52435e+00]],

      [[ 1.39734e+00, -1.13126e+00, -5.63834e-01,  ...,  3.51796e-01, -9.43075e-02, -2.73288e+00],
       [ 2.31755e-01, -2.07655e+00, -2.25739e-01,  ..., -1.25420e+00, -2.94663e-01, -3.53927e+00],
       [ 3.80731e-01, -2.20935e+00, -4.02237e-01,  ..., -3.65492e-01,  7.10430e-01, -3.09795e+00],
       ...,
       [-2.39816e-02, -2.33372e+00, -3.02039e-01,  ..., -8.52325e-01,  1.37913e+00, -3.17775e+00],
       [ 3.96966e-01, -2.36047e+00, -2.75289e-01,  ..., -1.12014e+00,  1.18490e+00, -3.45052e+00],
       [-6.04296e-01, -1.50652e+00, -3.81873e-01,  ...,  2.46018e-02,  1.29240e+00, -2.72499e+00]]],

     [[[ 1.55555e+00, -1.11915e-01, -1.08571e+00,  ...,  3.09336e-03,  1.17408e+00, -2.02302e+00],
       [ 9.66305e-01,  5.31032e-01, -7.44643e-01,  ..., -3.20374e-01,  1.44717e+00, -2.40270e+00],
       [ 9.41903e-01,  3.03630e-01, -8.77821e-01,  ..., -2.79603e-01,  2.62305e+00, -2.44937e+00],
       ...,
       [ 5.42702e-01,  3.82635e-01, -8.07605e-01,  ..., -7.17945e-01,  2.72194e+00, -2.93159e+00],
       [ 1.03743e+00,  5.50902e-01, -7.18177e-01,  ..., -3.86481e-01,  1.75410e+00, -2.58988e+00],
       [ 2.15776e-01,  9.78998e-02, -9.88537e-01,  ..., -3.24604e-01,  1.50248e+00, -2.46980e+00]],

      [[ 2.28143e+00, -3.38080e-01, -1.13177e+00,  ..., -3.00541e-01,  1.12165e+00, -2.71121e+00],
       [ 6.82154e-01, -1.89542e-01, -7.81992e-01,  ..., -1.04336e+00,  2.58719e+00, -2.78855e+00],
       [ 5.73257e-01, -2.52963e-01, -8.76512e-01,  ..., -1.78358e+00,  4.24755e+00, -3.38407e+00],
       ...,
       [ 1.89349e-01, -2.11963e-01, -8.30279e-01,  ..., -2.23527e+00,  5.21887e+00, -3.68530e+00],
       [ 5.64930e-01, -2.42643e-01, -7.73705e-01,  ..., -1.42865e+00,  2.84132e+00, -3.35850e+00],
       [-6.16732e-01, -3.27478e-01, -9.74793e-01,  ..., -7.61636e-01,  1.65759e+00, -3.21198e+00]],

      [[ 2.39506e+00, -1.87320e-01, -1.18463e+00,  ..., -3.66115e-01,  1.17296e+00, -3.06830e+00],
       [ 7.87602e-01, -6.65558e-02, -8.62476e-01,  ..., -1.30794e+00,  3.89594e+00, -3.02794e+00],
       [ 7.34185e-01, -1.13199e-01, -9.82304e-01,  ..., -1.75238e+00,  4.68581e+00, -3.50552e+00],
       ...,
       [ 2.11047e-01, -1.13680e-01, -9.34347e-01,  ..., -2.22963e+00,  5.73514e+00, -4.06718e+00],
       [ 5.92264e-01, -1.77221e-01, -8.02406e-01,  ..., -1.46749e+00,  4.82816e+00, -3.86546e+00],
       [-5.89894e-01, -3.96168e-01, -1.05357e+00,  ..., -7.24036e-01,  3.04416e+00, -3.21771e+00]],

      ...,

      [[ 2.63585e+00, -1.05996e-01, -1.15732e+00,  ..., -1.21991e-01,  1.46600e+00, -2.88012e+00],
       [ 1.00903e+00, -3.09077e-01, -8.49151e-01,  ..., -1.18295e+00,  4.56479e+00, -3.14166e+00],
       [ 8.84926e-01, -4.70902e-01, -9.61101e-01,  ..., -1.58560e+00,  4.20666e+00, -2.78134e+00],
       ...,
       [-1.56609e-01, -7.19986e-01, -8.92517e-01,  ..., -1.65274e+00,  3.83103e+00, -3.43452e+00],
       [ 3.72595e-02, -6.58900e-02, -7.38653e-01,  ..., -1.28064e+00,  4.38066e+00, -3.73436e+00],
       [-7.14882e-01, -1.18907e-01, -1.03248e+00,  ..., -7.66584e-01,  3.18614e+00, -3.36909e+00]],

      [[ 2.62142e+00,  1.99106e-01, -1.03704e+00,  ..., -1.43266e+00, -2.91442e-01, -3.40783e+00],
       [ 9.04530e-01,  3.50617e-01, -7.21058e-01,  ..., -2.06518e+00,  1.08509e+00, -3.56615e+00],
       [ 7.38782e-01,  9.04017e-02, -8.34156e-01,  ..., -2.54181e+00,  2.19155e+00, -4.03486e+00],
       ...,
       [-4.18948e-03, -2.59851e-02, -7.74642e-01,  ..., -2.60993e+00,  3.36851e+00, -4.03584e+00],
       [ 4.79495e-01,  2.11216e-01, -6.93933e-01,  ..., -1.87568e+00,  2.70199e+00, -3.72337e+00],
       [-9.51958e-01,  1.59488e-02, -9.04933e-01,  ..., -1.43045e+00,  1.52769e+00, -3.32127e+00]],

      [[ 2.10382e+00, -9.15280e-01, -1.08362e+00,  ..., -1.27011e-01,  7.95739e-01, -3.27411e+00],
       [ 8.24662e-01, -1.94892e+00, -7.47737e-01,  ..., -1.50571e+00,  2.20263e-01, -3.58989e+00],
       [ 8.89856e-01, -2.08303e+00, -9.15678e-01,  ..., -1.38547e+00,  1.14855e+00, -3.77685e+00],
       ...,
       [ 4.08392e-01, -2.28112e+00, -8.35204e-01,  ..., -1.58708e+00,  1.79779e+00, -3.80804e+00],
       [ 8.02353e-01, -2.24068e+00, -7.89665e-01,  ..., -1.28347e+00,  1.63641e+00, -3.60500e+00],
       [ 8.16434e-02, -1.32416e+00, -9.07675e-01,  ..., -1.96619e-01,  1.90694e+00, -3.10842e+00]]]]]), tensor([[[[[ 6.89453e-01,  8.19994e-01, -5.47395e-01,  ..., -2.81120e+00,  1.31253e+00, -4.13477e+00],
       [-9.02833e-02,  1.73158e+00, -4.20647e-01,  ..., -3.26589e+00,  3.17073e+00, -4.43095e+00],
       [-3.50521e-01,  1.73973e+00, -1.17367e-01,  ..., -2.45306e+00,  3.59818e+00, -4.32461e+00],
       ...,
       [-1.11871e-01,  1.65261e+00, -1.05441e-01,  ..., -2.47966e+00,  4.04109e+00, -4.38976e+00],
       [ 1.51748e-01,  1.60373e+00, -4.06148e-01,  ..., -2.98852e+00,  3.74366e+00, -4.49184e+00],
       [-6.88955e-01,  9.92839e-01, -5.14094e-01,  ..., -2.39312e+00,  2.07144e+00, -4.12880e+00]],

      [[ 1.38097e+00, -5.35230e-02, -5.96301e-01,  ..., -3.55773e+00,  2.54009e+00, -4.41275e+00],
       [-9.91016e-02,  1.51239e-01, -4.48749e-01,  ..., -3.36622e+00,  4.43207e+00, -3.25542e+00],
       [-2.46814e-01,  1.54045e-01, -8.18962e-02,  ..., -2.97794e+00,  3.78779e+00, -3.72133e+00],
       ...,
       [-5.87575e-02,  2.13048e-01, -1.07047e-01,  ..., -2.99477e+00,  5.48545e+00, -3.93879e+00],
       [ 1.35103e-01,  1.31488e-01, -4.39951e-01,  ..., -3.06798e+00,  5.59508e+00, -4.38943e+00],
       [-1.32230e+00, -1.04805e-01, -6.06642e-01,  ..., -2.83907e+00,  3.80004e+00, -4.03890e+00]],

      [[ 1.52054e+00,  2.61626e-02, -5.85517e-01,  ..., -2.68152e+00,  1.73602e+00, -4.74730e+00],
       [ 2.06851e-02,  2.46295e-01, -4.24561e-01,  ..., -2.53839e+00,  2.30028e+00, -3.40504e+00],
       [ 1.08701e-01,  1.56986e-01,  1.42243e-02,  ..., -2.57858e+00,  8.66213e-01, -3.51936e+00],
       ...,
       [-1.19504e-01,  2.69389e-01, -9.26682e-03,  ..., -2.79625e+00,  2.39180e+00, -3.73654e+00],
       [ 1.00909e-01,  8.73255e-02, -4.26041e-01,  ..., -2.60887e+00,  3.21019e+00, -4.05745e+00],
       [-1.31338e+00,  4.78064e-02, -6.28710e-01,  ..., -2.09334e+00,  2.96197e+00, -4.19277e+00]],

      ...,

      [[ 1.56709e+00,  1.75426e-01, -5.69327e-01,  ..., -2.91149e+00,  2.64466e+00, -4.05187e+00],
       [ 6.81030e-02,  3.54531e-01, -4.20971e-01,  ..., -3.01514e+00,  4.70841e+00, -2.36303e+00],
       [ 1.20090e-01,  4.80990e-02,  3.37213e-02,  ..., -2.85958e+00,  2.96200e+00, -2.06378e+00],
       ...,
       [-1.12342e-01,  1.00438e-01, -4.19420e-02,  ..., -2.87443e+00,  2.33099e+00, -2.47859e+00],
       [ 7.45863e-02,  2.83304e-01, -4.16078e-01,  ..., -2.79411e+00,  3.82716e+00, -3.33733e+00],
       [-1.48393e+00,  3.96536e-01, -5.79180e-01,  ..., -2.48558e+00,  3.34801e+00, -4.17121e+00]],

      [[ 1.75763e+00,  1.08631e-01, -5.16917e-01,  ..., -2.99500e+00,  2.93685e+00, -4.54236e+00],
       [-1.16291e-02, -9.15394e-02, -4.26565e-01,  ..., -2.81112e+00,  5.30727e+00, -2.97665e+00],
       [-4.20046e-02, -4.19987e-01, -1.79598e-03,  ..., -2.41489e+00,  4.40215e+00, -2.83205e+00],
       ...,
       [ 1.14690e-01, -3.31108e-01, -6.28824e-02,  ..., -2.36704e+00,  3.81778e+00, -3.21699e+00],
       [ 2.99319e-01, -2.03581e-01, -4.41284e-01,  ..., -2.36653e+00,  4.43965e+00, -3.90094e+00],
       [-1.45214e+00,  6.66651e-02, -5.43555e-01,  ..., -2.59475e+00,  3.13480e+00, -4.17770e+00]],

      [[ 1.08398e+00, -6.62469e-01, -5.18035e-01,  ..., -2.55819e+00,  1.95962e+00, -5.02971e+00],
       [-1.73250e-01, -1.41797e+00, -4.59350e-01,  ..., -2.92395e+00,  3.78546e+00, -4.76472e+00],
       [-2.06157e-01, -1.64838e+00, -5.17136e-02,  ..., -2.33015e+00,  3.59640e+00, -4.29707e+00],
       ...,
       [ 1.99323e-01, -1.48360e+00, -8.39590e-02,  ..., -2.22402e+00,  3.92343e+00, -4.32756e+00],
       [ 2.84445e-01, -1.56401e+00, -4.43266e-01,  ..., -2.79308e+00,  3.30923e+00, -4.72992e+00],
       [-7.75658e-01, -8.76278e-01, -5.50824e-01,  ..., -2.58314e+00,  2.03195e+00, -4.81671e+00]]],

     [[[ 7.15436e-01,  8.15682e-01, -1.06564e+00,  ..., -3.06415e+00,  1.41567e+00, -3.73831e+00],
       [-1.10488e-01,  1.69874e+00, -1.00576e+00,  ..., -3.52134e+00,  2.66696e+00, -4.26693e+00],
       [-3.72953e-01,  1.69376e+00, -7.95376e-01,  ..., -2.78257e+00,  3.38762e+00, -4.12275e+00],
       ...,
       [-1.17359e-01,  1.61259e+00, -7.99039e-01,  ..., -2.83198e+00,  3.72293e+00, -4.15757e+00],
       [ 9.20912e-02,  1.59250e+00, -9.85302e-01,  ..., -3.18493e+00,  3.52055e+00, -4.19832e+00],
       [-7.87422e-01,  9.96894e-01, -1.01599e+00,  ..., -2.50232e+00,  2.19810e+00, -4.01642e+00]],

      [[ 1.45414e+00, -9.45391e-02, -1.06425e+00,  ..., -3.87742e+00,  2.73128e+00, -4.01763e+00],
       [-4.92745e-02,  9.45988e-02, -9.74300e-01,  ..., -3.75014e+00,  3.98081e+00, -3.39420e+00],
       [-2.14412e-01,  1.15824e-01, -7.16154e-01,  ..., -3.34917e+00,  3.65975e+00, -3.77720e+00],
       ...,
       [ 6.26777e-03,  1.86461e-01, -7.58636e-01,  ..., -3.32463e+00,  5.24866e+00, -3.91904e+00],
       [ 6.12421e-02,  1.31659e-01, -9.62199e-01,  ..., -3.28197e+00,  5.48170e+00, -4.30868e+00],
       [-1.36672e+00, -8.24564e-02, -1.04956e+00,  ..., -2.98627e+00,  3.91709e+00, -4.11292e+00]],

      [[ 1.59118e+00, -1.19618e-02, -1.05398e+00,  ..., -3.13491e+00,  1.89390e+00, -4.41634e+00],
       [ 4.10931e-02,  1.94947e-01, -9.56506e-01,  ..., -2.91796e+00,  1.88371e+00, -3.60911e+00],
       [ 1.19924e-01,  1.15981e-01, -6.39971e-01,  ..., -2.98840e+00,  7.31990e-01, -3.70605e+00],
       ...,
       [-6.67303e-02,  2.66637e-01, -6.79519e-01,  ..., -3.13224e+00,  2.14776e+00, -3.87013e+00],
       [ 3.66589e-02,  8.38747e-02, -9.56729e-01,  ..., -2.85526e+00,  3.13828e+00, -4.02898e+00],
       [-1.38386e+00,  6.97111e-02, -1.05186e+00,  ..., -2.31606e+00,  3.12314e+00, -4.27515e+00]],

      ...,

      [[ 1.61468e+00,  1.25336e-01, -1.04051e+00,  ..., -3.40704e+00,  2.70673e+00, -3.74628e+00],
       [ 8.87758e-02,  3.10939e-01, -9.51191e-01,  ..., -3.59570e+00,  4.20525e+00, -2.59008e+00],
       [ 1.21105e-01,  1.08763e-02, -6.23037e-01,  ..., -3.44363e+00,  2.76088e+00, -2.25335e+00],
       ...,
       [-7.68137e-02,  1.11653e-01, -7.03732e-01,  ..., -3.40072e+00,  2.01464e+00, -2.55085e+00],
       [ 9.59173e-04,  2.82754e-01, -9.45777e-01,  ..., -3.24536e+00,  3.59745e+00, -3.35075e+00],
       [-1.54613e+00,  4.13537e-01, -1.01641e+00,  ..., -2.80218e+00,  3.37504e+00, -4.27166e+00]],

      [[ 1.76468e+00,  4.67952e-02, -1.01341e+00,  ..., -3.54050e+00,  3.15463e+00, -4.20530e+00],
       [-2.39984e-02, -1.65638e-01, -9.62698e-01,  ..., -3.41986e+00,  4.91370e+00, -3.17044e+00],
       [-4.89825e-02, -4.89403e-01, -6.63066e-01,  ..., -2.98488e+00,  4.40396e+00, -2.98814e+00],
       ...,
       [ 1.48487e-01, -3.58134e-01, -7.24502e-01,  ..., -2.87908e+00,  3.70764e+00, -3.28059e+00],
       [ 1.86871e-01, -2.06472e-01, -9.66651e-01,  ..., -2.84974e+00,  4.46289e+00, -3.79174e+00],
       [-1.52183e+00,  8.62337e-02, -1.02043e+00,  ..., -2.85710e+00,  3.23966e+00, -4.27743e+00]],

      [[ 1.15066e+00, -6.51105e-01, -1.04855e+00,  ..., -2.71137e+00,  2.14691e+00, -4.74436e+00],
       [-1.56587e-01, -1.43924e+00, -1.00862e+00,  ..., -3.01026e+00,  3.41163e+00, -4.69165e+00],
       [-1.66052e-01, -1.68058e+00, -7.35079e-01,  ..., -2.43147e+00,  3.57443e+00, -4.25417e+00],
       ...,
       [ 2.57203e-01, -1.50827e+00, -7.70623e-01,  ..., -2.38340e+00,  3.72987e+00, -4.24101e+00],
       [ 2.96140e-01, -1.54395e+00, -1.00853e+00,  ..., -2.88979e+00,  3.23558e+00, -4.55618e+00],
       [-8.19594e-01, -8.45250e-01, -1.03661e+00,  ..., -2.53689e+00,  2.17142e+00, -4.72064e+00]]],

     [[[ 4.12463e-01,  1.15113e-01, -5.68993e-01,  ..., -2.06719e+00, -4.78436e-01, -2.04569e+00],
       [ 7.41843e-02,  6.01761e-01, -4.27095e-01,  ..., -2.01272e+00,  2.85216e-01, -2.41251e+00],
       [-1.54174e-01,  6.20097e-01, -2.62571e-01,  ..., -1.83312e+00,  5.50643e-01, -2.37774e+00],
       ...,
       [-2.98118e-02,  5.54996e-01, -2.36485e-01,  ..., -1.90047e+00,  6.43932e-01, -2.35635e+00],
       [-5.99789e-02,  6.60994e-01, -4.53854e-01,  ..., -2.06117e+00,  8.20722e-01, -2.40235e+00],
       [-4.23174e-01,  2.34648e-01, -4.48372e-01,  ..., -1.79558e+00, -1.66097e-01, -2.12730e+00]],

      [[ 9.29643e-01,  3.64374e-01, -6.76913e-01,  ..., -2.88955e+00,  3.82818e-01, -2.41070e+00],
       [ 1.37871e-01,  7.61467e-01, -4.98203e-01,  ..., -2.92192e+00,  1.16933e+00, -2.00414e+00],
       [-3.80658e-02,  7.24536e-01, -2.80671e-01,  ..., -2.78975e+00,  7.62987e-01, -2.19583e+00],
       ...,
       [-7.14942e-02,  7.37439e-01, -2.71873e-01,  ..., -2.82050e+00,  1.61067e+00, -2.28392e+00],
       [-1.72246e-01,  9.14436e-01, -5.62681e-01,  ..., -2.73458e+00,  1.83794e+00, -2.31648e+00],
       [-9.90997e-01,  3.89740e-01, -5.31008e-01,  ..., -2.41096e+00,  9.26999e-01, -2.16774e+00]],

      [[ 1.17913e+00, -2.38138e-02, -7.48866e-01,  ..., -3.07782e+00, -6.74025e-02, -2.48266e+00],
       [ 1.44954e-01,  3.64821e-01, -5.65456e-01,  ..., -3.08631e+00,  7.64323e-02, -1.93444e+00],
       [ 2.05890e-01,  4.24684e-01, -2.91673e-01,  ..., -3.09029e+00, -8.99682e-01, -2.13254e+00],
       ...,
       [-1.20473e-01,  4.42786e-01, -3.12334e-01,  ..., -3.26534e+00,  7.44760e-02, -2.21113e+00],
       [-1.90720e-01,  3.19751e-01, -6.38874e-01,  ..., -3.10174e+00,  6.87708e-01, -2.09516e+00],
       [-1.17269e+00,  8.83184e-02, -6.16905e-01,  ..., -2.41893e+00,  3.62949e-01, -2.22741e+00]],

      ...,

      [[ 1.29079e+00, -7.23099e-02, -7.35939e-01,  ..., -2.63339e+00,  2.23173e-01, -2.22273e+00],
       [ 1.83048e-01,  1.37653e-01, -5.70201e-01,  ..., -2.70286e+00,  1.06945e+00, -1.46700e+00],
       [ 2.45448e-01, -4.39390e-02, -2.81482e-01,  ..., -2.80440e+00,  6.00521e-02, -1.30636e+00],
       ...,
       [-1.53797e-01, -1.13922e-01, -3.24183e-01,  ..., -2.73806e+00, -2.05790e-01, -1.56518e+00],
       [-2.25687e-01,  2.11394e-01, -6.46898e-01,  ..., -2.63373e+00,  5.67977e-01, -1.69431e+00],
       [-1.40978e+00,  2.46955e-01, -5.95548e-01,  ..., -2.38240e+00,  4.66745e-01, -2.26023e+00]],

      [[ 1.19810e+00, -4.53988e-01, -6.80504e-01,  ..., -2.76570e+00,  5.78450e-01, -2.50427e+00],
       [ 2.20658e-01, -7.24626e-01, -5.60108e-01,  ..., -2.64023e+00,  1.37518e+00, -1.96604e+00],
       [ 1.85890e-01, -9.04898e-01, -3.06435e-01,  ..., -2.54630e+00,  8.95478e-01, -1.88214e+00],
       ...,
       [ 6.63931e-02, -9.12040e-01, -3.18863e-01,  ..., -2.52687e+00,  6.18810e-01, -2.01991e+00],
       [-2.74894e-02, -7.48897e-01, -6.19398e-01,  ..., -2.51468e+00,  1.06563e+00, -2.03342e+00],
       [-1.15030e+00, -3.58563e-01, -5.44651e-01,  ..., -2.35801e+00,  6.02540e-01, -2.27689e+00]],

      [[ 7.05402e-01, -3.01978e-01, -5.28577e-01,  ..., -2.20664e+00,  2.62612e-02, -2.41362e+00],
       [ 4.46921e-02, -6.58611e-01, -3.72844e-01,  ..., -2.22709e+00,  7.48776e-01, -2.45888e+00],
       [ 7.65167e-02, -8.27883e-01, -1.59959e-01,  ..., -2.09515e+00,  6.91639e-01, -2.41154e+00],
       ...,
       [ 3.06763e-01, -7.86351e-01, -1.60499e-01,  ..., -2.05941e+00,  7.44825e-01, -2.39758e+00],
       [ 1.28848e-01, -6.43959e-01, -4.57112e-01,  ..., -2.28288e+00,  7.27487e-01, -2.46918e+00],
       [-4.46692e-01, -3.50273e-01, -4.13260e-01,  ..., -2.03499e+00,  7.94328e-02, -2.30357e+00]]]]]), tensor([[[[[ 9.42430e-01,  9.63071e-01, -8.97984e-01,  ..., -2.02751e+00, -4.98131e-01, -1.47186e+00],
       [ 5.57714e-02,  1.25856e+00, -7.27494e-01,  ..., -2.44291e+00,  1.81403e-01, -1.81278e+00],
       [-2.32369e-01,  1.11604e+00, -3.74938e-01,  ..., -1.96185e+00, -1.65811e-01, -1.40312e+00],
       ...,
       [ 3.11451e-01,  1.12483e+00, -3.66681e-01,  ..., -2.32544e+00,  6.06323e-01, -1.66383e+00],
       [-2.74272e-02,  1.28297e+00, -7.13617e-01,  ..., -2.68100e+00,  6.99806e-01, -2.05444e+00],
       [-1.03596e+00,  9.17992e-01, -9.18891e-01,  ..., -2.12563e+00,  3.95310e-02, -1.80489e+00]],

      [[ 1.36209e+00,  8.82454e-02, -9.10384e-01,  ..., -2.41722e+00, -6.15331e-01, -2.02004e+00],
       [ 1.30603e-01,  1.67846e-01, -7.24540e-01,  ..., -2.76219e+00, -1.07358e+00, -2.21890e+00],
       [-1.76619e-01,  2.24706e-01, -3.18522e-01,  ..., -2.21735e+00, -1.19233e+00, -1.86104e+00],
       ...,
       [ 1.45481e-01,  2.90356e-01, -3.05915e-01,  ..., -2.61043e+00, -4.25905e-02, -2.04483e+00],
       [-2.10735e-01,  2.34735e-01, -6.80680e-01,  ..., -3.01527e+00, -3.34737e-01, -2.38576e+00],
       [-1.50473e+00,  1.21357e-01, -9.04706e-01,  ..., -2.62704e+00, -3.45500e-01, -2.32824e+00]],

      [[ 1.18541e+00, -2.25211e-01, -9.05398e-01,  ..., -2.04732e+00, -1.07960e+00, -1.74247e+00],
       [ 2.04415e-01, -2.12251e-01, -7.06042e-01,  ..., -2.39830e+00, -1.81614e+00, -1.96638e+00],
       [ 2.05903e-02, -4.18067e-02, -2.70389e-01,  ..., -1.69382e+00, -1.98309e+00, -1.74739e+00],
       ...,
       [-1.29090e-01,  2.03900e-02, -2.58953e-01,  ..., -2.01698e+00, -5.41621e-01, -1.69008e+00],
       [-3.16899e-01, -2.29031e-01, -6.61441e-01,  ..., -2.54465e+00, -1.12906e+00, -2.02318e+00],
       [-1.38884e+00, -2.42128e-01, -9.15534e-01,  ..., -2.51323e+00, -9.86188e-01, -2.15521e+00]],

      ...,

      [[ 1.20630e+00,  2.72283e-01, -9.02342e-01,  ..., -2.22177e+00, -2.48450e-01, -1.59151e+00],
       [ 3.14838e-01, -1.84982e-01, -6.82767e-01,  ..., -2.02823e+00, -4.98356e-01, -1.26158e+00],
       [ 2.10594e-01, -3.01237e-01, -2.44446e-01,  ..., -1.19439e+00, -7.24995e-01, -8.33837e-01],
       ...,
       [-9.03732e-02, -1.81963e-01, -2.67140e-01,  ..., -1.62290e+00, -1.78406e-01, -1.27666e+00],
       [-3.48984e-01,  1.52712e-02, -6.56935e-01,  ..., -2.03326e+00, -8.26056e-01, -1.54756e+00],
       [-1.28668e+00,  3.02744e-01, -9.24881e-01,  ..., -2.30221e+00, -9.03314e-01, -1.81645e+00]],

      [[ 1.34427e+00,  1.76064e-01, -9.09456e-01,  ..., -2.52560e+00, -3.76163e-02, -1.76576e+00],
       [ 1.38035e-01, -8.65171e-02, -7.31915e-01,  ..., -2.55777e+00, -1.16280e-01, -1.42910e+00],
       [-1.36167e-01, -9.10429e-02, -3.15400e-01,  ..., -1.63596e+00, -4.48259e-01, -8.27267e-01],
       ...,
       [ 2.36701e-01,  1.16846e-02, -3.43185e-01,  ..., -2.12809e+00, -7.88776e-02, -1.51909e+00],
       [-9.89413e-02,  2.20459e-02, -7.18287e-01,  ..., -2.62676e+00, -5.64226e-01, -1.70836e+00],
       [-1.49135e+00,  1.30770e-01, -9.12434e-01,  ..., -2.72786e+00, -7.29371e-01, -2.18945e+00]],

      [[ 1.00402e+00, -8.44526e-01, -8.94482e-01,  ..., -2.64645e+00, -1.62521e-01, -1.84122e+00],
       [ 4.09335e-02, -1.31596e+00, -7.35312e-01,  ..., -3.05447e+00,  6.56089e-01, -1.98048e+00],
       [-4.20862e-01, -1.13238e+00, -4.04635e-01,  ..., -2.57353e+00,  2.19588e-01, -1.39129e+00],
       ...,
       [ 3.04473e-01, -1.03076e+00, -3.79914e-01,  ..., -2.59787e+00,  5.55781e-01, -1.56636e+00],
       [-6.04132e-02, -1.21320e+00, -7.23111e-01,  ..., -3.04008e+00,  3.81797e-01, -2.04550e+00],
       [-1.25319e+00, -9.14017e-01, -9.11772e-01,  ..., -2.87136e+00, -3.90458e-01, -2.12152e+00]]],

     [[[ 9.56946e-01,  9.58358e-01, -1.21939e+00,  ..., -1.62897e+00, -2.37883e-01, -1.34706e+00],
       [ 7.48386e-02,  1.27206e+00, -1.06284e+00,  ..., -2.08623e+00,  3.57879e-01, -1.61664e+00],
       [-2.11208e-01,  1.13688e+00, -7.67556e-01,  ..., -1.67870e+00,  1.97221e-02, -1.31759e+00],
       ...,
       [ 2.78476e-01,  1.12260e+00, -7.69433e-01,  ..., -2.01889e+00,  6.74058e-01, -1.50713e+00],
       [-4.74577e-02,  1.26302e+00, -1.05297e+00,  ..., -2.29831e+00,  7.83114e-01, -1.83475e+00],
       [-1.05189e+00,  9.15727e-01, -1.23269e+00,  ..., -1.64032e+00,  3.26987e-01, -1.68836e+00]],

      [[ 1.39311e+00,  5.50419e-02, -1.22904e+00,  ..., -1.89388e+00, -3.37426e-01, -1.73198e+00],
       [ 1.77634e-01,  1.55570e-01, -1.07157e+00,  ..., -2.36501e+00, -8.25964e-01, -1.92951e+00],
       [-1.25840e-01,  2.17833e-01, -7.26409e-01,  ..., -1.91660e+00, -1.03640e+00, -1.63312e+00],
       ...,
       [ 1.51862e-01,  2.43792e-01, -7.06467e-01,  ..., -2.35904e+00,  2.49594e-03, -1.70573e+00],
       [-2.42125e-01,  2.03230e-01, -1.02626e+00,  ..., -2.62684e+00, -1.91177e-01, -2.05156e+00],
       [-1.53285e+00,  8.88940e-02, -1.22487e+00,  ..., -2.01064e+00, -4.85771e-02, -2.01598e+00]],

      [[ 1.21205e+00, -2.60789e-01, -1.22618e+00,  ..., -1.56879e+00, -8.21974e-01, -1.51232e+00],
       [ 2.59519e-01, -2.15679e-01, -1.04953e+00,  ..., -2.01658e+00, -1.49387e+00, -1.66184e+00],
       [ 1.11646e-01, -3.76160e-02, -6.86054e-01,  ..., -1.40585e+00, -1.76348e+00, -1.51684e+00],
       ...,
       [-1.18679e-01, -2.61027e-02, -6.61362e-01,  ..., -1.82180e+00, -4.32834e-01, -1.35034e+00],
       [-3.25849e-01, -2.66500e-01, -1.00189e+00,  ..., -2.24083e+00, -9.12287e-01, -1.68527e+00],
       [-1.41338e+00, -2.75183e-01, -1.23583e+00,  ..., -1.91041e+00, -7.30031e-01, -1.85354e+00]],

      ...,

      [[ 1.19289e+00,  2.62973e-01, -1.21974e+00,  ..., -1.71495e+00, -6.27619e-03, -1.36299e+00],
       [ 3.48887e-01, -1.77478e-01, -1.01835e+00,  ..., -1.61764e+00, -2.81667e-01, -1.04745e+00],
       [ 3.18929e-01, -2.94249e-01, -6.49698e-01,  ..., -8.80970e-01, -5.66914e-01, -7.32610e-01],
       ...,
       [-1.00906e-01, -2.22221e-01, -6.64617e-01,  ..., -1.32223e+00, -1.10273e-01, -1.03332e+00],
       [-3.59595e-01, -1.52743e-02, -9.92114e-01,  ..., -1.61986e+00, -6.43509e-01, -1.33268e+00],
       [-1.32170e+00,  2.91809e-01, -1.24229e+00,  ..., -1.69771e+00, -6.00016e-01, -1.56806e+00]],

      [[ 1.33736e+00,  1.46063e-01, -1.22496e+00,  ..., -2.02578e+00,  8.49723e-02, -1.58169e+00],
       [ 1.58672e-01, -9.90473e-02, -1.06885e+00,  ..., -2.17800e+00, -6.14706e-02, -1.31761e+00],
       [-5.96869e-02, -9.82533e-02, -7.11666e-01,  ..., -1.34507e+00, -4.14104e-01, -8.17019e-01],
       ...,
       [ 2.29975e-01, -3.59555e-02, -7.30578e-01,  ..., -1.85685e+00, -1.34863e-01, -1.38268e+00],
       [-1.37310e-01, -1.88740e-02, -1.04994e+00,  ..., -2.24462e+00, -5.43083e-01, -1.60911e+00],
       [-1.54594e+00,  1.02519e-01, -1.23223e+00,  ..., -2.08063e+00, -5.35204e-01, -1.95420e+00]],

      [[ 9.93696e-01, -8.67172e-01, -1.22357e+00,  ..., -2.13176e+00, -7.03782e-02, -1.72153e+00],
       [ 2.11518e-02, -1.31173e+00, -1.07884e+00,  ..., -2.63884e+00,  5.89668e-01, -1.83545e+00],
       [-4.40688e-01, -1.10464e+00, -7.99756e-01,  ..., -2.25313e+00,  2.67985e-01, -1.36998e+00],
       ...,
       [ 2.43824e-01, -1.02811e+00, -7.81124e-01,  ..., -2.30306e+00,  5.37032e-01, -1.50271e+00],
       [-1.24772e-01, -1.22995e+00, -1.06831e+00,  ..., -2.63288e+00,  3.18468e-01, -1.94365e+00],
       [-1.29877e+00, -9.49796e-01, -1.23546e+00,  ..., -2.25683e+00, -3.13450e-01, -1.94474e+00]]],

     [[[ 9.13085e-01,  8.02369e-01, -1.33858e+00,  ..., -1.48166e+00, -2.52465e-01, -1.77567e+00],
       [ 1.19479e-01,  1.15449e+00, -1.20707e+00,  ..., -1.86274e+00,  3.64701e-01, -1.85339e+00],
       [-1.28456e-01,  1.02041e+00, -9.39507e-01,  ..., -1.59720e+00,  1.71411e-03, -1.56030e+00],
       ...,
       [ 2.33778e-01,  9.85361e-01, -9.34782e-01,  ..., -1.85773e+00,  5.39307e-01, -1.75255e+00],
       [-1.02172e-01,  1.12721e+00, -1.18852e+00,  ..., -2.01567e+00,  7.31285e-01, -2.08549e+00],
       [-1.02587e+00,  8.38115e-01, -1.35563e+00,  ..., -1.53519e+00,  1.66135e-01, -2.03650e+00]],

      [[ 1.34915e+00,  2.01809e-02, -1.34381e+00,  ..., -1.71274e+00, -2.40860e-01, -2.07565e+00],
       [ 2.15399e-01,  1.75886e-01, -1.21595e+00,  ..., -2.12355e+00, -5.73824e-01, -2.03019e+00],
       [-8.94462e-02,  2.52814e-01, -8.99362e-01,  ..., -1.85715e+00, -7.88601e-01, -1.72542e+00],
       ...,
       [ 4.83254e-02,  2.55908e-01, -8.80975e-01,  ..., -2.17645e+00,  7.79206e-02, -1.85132e+00],
       [-3.30010e-01,  2.06944e-01, -1.16569e+00,  ..., -2.33987e+00, -3.33729e-02, -2.19512e+00],
       [-1.53261e+00,  8.46040e-02, -1.33875e+00,  ..., -1.86365e+00, -5.20070e-02, -2.26135e+00]],

      [[ 1.19275e+00, -2.58370e-01, -1.33507e+00,  ..., -1.49703e+00, -7.55682e-01, -1.95498e+00],
       [ 2.98500e-01, -2.16011e-01, -1.19031e+00,  ..., -1.90921e+00, -1.18223e+00, -1.82853e+00],
       [ 1.67658e-01, -5.67821e-03, -8.53540e-01,  ..., -1.47550e+00, -1.45774e+00, -1.65915e+00],
       ...,
       [-2.01114e-01, -1.45985e-02, -8.32220e-01,  ..., -1.77265e+00, -3.59374e-01, -1.62462e+00],
       [-3.86440e-01, -2.84687e-01, -1.13902e+00,  ..., -2.07495e+00, -6.95949e-01, -1.97121e+00],
       [-1.35509e+00, -2.53751e-01, -1.34467e+00,  ..., -1.84042e+00, -7.08735e-01, -2.23002e+00]],

      ...,

      [[ 1.15486e+00,  9.55505e-02, -1.33183e+00,  ..., -1.63754e+00, -1.37068e-01, -1.89226e+00],
       [ 3.39635e-01, -2.51457e-01, -1.16453e+00,  ..., -1.58408e+00, -2.61938e-01, -1.43992e+00],
       [ 3.09579e-01, -3.19675e-01, -8.28984e-01,  ..., -1.03358e+00, -5.40264e-01, -1.17100e+00],
       ...,
       [-2.13917e-01, -2.98768e-01, -8.35706e-01,  ..., -1.36832e+00, -1.47103e-01, -1.53449e+00],
       [-4.35669e-01, -1.32319e-01, -1.13036e+00,  ..., -1.58153e+00, -5.26270e-01, -1.79008e+00],
       [-1.27568e+00,  1.48716e-01, -1.35242e+00,  ..., -1.70660e+00, -6.71950e-01, -2.07572e+00]],

      [[ 1.29225e+00,  1.24919e-02, -1.34257e+00,  ..., -1.77297e+00,  6.94638e-02, -1.99856e+00],
       [ 1.88288e-01, -1.98867e-01, -1.21435e+00,  ..., -1.90350e+00,  4.16680e-02, -1.58773e+00],
       [-2.58382e-02, -1.18955e-01, -8.95279e-01,  ..., -1.32202e+00, -3.21198e-01, -1.15152e+00],
       ...,
       [ 1.24763e-01, -9.82707e-02, -9.05656e-01,  ..., -1.65856e+00, -5.16666e-02, -1.69712e+00],
       [-2.21111e-01, -1.25798e-01, -1.19056e+00,  ..., -1.93421e+00, -3.29139e-01, -1.91908e+00],
       [-1.50142e+00, -2.30715e-04, -1.34965e+00,  ..., -1.88342e+00, -4.75151e-01, -2.26907e+00]],

      [[ 9.89151e-01, -8.88881e-01, -1.34686e+00,  ..., -1.84643e+00, -1.55948e-01, -2.13232e+00],
       [ 6.31064e-02, -1.32326e+00, -1.22502e+00,  ..., -2.25150e+00,  5.46254e-01, -2.05183e+00],
       [-3.48674e-01, -1.09971e+00, -9.78129e-01,  ..., -1.98615e+00,  1.42483e-01, -1.64473e+00],
       ...,
       [ 1.97233e-01, -1.03248e+00, -9.55822e-01,  ..., -2.00359e+00,  3.59097e-01, -1.81683e+00],
       [-1.61952e-01, -1.25554e+00, -1.21019e+00,  ..., -2.20722e+00,  3.06103e-01, -2.21070e+00],
       [-1.22079e+00, -9.32641e-01, -1.36126e+00,  ..., -2.00728e+00, -3.47121e-01, -2.27745e+00]]]]])) (68.5 MB)

TensorFlow version 2.9.2 has not been tested with coremltools. You may run into unexpected errors. TensorFlow 2.8.0 is the most recent version that has been tested.

CoreML: starting export with coremltools 6.0... Tuple detected at graph output. This will be flattened in the converted model. Converting graph. Adding op 'model.0.conv.bias' of type const Adding op 'model.0.conv.weight' of type const Adding op 'model.1.conv.bias' of type const Adding op 'model.1.conv.weight' of type const Adding op 'model.2.cv1.conv.bias' of type const Adding op 'model.2.cv1.conv.weight' of type const Adding op 'model.2.m.0.cv1.conv.bias' of type const Adding op 'model.2.m.0.cv1.conv.weight' of type const Adding op 'model.2.m.0.cv2.conv.bias' of type const Adding op 'model.2.m.0.cv2.conv.weight' of type const Adding op 'model.2.m.1.cv1.conv.bias' of type const Adding op 'model.2.m.1.cv1.conv.weight' of type const Adding op 'model.2.m.1.cv2.conv.bias' of type const Adding op 'model.2.m.1.cv2.conv.weight' of type const Adding op 'model.2.cv2.conv.bias' of type const Adding op 'model.2.cv2.conv.weight' of type const Adding op 'model.2.cv3.conv.bias' of type const Adding op 'model.2.cv3.conv.weight' of type const Adding op 'model.3.conv.bias' of type const Adding op 'model.3.conv.weight' of type const Adding op 'model.4.cv1.conv.bias' of type const Adding op 'model.4.cv1.conv.weight' of type const Adding op 'model.4.m.0.cv1.conv.bias' of type const Adding op 'model.4.m.0.cv1.conv.weight' of type const Adding op 'model.4.m.0.cv2.conv.bias' of type const Adding op 'model.4.m.0.cv2.conv.weight' of type const Adding op 'model.4.m.1.cv1.conv.bias' of type const Adding op 'model.4.m.1.cv1.conv.weight' of type const Adding op 'model.4.m.1.cv2.conv.bias' of type const Adding op 'model.4.m.1.cv2.conv.weight' of type const Adding op 'model.4.m.2.cv1.conv.bias' of type const Adding op 'model.4.m.2.cv1.conv.weight' of type const Adding op 'model.4.m.2.cv2.conv.bias' of type const Adding op 'model.4.m.2.cv2.conv.weight' of type const Adding op 'model.4.m.3.cv1.conv.bias' of type const Adding op 'model.4.m.3.cv1.conv.weight' of type const Adding op 'model.4.m.3.cv2.conv.bias' of type const Adding op 'model.4.m.3.cv2.conv.weight' of type const Adding op 'model.4.cv2.conv.bias' of type const Adding op 'model.4.cv2.conv.weight' of type const Adding op 'model.4.cv3.conv.bias' of type const Adding op 'model.4.cv3.conv.weight' of type const Adding op 'model.5.conv.bias' of type const Adding op 'model.5.conv.weight' of type const Adding op 'model.6.cv1.conv.bias' of type const Adding op 'model.6.cv1.conv.weight' of type const Adding op 'model.6.m.0.cv1.conv.bias' of type const Adding op 'model.6.m.0.cv1.conv.weight' of type const Adding op 'model.6.m.0.cv2.conv.bias' of type const Adding op 'model.6.m.0.cv2.conv.weight' of type const Adding op 'model.6.m.1.cv1.conv.bias' of type const Adding op 'model.6.m.1.cv1.conv.weight' of type const Adding op 'model.6.m.1.cv2.conv.bias' of type const Adding op 'model.6.m.1.cv2.conv.weight' of type const Adding op 'model.6.m.2.cv1.conv.bias' of type const Adding op 'model.6.m.2.cv1.conv.weight' of type const Adding op 'model.6.m.2.cv2.conv.bias' of type const Adding op 'model.6.m.2.cv2.conv.weight' of type const Adding op 'model.6.m.3.cv1.conv.bias' of type const Adding op 'model.6.m.3.cv1.conv.weight' of type const Adding op 'model.6.m.3.cv2.conv.bias' of type const Adding op 'model.6.m.3.cv2.conv.weight' of type const Adding op 'model.6.m.4.cv1.conv.bias' of type const Adding op 'model.6.m.4.cv1.conv.weight' of type const Adding op 'model.6.m.4.cv2.conv.bias' of type const Adding op 'model.6.m.4.cv2.conv.weight' of type const Adding op 'model.6.m.5.cv1.conv.bias' of type const Adding op 'model.6.m.5.cv1.conv.weight' of type const Adding op 'model.6.m.5.cv2.conv.bias' of type const Adding op 'model.6.m.5.cv2.conv.weight' of type const Adding op 'model.6.cv2.conv.bias' of type const Adding op 'model.6.cv2.conv.weight' of type const Adding op 'model.6.cv3.conv.bias' of type const Adding op 'model.6.cv3.conv.weight' of type const Adding op 'model.7.conv.bias' of type const Adding op 'model.7.conv.weight' of type const Adding op 'model.8.cv1.conv.bias' of type const Adding op 'model.8.cv1.conv.weight' of type const Adding op 'model.8.m.0.cv1.conv.bias' of type const Adding op 'model.8.m.0.cv1.conv.weight' of type const Adding op 'model.8.m.0.cv2.conv.bias' of type const Adding op 'model.8.m.0.cv2.conv.weight' of type const Adding op 'model.8.m.1.cv1.conv.bias' of type const Adding op 'model.8.m.1.cv1.conv.weight' of type const Adding op 'model.8.m.1.cv2.conv.bias' of type const Adding op 'model.8.m.1.cv2.conv.weight' of type const Adding op 'model.8.cv2.conv.bias' of type const Adding op 'model.8.cv2.conv.weight' of type const Adding op 'model.8.cv3.conv.bias' of type const Adding op 'model.8.cv3.conv.weight' of type const Adding op 'model.9.conv.bias' of type const Adding op 'model.9.conv.weight' of type const Adding op 'model.10.cv1.conv.bias' of type const Adding op 'model.10.cv1.conv.weight' of type const Adding op 'model.10.m.0.cv1.conv.bias' of type const Adding op 'model.10.m.0.cv1.conv.weight' of type const Adding op 'model.10.m.0.cv2.conv.bias' of type const Adding op 'model.10.m.0.cv2.conv.weight' of type const Adding op 'model.10.m.1.cv1.conv.bias' of type const Adding op 'model.10.m.1.cv1.conv.weight' of type const Adding op 'model.10.m.1.cv2.conv.bias' of type const Adding op 'model.10.m.1.cv2.conv.weight' of type const Adding op 'model.10.cv2.conv.bias' of type const Adding op 'model.10.cv2.conv.weight' of type const Adding op 'model.10.cv3.conv.bias' of type const Adding op 'model.10.cv3.conv.weight' of type const Adding op 'model.11.cv1.conv.bias' of type const Adding op 'model.11.cv1.conv.weight' of type const Adding op 'model.11.cv2.conv.bias' of type const Adding op 'model.11.cv2.conv.weight' of type const Adding op 'model.12.conv.bias' of type const Adding op 'model.12.conv.weight' of type const Adding op 'model.15.cv1.conv.bias' of type const Adding op 'model.15.cv1.conv.weight' of type const Adding op 'model.15.m.0.cv1.conv.bias' of type const Adding op 'model.15.m.0.cv1.conv.weight' of type const Adding op 'model.15.m.0.cv2.conv.bias' of type const Adding op 'model.15.m.0.cv2.conv.weight' of type const Adding op 'model.15.m.1.cv1.conv.bias' of type const Adding op 'model.15.m.1.cv1.conv.weight' of type const Adding op 'model.15.m.1.cv2.conv.bias' of type const Adding op 'model.15.m.1.cv2.conv.weight' of type const Adding op 'model.15.cv2.conv.bias' of type const Adding op 'model.15.cv2.conv.weight' of type const Adding op 'model.15.cv3.conv.bias' of type const Adding op 'model.15.cv3.conv.weight' of type const Adding op 'model.16.conv.bias' of type const Adding op 'model.16.conv.weight' of type const Adding op 'model.19.cv1.conv.bias' of type const Adding op 'model.19.cv1.conv.weight' of type const Adding op 'model.19.m.0.cv1.conv.bias' of type const Adding op 'model.19.m.0.cv1.conv.weight' of type const Adding op 'model.19.m.0.cv2.conv.bias' of type const Adding op 'model.19.m.0.cv2.conv.weight' of type const Adding op 'model.19.m.1.cv1.conv.bias' of type const Adding op 'model.19.m.1.cv1.conv.weight' of type const Adding op 'model.19.m.1.cv2.conv.bias' of type const Adding op 'model.19.m.1.cv2.conv.weight' of type const Adding op 'model.19.cv2.conv.bias' of type const Adding op 'model.19.cv2.conv.weight' of type const Adding op 'model.19.cv3.conv.bias' of type const Adding op 'model.19.cv3.conv.weight' of type const Adding op 'model.20.conv.bias' of type const Adding op 'model.20.conv.weight' of type const Adding op 'model.23.cv1.conv.bias' of type const Adding op 'model.23.cv1.conv.weight' of type const Adding op 'model.23.m.0.cv1.conv.bias' of type const Adding op 'model.23.m.0.cv1.conv.weight' of type const Adding op 'model.23.m.0.cv2.conv.bias' of type const Adding op 'model.23.m.0.cv2.conv.weight' of type const Adding op 'model.23.m.1.cv1.conv.bias' of type const Adding op 'model.23.m.1.cv1.conv.weight' of type const Adding op 'model.23.m.1.cv2.conv.bias' of type const Adding op 'model.23.m.1.cv2.conv.weight' of type const Adding op 'model.23.cv2.conv.bias' of type const Adding op 'model.23.cv2.conv.weight' of type const Adding op 'model.23.cv3.conv.bias' of type const Adding op 'model.23.cv3.conv.weight' of type const Adding op 'model.24.conv.bias' of type const Adding op 'model.24.conv.weight' of type const Adding op 'model.26.cv1.conv.bias' of type const Adding op 'model.26.cv1.conv.weight' of type const Adding op 'model.26.m.0.cv1.conv.bias' of type const Adding op 'model.26.m.0.cv1.conv.weight' of type const Adding op 'model.26.m.0.cv2.conv.bias' of type const Adding op 'model.26.m.0.cv2.conv.weight' of type const Adding op 'model.26.m.1.cv1.conv.bias' of type const Adding op 'model.26.m.1.cv1.conv.weight' of type const Adding op 'model.26.m.1.cv2.conv.bias' of type const Adding op 'model.26.m.1.cv2.conv.weight' of type const Adding op 'model.26.cv2.conv.bias' of type const Adding op 'model.26.cv2.conv.weight' of type const Adding op 'model.26.cv3.conv.bias' of type const Adding op 'model.26.cv3.conv.weight' of type const Adding op 'model.27.conv.bias' of type const Adding op 'model.27.conv.weight' of type const Adding op 'model.29.cv1.conv.bias' of type const Adding op 'model.29.cv1.conv.weight' of type const Adding op 'model.29.m.0.cv1.conv.bias' of type const Adding op 'model.29.m.0.cv1.conv.weight' of type const Adding op 'model.29.m.0.cv2.conv.bias' of type const Adding op 'model.29.m.0.cv2.conv.weight' of type const Adding op 'model.29.m.1.cv1.conv.bias' of type const Adding op 'model.29.m.1.cv1.conv.weight' of type const Adding op 'model.29.m.1.cv2.conv.bias' of type const Adding op 'model.29.m.1.cv2.conv.weight' of type const Adding op 'model.29.cv2.conv.bias' of type const Adding op 'model.29.cv2.conv.weight' of type const Adding op 'model.29.cv3.conv.bias' of type const Adding op 'model.29.cv3.conv.weight' of type const Adding op 'model.30.conv.bias' of type const Adding op 'model.30.conv.weight' of type const Adding op 'model.32.cv1.conv.bias' of type const Adding op 'model.32.cv1.conv.weight' of type const Adding op 'model.32.m.0.cv1.conv.bias' of type const Adding op 'model.32.m.0.cv1.conv.weight' of type const Adding op 'model.32.m.0.cv2.conv.bias' of type const Adding op 'model.32.m.0.cv2.conv.weight' of type const Adding op 'model.32.m.1.cv1.conv.bias' of type const Adding op 'model.32.m.1.cv1.conv.weight' of type const Adding op 'model.32.m.1.cv2.conv.bias' of type const Adding op 'model.32.m.1.cv2.conv.weight' of type const Adding op 'model.32.cv2.conv.bias' of type const Adding op 'model.32.cv2.conv.weight' of type const Adding op 'model.32.cv3.conv.bias' of type const Adding op 'model.32.cv3.conv.weight' of type const Adding op 'model.33.m.0.bias' of type const Adding op 'model.33.m.0.weight' of type const Adding op 'model.33.m.1.bias' of type const Adding op 'model.33.m.1.weight' of type const Adding op 'model.33.m.2.bias' of type const Adding op 'model.33.m.2.weight' of type const Adding op 'model.33.m.3.bias' of type const Adding op 'model.33.m.3.weight' of type const Converting PyTorch Frontend ==> MIL Ops: 0% 0/873 [00:00<?, ? ops/s]Converting op 52 : constant Adding op '52' of type const Converting op 53 : constant Adding op '53' of type const Converting op 54 : constant Adding op '54' of type const Converting op 55 : constant Adding op '55' of type const Converting op 56 : constant Adding op '56' of type const Converting op 60 : listconstruct Adding op '60' of type const Converting op 61 : listconstruct Adding op '61' of type const Converting op 62 : listconstruct Adding op '62' of type const Converting op 63 : listconstruct Adding op '63' of type const Converting op input.1 : _convolution Adding op 'input.1' of type conv Adding op 'input.1_pad_type_0' of type const Adding op 'input.1_pad0' of type const Converting op input.3 : silu Adding op 'input.3' of type silu Converting op 66 : constant Adding op '66' of type const Converting op 67 : constant Adding op '67' of type const Converting op 68 : constant Adding op '68' of type const Converting op 69 : constant Adding op '69' of type const Converting op 70 : constant Adding op '70' of type const Converting op 74 : listconstruct Adding op '74' of type const Converting op 75 : listconstruct Adding op '75' of type const Converting op 76 : listconstruct Adding op '76' of type const Converting op 77 : listconstruct Adding op '77' of type const Converting op input.5 : _convolution Adding op 'input.5' of type conv Adding op 'input.5_pad_type_0' of type const Adding op 'input.5_pad0' of type const Converting op input.7 : silu Adding op 'input.7' of type silu Converting op 80 : constant Adding op '80' of type const Converting op 81 : constant Adding op '81' of type const Converting op 82 : constant Adding op '82' of type const Converting op 83 : constant Adding op '83' of type const Converting op 91 : listconstruct Adding op '91' of type const Converting op 92 : listconstruct Adding op '92' of type const Converting op 93 : listconstruct Adding op '93' of type const Converting op 94 : listconstruct Adding op '94' of type const Converting op input.9 : _convolution Adding op 'input.9' of type conv Adding op 'input.9_pad_type_0' of type const Adding op 'input.9_pad0' of type const Converting op input.11 : silu Adding op 'input.11' of type silu Converting op 104 : listconstruct Adding op '104' of type const Converting op 105 : listconstruct Adding op '105' of type const Converting op 106 : listconstruct Adding op '106' of type const Converting op 107 : listconstruct Adding op '107' of type const Converting op input.13 : _convolution Adding op 'input.13' of type conv Adding op 'input.13_pad_type_0' of type const Adding op 'input.13_pad0' of type const Converting op input.15 : silu Adding op 'input.15' of type silu Converting op 113 : listconstruct Adding op '113' of type const Converting op 114 : listconstruct Adding op '114' of type const Converting op 115 : listconstruct Adding op '115' of type const Converting op 116 : listconstruct Adding op '116' of type const Converting op input.17 : _convolution Adding op 'input.17' of type conv Adding op 'input.17_pad_type_0' of type const Adding op 'input.17_pad0' of type const Converting op 118 : silu Adding op '118' of type silu Converting op input.19 : add Adding op 'input.19' of type add Converting op 125 : listconstruct Adding op '125' of type const Converting op 126 : listconstruct Adding op '126' of type const Converting op 127 : listconstruct Adding op '127' of type const Converting op 128 : listconstruct Adding op '128' of type const Converting op input.21 : _convolution Adding op 'input.21' of type conv Adding op 'input.21_pad_type_0' of type const Adding op 'input.21_pad0' of type const Converting op input.23 : silu Adding op 'input.23' of type silu Converting op 134 : listconstruct Adding op '134' of type const Converting op 135 : listconstruct Adding op '135' of type const Converting op 136 : listconstruct Adding op '136' of type const Converting op 137 : listconstruct Adding op '137' of type const Converting op input.25 : _convolution Adding op 'input.25' of type conv Adding op 'input.25_pad_type_0' of type const Adding op 'input.25_pad0' of type const Converting op 139 : silu Adding op '139' of type silu Converting op 140 : add Adding op '140' of type add Converting op 144 : listconstruct Adding op '144' of type const Converting op 145 : listconstruct Adding op '145' of type const Converting op 146 : listconstruct Adding op '146' of type const Converting op 147 : listconstruct Adding op '147' of type const Converting op input.27 : _convolution Adding op 'input.27' of type conv Adding op 'input.27_pad_type_0' of type const Adding op 'input.27_pad0' of type const Converting op 149 : silu Adding op '149' of type silu Converting op 150 : listconstruct Converting op input.29 : cat Adding op 'input.29' of type concat Adding op 'input.29_interleave_0' of type const Converting op 155 : listconstruct Adding op '155' of type const Converting op 156 : listconstruct Adding op '156' of type const Converting op 157 : listconstruct Adding op '157' of type const Converting op 158 : listconstruct Adding op '158' of type const Converting op input.31 : _convolution Adding op 'input.31' of type conv Adding op 'input.31_pad_type_0' of type const Adding op 'input.31_pad0' of type const Converting op input.33 : silu Adding op 'input.33' of type silu Converting op 161 : constant Adding op '161' of type const Converting op 162 : constant Adding op '162' of type const Converting op 163 : constant Adding op '163' of type const Converting op 164 : constant Adding op '164' of type const Converting op 165 : constant Adding op '165' of type const Converting op 169 : listconstruct Adding op '169' of type const Converting op 170 : listconstruct Adding op '170' of type const Converting op 171 : listconstruct Adding op '171' of type const Converting op 172 : listconstruct Adding op '172' of type const Converting op input.35 : _convolution Adding op 'input.35' of type conv Adding op 'input.35_pad_type_0' of type const Adding op 'input.35_pad0' of type const Converting op input.37 : silu Adding op 'input.37' of type silu Converting op 175 : constant Adding op '175' of type const Converting op 176 : constant Adding op '176' of type const Converting op 177 : constant Adding op '177' of type const Converting op 178 : constant Adding op '178' of type const Converting op 186 : listconstruct Adding op '186' of type const Converting op 187 : listconstruct Adding op '187' of type const Converting op 188 : listconstruct Adding op '188' of type const Converting op 189 : listconstruct Adding op '189' of type const Converting op input.39 : _convolution Adding op 'input.39' of type conv Adding op 'input.39_pad_type_0' of type const Adding op 'input.39_pad0' of type const Converting op input.41 : silu Adding op 'input.41' of type silu Converting op 201 : listconstruct Adding op '201' of type const Converting op 202 : listconstruct Adding op '202' of type const Converting op 203 : listconstruct Adding op '203' of type const Converting op 204 : listconstruct Adding op '204' of type const Converting op input.43 : _convolution Adding op 'input.43' of type conv Adding op 'input.43_pad_type_0' of type const Adding op 'input.43_pad0' of type const Converting op input.45 : silu Adding op 'input.45' of type silu Converting op 210 : listconstruct Adding op '210' of type const Converting op 211 : listconstruct Adding op '211' of type const Converting op 212 : listconstruct Adding op '212' of type const Converting op 213 : listconstruct Adding op '213' of type const Converting op input.47 : _convolution Adding op 'input.47' of type conv Adding op 'input.47_pad_type_0' of type const Adding op 'input.47_pad0' of type const Converting op 215 : silu Adding op '215' of type silu Converting op input.49 : add Adding op 'input.49' of type add Converting op 222 : listconstruct Adding op '222' of type const Converting op 223 : listconstruct Adding op '223' of type const Converting op 224 : listconstruct Adding op '224' of type const Converting op 225 : listconstruct Adding op '225' of type const Converting op input.51 : _convolution Adding op 'input.51' of type conv Adding op 'input.51_pad_type_0' of type const Adding op 'input.51_pad0' of type const Converting op input.53 : silu Adding op 'input.53' of type silu Converting op 231 : listconstruct Adding op '231' of type const Converting op 232 : listconstruct Adding op '232' of type const Converting op 233 : listconstruct Adding op '233' of type const Converting op 234 : listconstruct Adding op '234' of type const Converting op input.55 : _convolution Adding op 'input.55' of type conv Adding op 'input.55_pad_type_0' of type const Adding op 'input.55_pad0' of type const Converting op 236 : silu Adding op '236' of type silu Converting op input.57 : add Adding op 'input.57' of type add Converting op 243 : listconstruct Adding op '243' of type const Converting op 244 : listconstruct Adding op '244' of type const Converting op 245 : listconstruct Adding op '245' of type const Converting op 246 : listconstruct Adding op '246' of type const Converting op input.59 : _convolution Adding op 'input.59' of type conv Adding op 'input.59_pad_type_0' of type const Adding op 'input.59_pad0' of type const Converting op input.61 : silu Adding op 'input.61' of type silu Converting op 252 : listconstruct Adding op '252' of type const Converting op 253 : listconstruct Adding op '253' of type const Converting op 254 : listconstruct Adding op '254' of type const Converting op 255 : listconstruct Adding op '255' of type const Converting op input.63 : _convolution Adding op 'input.63' of type conv Adding op 'input.63_pad_type_0' of type const Adding op 'input.63_pad0' of type const Converting op 257 : silu Adding op '257' of type silu Converting op input.65 : add Adding op 'input.65' of type add Converting PyTorch Frontend ==> MIL Ops: 15% 132/873 [00:00<00:00, 1312.69 ops/s]Converting op 264 : listconstruct Adding op '264' of type const Converting op 265 : listconstruct Adding op '265' of type const Converting op 266 : listconstruct Adding op '266' of type const Converting op 267 : listconstruct Adding op '267' of type const Converting op input.67 : _convolution Adding op 'input.67' of type conv Adding op 'input.67_pad_type_0' of type const Adding op 'input.67_pad0' of type const Converting op input.69 : silu Adding op 'input.69' of type silu Converting op 273 : listconstruct Adding op '273' of type const Converting op 274 : listconstruct Adding op '274' of type const Converting op 275 : listconstruct Adding op '275' of type const Converting op 276 : listconstruct Adding op '276' of type const Converting op input.71 : _convolution Adding op 'input.71' of type conv Adding op 'input.71_pad_type_0' of type const Adding op 'input.71_pad0' of type const Converting op 278 : silu Adding op '278' of type silu Converting op 279 : add Adding op '279' of type add Converting op 283 : listconstruct Adding op '283' of type const Converting op 284 : listconstruct Adding op '284' of type const Converting op 285 : listconstruct Adding op '285' of type const Converting op 286 : listconstruct Adding op '286' of type const Converting op input.73 : _convolution Adding op 'input.73' of type conv Adding op 'input.73_pad_type_0' of type const Adding op 'input.73_pad0' of type const Converting op 288 : silu Adding op '288' of type silu Converting op 289 : listconstruct Converting op input.75 : cat Adding op 'input.75' of type concat Adding op 'input.75_interleave_0' of type const Converting op 294 : listconstruct Adding op '294' of type const Converting op 295 : listconstruct Adding op '295' of type const Converting op 296 : listconstruct Adding op '296' of type const Converting op 297 : listconstruct Adding op '297' of type const Converting op input.77 : _convolution Adding op 'input.77' of type conv Adding op 'input.77_pad_type_0' of type const Adding op 'input.77_pad0' of type const Converting op input.79 : silu Adding op 'input.79' of type silu Converting op 300 : constant Adding op '300' of type const Converting op 301 : constant Adding op '301' of type const Converting op 302 : constant Adding op '302' of type const Converting op 303 : constant Adding op '303' of type const Converting op 304 : constant Adding op '304' of type const Converting op 308 : listconstruct Adding op '308' of type const Converting op 309 : listconstruct Adding op '309' of type const Converting op 310 : listconstruct Adding op '310' of type const Converting op 311 : listconstruct Adding op '311' of type const Converting op input.81 : _convolution Adding op 'input.81' of type conv Adding op 'input.81_pad_type_0' of type const Adding op 'input.81_pad0' of type const Converting op input.83 : silu Adding op 'input.83' of type silu Converting op 314 : constant Adding op '314' of type const Converting op 315 : constant Adding op '315' of type const Converting op 316 : constant Adding op '316' of type const Converting op 317 : constant Adding op '317' of type const Converting op 325 : listconstruct Adding op '325' of type const Converting op 326 : listconstruct Adding op '326' of type const Converting op 327 : listconstruct Adding op '327' of type const Converting op 328 : listconstruct Adding op '328' of type const Converting op input.85 : _convolution Adding op 'input.85' of type conv Adding op 'input.85_pad_type_0' of type const Adding op 'input.85_pad0' of type const Converting op input.87 : silu Adding op 'input.87' of type silu Converting op 342 : listconstruct Adding op '342' of type const Converting op 343 : listconstruct Adding op '343' of type const Converting op 344 : listconstruct Adding op '344' of type const Converting op 345 : listconstruct Adding op '345' of type const Converting op input.89 : _convolution Adding op 'input.89' of type conv Adding op 'input.89_pad_type_0' of type const Adding op 'input.89_pad0' of type const Converting op input.91 : silu Adding op 'input.91' of type silu Converting op 351 : listconstruct Adding op '351' of type const Converting op 352 : listconstruct Adding op '352' of type const Converting op 353 : listconstruct Adding op '353' of type const Converting op 354 : listconstruct Adding op '354' of type const Converting op input.93 : _convolution Adding op 'input.93' of type conv Adding op 'input.93_pad_type_0' of type const Adding op 'input.93_pad0' of type const Converting op 356 : silu Adding op '356' of type silu Converting op input.95 : add Adding op 'input.95' of type add Converting op 363 : listconstruct Adding op '363' of type const Converting op 364 : listconstruct Adding op '364' of type const Converting op 365 : listconstruct Adding op '365' of type const Converting op 366 : listconstruct Adding op '366' of type const Converting op input.97 : _convolution Adding op 'input.97' of type conv Adding op 'input.97_pad_type_0' of type const Adding op 'input.97_pad0' of type const Converting op input.99 : silu Adding op 'input.99' of type silu Converting op 372 : listconstruct Adding op '372' of type const Converting op 373 : listconstruct Adding op '373' of type const Converting op 374 : listconstruct Adding op '374' of type const Converting op 375 : listconstruct Adding op '375' of type const Converting op input.101 : _convolution Adding op 'input.101' of type conv Adding op 'input.101_pad_type_0' of type const Adding op 'input.101_pad0' of type const Converting op 377 : silu Adding op '377' of type silu Converting op input.103 : add Adding op 'input.103' of type add Converting op 384 : listconstruct Adding op '384' of type const Converting op 385 : listconstruct Adding op '385' of type const Converting op 386 : listconstruct Adding op '386' of type const Converting op 387 : listconstruct Adding op '387' of type const Converting op input.105 : _convolution Adding op 'input.105' of type conv Adding op 'input.105_pad_type_0' of type const Adding op 'input.105_pad0' of type const Converting op input.107 : silu Adding op 'input.107' of type silu Converting op 393 : listconstruct Adding op '393' of type const Converting op 394 : listconstruct Adding op '394' of type const Converting op 395 : listconstruct Adding op '395' of type const Converting op 396 : listconstruct Adding op '396' of type const Converting op input.109 : _convolution Adding op 'input.109' of type conv Adding op 'input.109_pad_type_0' of type const Adding op 'input.109_pad0' of type const Converting op 398 : silu Adding op '398' of type silu Converting op input.111 : add Adding op 'input.111' of type add Converting op 405 : listconstruct Adding op '405' of type const Converting op 406 : listconstruct Adding op '406' of type const Converting op 407 : listconstruct Adding op '407' of type const Converting op 408 : listconstruct Adding op '408' of type const Converting op input.113 : _convolution Adding op 'input.113' of type conv Adding op 'input.113_pad_type_0' of type const Adding op 'input.113_pad0' of type const Converting op input.115 : silu Adding op 'input.115' of type silu Converting op 414 : listconstruct Adding op '414' of type const Converting op 415 : listconstruct Adding op '415' of type const Converting op 416 : listconstruct Adding op '416' of type const Converting op 417 : listconstruct Adding op '417' of type const Converting op input.117 : _convolution Adding op 'input.117' of type conv Adding op 'input.117_pad_type_0' of type const Adding op 'input.117_pad0' of type const Converting op 419 : silu Adding op '419' of type silu Converting op input.119 : add Adding op 'input.119' of type add Converting op 426 : listconstruct Adding op '426' of type const Converting op 427 : listconstruct Adding op '427' of type const Converting op 428 : listconstruct Adding op '428' of type const Converting op 429 : listconstruct Adding op '429' of type const Converting op input.121 : _convolution Adding op 'input.121' of type conv Adding op 'input.121_pad_type_0' of type const Adding op 'input.121_pad0' of type const Converting op input.123 : silu Adding op 'input.123' of type silu Converting op 435 : listconstruct Adding op '435' of type const Converting op 436 : listconstruct Adding op '436' of type const Converting op 437 : listconstruct Adding op '437' of type const Converting op 438 : listconstruct Adding op '438' of type const Converting op input.125 : _convolution Adding op 'input.125' of type conv Adding op 'input.125_pad_type_0' of type const Adding op 'input.125_pad0' of type const Converting op 440 : silu Adding op '440' of type silu Converting op input.127 : add Adding op 'input.127' of type add Converting op 447 : listconstruct Adding op '447' of type const Converting op 448 : listconstruct Adding op '448' of type const Converting op 449 : listconstruct Adding op '449' of type const Converting op 450 : listconstruct Adding op '450' of type const Converting op input.129 : _convolution Adding op 'input.129' of type conv Adding op 'input.129_pad_type_0' of type const Adding op 'input.129_pad0' of type const Converting op input.131 : silu Adding op 'input.131' of type silu Converting op 456 : listconstruct Adding op '456' of type const Converting op 457 : listconstruct Adding op '457' of type const Converting op 458 : listconstruct Adding op '458' of type const Converting op 459 : listconstruct Adding op '459' of type const Converting op input.133 : _convolution Adding op 'input.133' of type conv Adding op 'input.133_pad_type_0' of type const Adding op 'input.133_pad0' of type const Converting op 461 : silu Adding op '461' of type silu Converting op 462 : add Adding op '462' of type add Converting op 466 : listconstruct Adding op '466' of type const Converting op 467 : listconstruct Adding op '467' of type const Converting op 468 : listconstruct Adding op '468' of type const Converting op 469 : listconstruct Adding op '469' of type const Converting op input.135 : _convolution Adding op 'input.135' of type conv Adding op 'input.135_pad_type_0' of type const Adding op 'input.135_pad0' of type const Converting op 471 : silu Adding op '471' of type silu Converting PyTorch Frontend ==> MIL Ops: 30% 264/873 [00:00<00:00, 1285.36 ops/s]Converting op 472 : listconstruct Converting op input.137 : cat Adding op 'input.137' of type concat Adding op 'input.137_interleave_0' of type const Converting op 477 : listconstruct Adding op '477' of type const Converting op 478 : listconstruct Adding op '478' of type const Converting op 479 : listconstruct Adding op '479' of type const Converting op 480 : listconstruct Adding op '480' of type const Converting op input.139 : _convolution Adding op 'input.139' of type conv Adding op 'input.139_pad_type_0' of type const Adding op 'input.139_pad0' of type const Converting op input.141 : silu Adding op 'input.141' of type silu Converting op 483 : constant Adding op '483' of type const Converting op 484 : constant Adding op '484' of type const Converting op 485 : constant Adding op '485' of type const Converting op 486 : constant Adding op '486' of type const Converting op 487 : constant Adding op '487' of type const Converting op 491 : listconstruct Adding op '491' of type const Converting op 492 : listconstruct Adding op '492' of type const Converting op 493 : listconstruct Adding op '493' of type const Converting op 494 : listconstruct Adding op '494' of type const Converting op input.143 : _convolution Adding op 'input.143' of type conv Adding op 'input.143_pad_type_0' of type const Adding op 'input.143_pad0' of type const Converting op input.145 : silu Adding op 'input.145' of type silu Converting op 497 : constant Adding op '497' of type const Converting op 498 : constant Adding op '498' of type const Converting op 499 : constant Adding op '499' of type const Converting op 500 : constant Adding op '500' of type const Converting op 508 : listconstruct Adding op '508' of type const Converting op 509 : listconstruct Adding op '509' of type const Converting op 510 : listconstruct Adding op '510' of type const Converting op 511 : listconstruct Adding op '511' of type const Converting op input.147 : _convolution Adding op 'input.147' of type conv Adding op 'input.147_pad_type_0' of type const Adding op 'input.147_pad0' of type const Converting op input.149 : silu Adding op 'input.149' of type silu Converting op 521 : listconstruct Adding op '521' of type const Converting op 522 : listconstruct Adding op '522' of type const Converting op 523 : listconstruct Adding op '523' of type const Converting op 524 : listconstruct Adding op '524' of type const Converting op input.151 : _convolution Adding op 'input.151' of type conv Adding op 'input.151_pad_type_0' of type const Adding op 'input.151_pad0' of type const Converting op input.153 : silu Adding op 'input.153' of type silu Converting op 530 : listconstruct Adding op '530' of type const Converting op 531 : listconstruct Adding op '531' of type const Converting op 532 : listconstruct Adding op '532' of type const Converting op 533 : listconstruct Adding op '533' of type const Converting op input.155 : _convolution Adding op 'input.155' of type conv Adding op 'input.155_pad_type_0' of type const Adding op 'input.155_pad0' of type const Converting op 535 : silu Adding op '535' of type silu Converting op input.157 : add Adding op 'input.157' of type add Converting op 542 : listconstruct Adding op '542' of type const Converting op 543 : listconstruct Adding op '543' of type const Converting op 544 : listconstruct Adding op '544' of type const Converting op 545 : listconstruct Adding op '545' of type const Converting op input.159 : _convolution Adding op 'input.159' of type conv Adding op 'input.159_pad_type_0' of type const Adding op 'input.159_pad0' of type const Converting op input.161 : silu Adding op 'input.161' of type silu Converting op 551 : listconstruct Adding op '551' of type const Converting op 552 : listconstruct Adding op '552' of type const Converting op 553 : listconstruct Adding op '553' of type const Converting op 554 : listconstruct Adding op '554' of type const Converting op input.163 : _convolution Adding op 'input.163' of type conv Adding op 'input.163_pad_type_0' of type const Adding op 'input.163_pad0' of type const Converting op 556 : silu Adding op '556' of type silu Converting op 557 : add Adding op '557' of type add Converting op 561 : listconstruct Adding op '561' of type const Converting op 562 : listconstruct Adding op '562' of type const Converting op 563 : listconstruct Adding op '563' of type const Converting op 564 : listconstruct Adding op '564' of type const Converting op input.165 : _convolution Adding op 'input.165' of type conv Adding op 'input.165_pad_type_0' of type const Adding op 'input.165_pad0' of type const Converting op 566 : silu Adding op '566' of type silu Converting op 567 : listconstruct Converting op input.167 : cat Adding op 'input.167' of type concat Adding op 'input.167_interleave_0' of type const Converting op 572 : listconstruct Adding op '572' of type const Converting op 573 : listconstruct Adding op '573' of type const Converting op 574 : listconstruct Adding op '574' of type const Converting op 575 : listconstruct Adding op '575' of type const Converting op input.169 : _convolution Adding op 'input.169' of type conv Adding op 'input.169_pad_type_0' of type const Adding op 'input.169_pad0' of type const Converting op input.171 : silu Adding op 'input.171' of type silu Converting op 578 : constant Adding op '578' of type const Converting op 579 : constant Adding op '579' of type const Converting op 580 : constant Adding op '580' of type const Converting op 581 : constant Adding op '581' of type const Converting op 582 : constant Adding op '582' of type const Converting op 586 : listconstruct Adding op '586' of type const Converting op 587 : listconstruct Adding op '587' of type const Converting op 588 : listconstruct Adding op '588' of type const Converting op 589 : listconstruct Adding op '589' of type const Converting op input.173 : _convolution Adding op 'input.173' of type conv Adding op 'input.173_pad_type_0' of type const Adding op 'input.173_pad0' of type const Converting op input.175 : silu Adding op 'input.175' of type silu Converting op 592 : constant Adding op '592' of type const Converting op 593 : constant Adding op '593' of type const Converting op 594 : constant Adding op '594' of type const Converting op 595 : constant Adding op '595' of type const Converting op 603 : listconstruct Adding op '603' of type const Converting op 604 : listconstruct Adding op '604' of type const Converting op 605 : listconstruct Adding op '605' of type const Converting op 606 : listconstruct Adding op '606' of type const Converting op input.177 : _convolution Adding op 'input.177' of type conv Adding op 'input.177_pad_type_0' of type const Adding op 'input.177_pad0' of type const Converting op input.179 : silu Adding op 'input.179' of type silu Converting op 616 : listconstruct Adding op '616' of type const Converting op 617 : listconstruct Adding op '617' of type const Converting op 618 : listconstruct Adding op '618' of type const Converting op 619 : listconstruct Adding op '619' of type const Converting op input.181 : _convolution Adding op 'input.181' of type conv Adding op 'input.181_pad_type_0' of type const Adding op 'input.181_pad0' of type const Converting op input.183 : silu Adding op 'input.183' of type silu Converting op 625 : listconstruct Adding op '625' of type const Converting op 626 : listconstruct Adding op '626' of type const Converting op 627 : listconstruct Adding op '627' of type const Converting op 628 : listconstruct Adding op '628' of type const Converting op input.185 : _convolution Adding op 'input.185' of type conv Adding op 'input.185_pad_type_0' of type const Adding op 'input.185_pad0' of type const Converting op 630 : silu Adding op '630' of type silu Converting op input.187 : add Adding op 'input.187' of type add Converting op 637 : listconstruct Adding op '637' of type const Converting op 638 : listconstruct Adding op '638' of type const Converting op 639 : listconstruct Adding op '639' of type const Converting op 640 : listconstruct Adding op '640' of type const Converting op input.189 : _convolution Adding op 'input.189' of type conv Adding op 'input.189_pad_type_0' of type const Adding op 'input.189_pad0' of type const Converting op input.191 : silu Adding op 'input.191' of type silu Converting op 646 : listconstruct Adding op '646' of type const Converting op 647 : listconstruct Adding op '647' of type const Converting op 648 : listconstruct Adding op '648' of type const Converting op 649 : listconstruct Adding op '649' of type const Converting op input.193 : _convolution Adding op 'input.193' of type conv Adding op 'input.193_pad_type_0' of type const Adding op 'input.193_pad0' of type const Converting op 651 : silu Adding op '651' of type silu Converting op 652 : add Adding op '652' of type add Converting op 656 : listconstruct Adding op '656' of type const Converting op 657 : listconstruct Adding op '657' of type const Converting op 658 : listconstruct Adding op '658' of type const Converting op 659 : listconstruct Adding op '659' of type const Converting op input.195 : _convolution Adding op 'input.195' of type conv Adding op 'input.195_pad_type_0' of type const Adding op 'input.195_pad0' of type const Converting op 661 : silu Adding op '661' of type silu Converting op 662 : listconstruct Converting op input.197 : cat Adding op 'input.197' of type concat Adding op 'input.197_interleave_0' of type const Converting op 667 : listconstruct Adding op '667' of type const Converting op 668 : listconstruct Adding op '668' of type const Converting op 669 : listconstruct Adding op '669' of type const Converting op 670 : listconstruct Adding op '670' of type const Converting op input.199 : _convolution Adding op 'input.199' of type conv Adding op 'input.199_pad_type_0' of type const Adding op 'input.199_pad0' of type const Converting PyTorch Frontend ==> MIL Ops: 45% 393/873 [00:00<00:00, 1266.98 ops/s]Converting op input.201 : silu Adding op 'input.201' of type silu Converting op 673 : constant Adding op '673' of type const Converting op 674 : constant Adding op '674' of type const Converting op 675 : constant Adding op '675' of type const Converting op 676 : constant Adding op '676' of type const Converting op 677 : constant Adding op '677' of type const Converting op 678 : constant Adding op '678' of type const Converting op 684 : listconstruct Adding op '684' of type const Converting op 685 : listconstruct Adding op '685' of type const Converting op 686 : listconstruct Adding op '686' of type const Converting op 687 : listconstruct Adding op '687' of type const Converting op input.203 : _convolution Adding op 'input.203' of type conv Adding op 'input.203_pad_type_0' of type const Adding op 'input.203_pad0' of type const Converting op input.205 : silu Adding op 'input.205' of type silu Converting op 690 : listconstruct Adding op '690' of type const Converting op 691 : listconstruct Adding op '691' of type const Converting op 692 : listconstruct Adding op '692' of type const Converting op 693 : listconstruct Adding op '693' of type const Converting op input.207 : max_pool2d Adding op 'input.207' of type max_pool Adding op 'input.207_pad_type_0' of type const Adding op 'input.207_pad_0' of type const Adding op 'input.207_ceil_mode_0' of type const Converting op 695 : listconstruct Adding op '695' of type const Converting op 696 : listconstruct Adding op '696' of type const Converting op 697 : listconstruct Adding op '697' of type const Converting op 698 : listconstruct Adding op '698' of type const Converting op input.209 : max_pool2d Adding op 'input.209' of type max_pool Adding op 'input.209_pad_type_0' of type const Adding op 'input.209_pad_0' of type const Adding op 'input.209_ceil_mode_0' of type const Converting op 700 : listconstruct Adding op '700' of type const Converting op 701 : listconstruct Adding op '701' of type const Converting op 702 : listconstruct Adding op '702' of type const Converting op 703 : listconstruct Adding op '703' of type const Converting op 704 : max_pool2d Adding op '704' of type max_pool Adding op '704_pad_type_0' of type const Adding op '704_pad_0' of type const Adding op '704_ceil_mode_0' of type const Converting op 705 : listconstruct Converting op input.211 : cat Adding op 'input.211' of type concat Adding op 'input.211_interleave_0' of type const Converting op 710 : listconstruct Adding op '710' of type const Converting op 711 : listconstruct Adding op '711' of type const Converting op 712 : listconstruct Adding op '712' of type const Converting op 713 : listconstruct Adding op '713' of type const Converting op input.213 : _convolution Adding op 'input.213' of type conv Adding op 'input.213_pad_type_0' of type const Adding op 'input.213_pad0' of type const Converting op input.215 : silu Adding op 'input.215' of type silu Converting op 716 : constant Adding op '716' of type const Converting op 717 : constant Adding op '717' of type const Converting op 718 : constant Adding op '718' of type const Converting op 719 : constant Adding op '719' of type const Converting op 723 : listconstruct Adding op '723' of type const Converting op 724 : listconstruct Adding op '724' of type const Converting op 725 : listconstruct Adding op '725' of type const Converting op 726 : listconstruct Adding op '726' of type const Converting op input.217 : _convolution Adding op 'input.217' of type conv Adding op 'input.217_pad_type_0' of type const Adding op 'input.217_pad0' of type const Converting op input.219 : silu Adding op 'input.219' of type silu Converting op 729 : constant Adding op '729' of type const Converting op 730 : constant Converting op 731 : listconstruct Adding op '731' of type const Converting op 732 : upsample_nearest2d Adding op '732' of type upsample_nearest_neighbor Adding op '732_scale_factor_height_0' of type const Adding op '732_scale_factor_width_0' of type const Converting op 733 : constant Adding op '733' of type const Converting op 734 : listconstruct Converting op input.221 : cat Adding op 'input.221' of type concat Adding op 'input.221_interleave_0' of type const Converting op 736 : constant Adding op '736' of type const Converting op 737 : constant Adding op '737' of type const Converting op 738 : constant Adding op '738' of type const Converting op 739 : constant Adding op '739' of type const Converting op 747 : listconstruct Adding op '747' of type const Converting op 748 : listconstruct Adding op '748' of type const Converting op 749 : listconstruct Adding op '749' of type const Converting op 750 : listconstruct Adding op '750' of type const Converting op input.223 : _convolution Adding op 'input.223' of type conv Adding op 'input.223_pad_type_0' of type const Adding op 'input.223_pad0' of type const Converting op input.225 : silu Adding op 'input.225' of type silu Converting op 760 : listconstruct Adding op '760' of type const Converting op 761 : listconstruct Adding op '761' of type const Converting op 762 : listconstruct Adding op '762' of type const Converting op 763 : listconstruct Adding op '763' of type const Converting op input.227 : _convolution Adding op 'input.227' of type conv Adding op 'input.227_pad_type_0' of type const Adding op 'input.227_pad0' of type const Converting op input.229 : silu Adding op 'input.229' of type silu Converting op 769 : listconstruct Adding op '769' of type const Converting op 770 : listconstruct Adding op '770' of type const Converting op 771 : listconstruct Adding op '771' of type const Converting op 772 : listconstruct Adding op '772' of type const Converting op input.231 : _convolution Adding op 'input.231' of type conv Adding op 'input.231_pad_type_0' of type const Adding op 'input.231_pad0' of type const Converting op input.233 : silu Adding op 'input.233' of type silu Converting op 780 : listconstruct Adding op '780' of type const Converting op 781 : listconstruct Adding op '781' of type const Converting op 782 : listconstruct Adding op '782' of type const Converting op 783 : listconstruct Adding op '783' of type const Converting op input.235 : _convolution Adding op 'input.235' of type conv Adding op 'input.235_pad_type_0' of type const Adding op 'input.235_pad0' of type const Converting op input.237 : silu Adding op 'input.237' of type silu Converting op 789 : listconstruct Adding op '789' of type const Converting op 790 : listconstruct Adding op '790' of type const Converting op 791 : listconstruct Adding op '791' of type const Converting op 792 : listconstruct Adding op '792' of type const Converting op input.239 : _convolution Adding op 'input.239' of type conv Adding op 'input.239_pad_type_0' of type const Adding op 'input.239_pad0' of type const Converting op 794 : silu Adding op '794' of type silu Converting op 798 : listconstruct Adding op '798' of type const Converting op 799 : listconstruct Adding op '799' of type const Converting op 800 : listconstruct Adding op '800' of type const Converting op 801 : listconstruct Adding op '801' of type const Converting op input.241 : _convolution Adding op 'input.241' of type conv Adding op 'input.241_pad_type_0' of type const Adding op 'input.241_pad0' of type const Converting op 803 : silu Adding op '803' of type silu Converting op 804 : listconstruct Converting op input.243 : cat Adding op 'input.243' of type concat Adding op 'input.243_interleave_0' of type const Converting op 809 : listconstruct Adding op '809' of type const Converting op 810 : listconstruct Adding op '810' of type const Converting op 811 : listconstruct Adding op '811' of type const Converting op 812 : listconstruct Adding op '812' of type const Converting op input.245 : _convolution Adding op 'input.245' of type conv Adding op 'input.245_pad_type_0' of type const Adding op 'input.245_pad0' of type const Converting op input.247 : silu Adding op 'input.247' of type silu Converting op 815 : constant Adding op '815' of type const Converting op 816 : constant Adding op '816' of type const Converting op 817 : constant Adding op '817' of type const Converting op 818 : constant Adding op '818' of type const Converting op 822 : listconstruct Adding op '822' of type const Converting op 823 : listconstruct Adding op '823' of type const Converting op 824 : listconstruct Adding op '824' of type const Converting op 825 : listconstruct Adding op '825' of type const Converting op input.249 : _convolution Adding op 'input.249' of type conv Adding op 'input.249_pad_type_0' of type const Adding op 'input.249_pad0' of type const Converting op input.251 : silu Adding op 'input.251' of type silu Converting op 828 : constant Adding op '828' of type const Converting op 829 : constant Converting op 830 : listconstruct Adding op '830' of type const Converting op 831 : upsample_nearest2d Adding op '831' of type upsample_nearest_neighbor Adding op '831_scale_factor_height_0' of type const Adding op '831_scale_factor_width_0' of type const Converting op 832 : constant Adding op '832' of type const Converting op 833 : listconstruct Converting op input.253 : cat Adding op 'input.253' of type concat Adding op 'input.253_interleave_0' of type const Converting op 835 : constant Adding op '835' of type const Converting op 836 : constant Adding op '836' of type const Converting op 837 : constant Adding op '837' of type const Converting op 838 : constant Adding op '838' of type const Converting op 846 : listconstruct Adding op '846' of type const Converting op 847 : listconstruct Adding op '847' of type const Converting op 848 : listconstruct Adding op '848' of type const Converting op 849 : listconstruct Adding op '849' of type const Converting op input.255 : _convolution Adding op 'input.255' of type conv Adding op 'input.255_pad_type_0' of type const Adding op 'input.255_pad0' of type const Converting PyTorch Frontend ==> MIL Ops: 60% 520/873 [00:00<00:00, 1249.63 ops/s]Converting op input.257 : silu Adding op 'input.257' of type silu Converting op 859 : listconstruct Adding op '859' of type const Converting op 860 : listconstruct Adding op '860' of type const Converting op 861 : listconstruct Adding op '861' of type const Converting op 862 : listconstruct Adding op '862' of type const Converting op input.259 : _convolution Adding op 'input.259' of type conv Adding op 'input.259_pad_type_0' of type const Adding op 'input.259_pad0' of type const Converting op input.261 : silu Adding op 'input.261' of type silu Converting op 868 : listconstruct Adding op '868' of type const Converting op 869 : listconstruct Adding op '869' of type const Converting op 870 : listconstruct Adding op '870' of type const Converting op 871 : listconstruct Adding op '871' of type const Converting op input.263 : _convolution Adding op 'input.263' of type conv Adding op 'input.263_pad_type_0' of type const Adding op 'input.263_pad0' of type const Converting op input.265 : silu Adding op 'input.265' of type silu Converting op 879 : listconstruct Adding op '879' of type const Converting op 880 : listconstruct Adding op '880' of type const Converting op 881 : listconstruct Adding op '881' of type const Converting op 882 : listconstruct Adding op '882' of type const Converting op input.267 : _convolution Adding op 'input.267' of type conv Adding op 'input.267_pad_type_0' of type const Adding op 'input.267_pad0' of type const Converting op input.269 : silu Adding op 'input.269' of type silu Converting op 888 : listconstruct Adding op '888' of type const Converting op 889 : listconstruct Adding op '889' of type const Converting op 890 : listconstruct Adding op '890' of type const Converting op 891 : listconstruct Adding op '891' of type const Converting op input.271 : _convolution Adding op 'input.271' of type conv Adding op 'input.271_pad_type_0' of type const Adding op 'input.271_pad0' of type const Converting op 893 : silu Adding op '893' of type silu Converting op 897 : listconstruct Adding op '897' of type const Converting op 898 : listconstruct Adding op '898' of type const Converting op 899 : listconstruct Adding op '899' of type const Converting op 900 : listconstruct Adding op '900' of type const Converting op input.273 : _convolution Adding op 'input.273' of type conv Adding op 'input.273_pad_type_0' of type const Adding op 'input.273_pad0' of type const Converting op 902 : silu Adding op '902' of type silu Converting op 903 : listconstruct Converting op input.275 : cat Adding op 'input.275' of type concat Adding op 'input.275_interleave_0' of type const Converting op 908 : listconstruct Adding op '908' of type const Converting op 909 : listconstruct Adding op '909' of type const Converting op 910 : listconstruct Adding op '910' of type const Converting op 911 : listconstruct Adding op '911' of type const Converting op input.277 : _convolution Adding op 'input.277' of type conv Adding op 'input.277_pad_type_0' of type const Adding op 'input.277_pad0' of type const Converting op input.279 : silu Adding op 'input.279' of type silu Converting op 914 : constant Adding op '914' of type const Converting op 915 : constant Adding op '915' of type const Converting op 916 : constant Adding op '916' of type const Converting op 917 : constant Adding op '917' of type const Converting op 921 : listconstruct Adding op '921' of type const Converting op 922 : listconstruct Adding op '922' of type const Converting op 923 : listconstruct Adding op '923' of type const Converting op 924 : listconstruct Adding op '924' of type const Converting op input.281 : _convolution Adding op 'input.281' of type conv Adding op 'input.281_pad_type_0' of type const Adding op 'input.281_pad0' of type const Converting op input.283 : silu Adding op 'input.283' of type silu Converting op 927 : constant Adding op '927' of type const Converting op 928 : constant Converting op 929 : listconstruct Adding op '929' of type const Converting op 930 : upsample_nearest2d Adding op '930' of type upsample_nearest_neighbor Adding op '930_scale_factor_height_0' of type const Adding op '930_scale_factor_width_0' of type const Converting op 931 : constant Adding op '931' of type const Converting op 932 : listconstruct Converting op input.285 : cat Adding op 'input.285' of type concat Adding op 'input.285_interleave_0' of type const Converting op 934 : constant Adding op '934' of type const Converting op 935 : constant Adding op '935' of type const Converting op 936 : constant Adding op '936' of type const Converting op 937 : constant Adding op '937' of type const Converting op 945 : listconstruct Adding op '945' of type const Converting op 946 : listconstruct Adding op '946' of type const Converting op 947 : listconstruct Adding op '947' of type const Converting op 948 : listconstruct Adding op '948' of type const Converting op input.287 : _convolution Adding op 'input.287' of type conv Adding op 'input.287_pad_type_0' of type const Adding op 'input.287_pad0' of type const Converting op input.289 : silu Adding op 'input.289' of type silu Converting op 958 : listconstruct Adding op '958' of type const Converting op 959 : listconstruct Adding op '959' of type const Converting op 960 : listconstruct Adding op '960' of type const Converting op 961 : listconstruct Adding op '961' of type const Converting op input.291 : _convolution Adding op 'input.291' of type conv Adding op 'input.291_pad_type_0' of type const Adding op 'input.291_pad0' of type const Converting op input.293 : silu Adding op 'input.293' of type silu Converting op 967 : listconstruct Adding op '967' of type const Converting op 968 : listconstruct Adding op '968' of type const Converting op 969 : listconstruct Adding op '969' of type const Converting op 970 : listconstruct Adding op '970' of type const Converting op input.295 : _convolution Adding op 'input.295' of type conv Adding op 'input.295_pad_type_0' of type const Adding op 'input.295_pad0' of type const Converting op input.297 : silu Adding op 'input.297' of type silu Converting op 978 : listconstruct Adding op '978' of type const Converting op 979 : listconstruct Adding op '979' of type const Converting op 980 : listconstruct Adding op '980' of type const Converting op 981 : listconstruct Adding op '981' of type const Converting op input.299 : _convolution Adding op 'input.299' of type conv Adding op 'input.299_pad_type_0' of type const Adding op 'input.299_pad0' of type const Converting op input.301 : silu Adding op 'input.301' of type silu Converting op 987 : listconstruct Adding op '987' of type const Converting op 988 : listconstruct Adding op '988' of type const Converting op 989 : listconstruct Adding op '989' of type const Converting op 990 : listconstruct Adding op '990' of type const Converting op input.303 : _convolution Adding op 'input.303' of type conv Adding op 'input.303_pad_type_0' of type const Adding op 'input.303_pad0' of type const Converting op 992 : silu Adding op '992' of type silu Converting op 996 : listconstruct Adding op '996' of type const Converting op 997 : listconstruct Adding op '997' of type const Converting op 998 : listconstruct Adding op '998' of type const Converting op 999 : listconstruct Adding op '999' of type const Converting op input.305 : _convolution Adding op 'input.305' of type conv Adding op 'input.305_pad_type_0' of type const Adding op 'input.305_pad0' of type const Converting op 1001 : silu Adding op '1001' of type silu Converting op 1002 : listconstruct Converting op input.307 : cat Adding op 'input.307' of type concat Adding op 'input.307_interleave_0' of type const Converting op 1007 : listconstruct Adding op '1007' of type const Converting op 1008 : listconstruct Adding op '1008' of type const Converting op 1009 : listconstruct Adding op '1009' of type const Converting op 1010 : listconstruct Adding op '1010' of type const Converting op input.309 : _convolution Adding op 'input.309' of type conv Adding op 'input.309_pad_type_0' of type const Adding op 'input.309_pad0' of type const Converting op input.311 : silu Adding op 'input.311' of type silu Converting op 1013 : constant Adding op '1013' of type const Converting op 1014 : constant Adding op '1014' of type const Converting op 1015 : constant Adding op '1015' of type const Converting op 1016 : constant Adding op '1016' of type const Converting op 1017 : constant Adding op '1017' of type const Converting op 1021 : listconstruct Adding op '1021' of type const Converting op 1022 : listconstruct Adding op '1022' of type const Converting op 1023 : listconstruct Adding op '1023' of type const Converting op 1024 : listconstruct Adding op '1024' of type const Converting op input.313 : _convolution Adding op 'input.313' of type conv Adding op 'input.313_pad_type_0' of type const Adding op 'input.313_pad0' of type const Converting op 1026 : silu Adding op '1026' of type silu Converting op 1027 : constant Adding op '1027' of type const Converting op 1028 : listconstruct Converting op input.315 : cat Adding op 'input.315' of type concat Adding op 'input.315_interleave_0' of type const Converting op 1030 : constant Adding op '1030' of type const Converting op 1031 : constant Adding op '1031' of type const Converting op 1032 : constant Adding op '1032' of type const Converting op 1033 : constant Adding op '1033' of type const Converting op 1041 : listconstruct Adding op '1041' of type const Converting op 1042 : listconstruct Adding op '1042' of type const Converting op 1043 : listconstruct Adding op '1043' of type const Converting PyTorch Frontend ==> MIL Ops: 74% 645/873 [00:00<00:00, 1145.11 ops/s]Converting op 1044 : listconstruct Adding op '1044' of type const Converting op input.317 : _convolution Adding op 'input.317' of type conv Adding op 'input.317_pad_type_0' of type const Adding op 'input.317_pad0' of type const Converting op input.319 : silu Adding op 'input.319' of type silu Converting op 1054 : listconstruct Adding op '1054' of type const Converting op 1055 : listconstruct Adding op '1055' of type const Converting op 1056 : listconstruct Adding op '1056' of type const Converting op 1057 : listconstruct Adding op '1057' of type const Converting op input.321 : _convolution Adding op 'input.321' of type conv Adding op 'input.321_pad_type_0' of type const Adding op 'input.321_pad0' of type const Converting op input.323 : silu Adding op 'input.323' of type silu Converting op 1063 : listconstruct Adding op '1063' of type const Converting op 1064 : listconstruct Adding op '1064' of type const Converting op 1065 : listconstruct Adding op '1065' of type const Converting op 1066 : listconstruct Adding op '1066' of type const Converting op input.325 : _convolution Adding op 'input.325' of type conv Adding op 'input.325_pad_type_0' of type const Adding op 'input.325_pad0' of type const Converting op input.327 : silu Adding op 'input.327' of type silu Converting op 1074 : listconstruct Adding op '1074' of type const Converting op 1075 : listconstruct Adding op '1075' of type const Converting op 1076 : listconstruct Adding op '1076' of type const Converting op 1077 : listconstruct Adding op '1077' of type const Converting op input.329 : _convolution Adding op 'input.329' of type conv Adding op 'input.329_pad_type_0' of type const Adding op 'input.329_pad0' of type const Converting op input.331 : silu Adding op 'input.331' of type silu Converting op 1083 : listconstruct Adding op '1083' of type const Converting op 1084 : listconstruct Adding op '1084' of type const Converting op 1085 : listconstruct Adding op '1085' of type const Converting op 1086 : listconstruct Adding op '1086' of type const Converting op input.333 : _convolution Adding op 'input.333' of type conv Adding op 'input.333_pad_type_0' of type const Adding op 'input.333_pad0' of type const Converting op 1088 : silu Adding op '1088' of type silu Converting op 1092 : listconstruct Adding op '1092' of type const Converting op 1093 : listconstruct Adding op '1093' of type const Converting op 1094 : listconstruct Adding op '1094' of type const Converting op 1095 : listconstruct Adding op '1095' of type const Converting op input.335 : _convolution Adding op 'input.335' of type conv Adding op 'input.335_pad_type_0' of type const Adding op 'input.335_pad0' of type const Converting op 1097 : silu Adding op '1097' of type silu Converting op 1098 : listconstruct Converting op input.337 : cat Adding op 'input.337' of type concat Adding op 'input.337_interleave_0' of type const Converting op 1103 : listconstruct Adding op '1103' of type const Converting op 1104 : listconstruct Adding op '1104' of type const Converting op 1105 : listconstruct Adding op '1105' of type const Converting op 1106 : listconstruct Adding op '1106' of type const Converting op input.339 : _convolution Adding op 'input.339' of type conv Adding op 'input.339_pad_type_0' of type const Adding op 'input.339_pad0' of type const Converting op input.341 : silu Adding op 'input.341' of type silu Converting op 1109 : constant Adding op '1109' of type const Converting op 1110 : constant Adding op '1110' of type const Converting op 1111 : constant Adding op '1111' of type const Converting op 1112 : constant Adding op '1112' of type const Converting op 1113 : constant Adding op '1113' of type const Converting op 1117 : listconstruct Adding op '1117' of type const Converting op 1118 : listconstruct Adding op '1118' of type const Converting op 1119 : listconstruct Adding op '1119' of type const Converting op 1120 : listconstruct Adding op '1120' of type const Converting op input.343 : _convolution Adding op 'input.343' of type conv Adding op 'input.343_pad_type_0' of type const Adding op 'input.343_pad0' of type const Converting op 1122 : silu Adding op '1122' of type silu Converting op 1123 : constant Adding op '1123' of type const Converting op 1124 : listconstruct Converting op input.345 : cat Adding op 'input.345' of type concat Adding op 'input.345_interleave_0' of type const Converting op 1126 : constant Adding op '1126' of type const Converting op 1127 : constant Adding op '1127' of type const Converting op 1128 : constant Adding op '1128' of type const Converting op 1129 : constant Adding op '1129' of type const Converting op 1137 : listconstruct Adding op '1137' of type const Converting op 1138 : listconstruct Adding op '1138' of type const Converting op 1139 : listconstruct Adding op '1139' of type const Converting op 1140 : listconstruct Adding op '1140' of type const Converting op input.347 : _convolution Adding op 'input.347' of type conv Adding op 'input.347_pad_type_0' of type const Adding op 'input.347_pad0' of type const Converting op input.349 : silu Adding op 'input.349' of type silu Converting op 1150 : listconstruct Adding op '1150' of type const Converting op 1151 : listconstruct Adding op '1151' of type const Converting op 1152 : listconstruct Adding op '1152' of type const Converting op 1153 : listconstruct Adding op '1153' of type const Converting op input.351 : _convolution Adding op 'input.351' of type conv Adding op 'input.351_pad_type_0' of type const Adding op 'input.351_pad0' of type const Converting op input.353 : silu Adding op 'input.353' of type silu Converting op 1159 : listconstruct Adding op '1159' of type const Converting op 1160 : listconstruct Adding op '1160' of type const Converting op 1161 : listconstruct Adding op '1161' of type const Converting op 1162 : listconstruct Adding op '1162' of type const Converting op input.355 : _convolution Adding op 'input.355' of type conv Adding op 'input.355_pad_type_0' of type const Adding op 'input.355_pad0' of type const Converting op input.357 : silu Adding op 'input.357' of type silu Converting op 1170 : listconstruct Adding op '1170' of type const Converting op 1171 : listconstruct Adding op '1171' of type const Converting op 1172 : listconstruct Adding op '1172' of type const Converting op 1173 : listconstruct Adding op '1173' of type const Converting op input.359 : _convolution Adding op 'input.359' of type conv Adding op 'input.359_pad_type_0' of type const Adding op 'input.359_pad0' of type const Converting op input.361 : silu Adding op 'input.361' of type silu Converting op 1179 : listconstruct Adding op '1179' of type const Converting op 1180 : listconstruct Adding op '1180' of type const Converting op 1181 : listconstruct Adding op '1181' of type const Converting op 1182 : listconstruct Adding op '1182' of type const Converting op input.363 : _convolution Adding op 'input.363' of type conv Adding op 'input.363_pad_type_0' of type const Adding op 'input.363_pad0' of type const Converting op 1184 : silu Adding op '1184' of type silu Converting op 1188 : listconstruct Adding op '1188' of type const Converting op 1189 : listconstruct Adding op '1189' of type const Converting op 1190 : listconstruct Adding op '1190' of type const Converting op 1191 : listconstruct Adding op '1191' of type const Converting op input.365 : _convolution Adding op 'input.365' of type conv Adding op 'input.365_pad_type_0' of type const Adding op 'input.365_pad0' of type const Converting op 1193 : silu Adding op '1193' of type silu Converting op 1194 : listconstruct Converting op input.367 : cat Adding op 'input.367' of type concat Adding op 'input.367_interleave_0' of type const Converting op 1199 : listconstruct Adding op '1199' of type const Converting op 1200 : listconstruct Adding op '1200' of type const Converting op 1201 : listconstruct Adding op '1201' of type const Converting op 1202 : listconstruct Adding op '1202' of type const Converting op input.369 : _convolution Adding op 'input.369' of type conv Adding op 'input.369_pad_type_0' of type const Adding op 'input.369_pad0' of type const Converting op input.371 : silu Adding op 'input.371' of type silu Converting op 1205 : constant Adding op '1205' of type const Converting op 1206 : constant Adding op '1206' of type const Converting op 1207 : constant Adding op '1207' of type const Converting op 1208 : constant Adding op '1208' of type const Converting op 1209 : constant Adding op '1209' of type const Converting op 1213 : listconstruct Adding op '1213' of type const Converting op 1214 : listconstruct Adding op '1214' of type const Converting op 1215 : listconstruct Adding op '1215' of type const Converting op 1216 : listconstruct Adding op '1216' of type const Converting op input.373 : _convolution Adding op 'input.373' of type conv Adding op 'input.373_pad_type_0' of type const Adding op 'input.373_pad0' of type const Converting op 1218 : silu Adding op '1218' of type silu Converting op 1219 : constant Adding op '1219' of type const Converting op 1220 : listconstruct Converting PyTorch Frontend ==> MIL Ops: 87% 761/873 [00:00<00:00, 1112.48 ops/s]Converting op input.375 : cat Adding op 'input.375' of type concat Adding op 'input.375_interleave_0' of type const Converting op 1222 : constant Adding op '1222' of type const Converting op 1223 : constant Adding op '1223' of type const Converting op 1224 : constant Adding op '1224' of type const Converting op 1225 : constant Adding op '1225' of type const Converting op 1233 : listconstruct Adding op '1233' of type const Converting op 1234 : listconstruct Adding op '1234' of type const Converting op 1235 : listconstruct Adding op '1235' of type const Converting op 1236 : listconstruct Adding op '1236' of type const Converting op input.377 : _convolution Adding op 'input.377' of type conv Adding op 'input.377_pad_type_0' of type const Adding op 'input.377_pad0' of type const Converting op input.379 : silu Adding op 'input.379' of type silu Converting op 1246 : listconstruct Adding op '1246' of type const Converting op 1247 : listconstruct Adding op '1247' of type const Converting op 1248 : listconstruct Adding op '1248' of type const Converting op 1249 : listconstruct Adding op '1249' of type const Converting op input.381 : _convolution Adding op 'input.381' of type conv Adding op 'input.381_pad_type_0' of type const Adding op 'input.381_pad0' of type const Converting op input.383 : silu Adding op 'input.383' of type silu Converting op 1255 : listconstruct Adding op '1255' of type const Converting op 1256 : listconstruct Adding op '1256' of type const Converting op 1257 : listconstruct Adding op '1257' of type const Converting op 1258 : listconstruct Adding op '1258' of type const Converting op input.385 : _convolution Adding op 'input.385' of type conv Adding op 'input.385_pad_type_0' of type const Adding op 'input.385_pad0' of type const Converting op input.387 : silu Adding op 'input.387' of type silu Converting op 1266 : listconstruct Adding op '1266' of type const Converting op 1267 : listconstruct Adding op '1267' of type const Converting op 1268 : listconstruct Adding op '1268' of type const Converting op 1269 : listconstruct Adding op '1269' of type const Converting op input.389 : _convolution Adding op 'input.389' of type conv Adding op 'input.389_pad_type_0' of type const Adding op 'input.389_pad0' of type const Converting op input.391 : silu Adding op 'input.391' of type silu Converting op 1275 : listconstruct Adding op '1275' of type const Converting op 1276 : listconstruct Adding op '1276' of type const Converting op 1277 : listconstruct Adding op '1277' of type const Converting op 1278 : listconstruct Adding op '1278' of type const Converting op input.393 : _convolution Adding op 'input.393' of type conv Adding op 'input.393_pad_type_0' of type const Adding op 'input.393_pad0' of type const Converting op 1280 : silu Adding op '1280' of type silu Converting op 1284 : listconstruct Adding op '1284' of type const Converting op 1285 : listconstruct Adding op '1285' of type const Converting op 1286 : listconstruct Adding op '1286' of type const Converting op 1287 : listconstruct Adding op '1287' of type const Converting op input.395 : _convolution Adding op 'input.395' of type conv Adding op 'input.395_pad_type_0' of type const Adding op 'input.395_pad0' of type const Converting op 1289 : silu Adding op '1289' of type silu Converting op 1290 : listconstruct Converting op input.397 : cat Adding op 'input.397' of type concat Adding op 'input.397_interleave_0' of type const Converting op 1295 : listconstruct Adding op '1295' of type const Converting op 1296 : listconstruct Adding op '1296' of type const Converting op 1297 : listconstruct Adding op '1297' of type const Converting op 1298 : listconstruct Adding op '1298' of type const Converting op input.399 : _convolution Adding op 'input.399' of type conv Adding op 'input.399_pad_type_0' of type const Adding op 'input.399_pad0' of type const Converting op input : silu Adding op 'input' of type silu Converting op 1301 : constant Adding op '1301' of type const Converting op 1302 : constant Adding op '1302' of type const Converting op 1303 : constant Adding op '1303' of type const Converting op 1304 : constant Adding op '1304' of type const Converting op 1305 : constant Adding op '1305' of type const Converting op 1306 : constant Adding op '1306' of type const Converting op 1307 : constant Adding op '1307' of type const Converting op 1308 : constant Adding op '1308' of type const Converting op 1319 : listconstruct Adding op '1319' of type const Converting op 1320 : listconstruct Adding op '1320' of type const Converting op 1321 : listconstruct Adding op '1321' of type const Converting op 1322 : listconstruct Adding op '1322' of type const Converting op 1323 : _convolution Adding op '1323' of type conv Adding op '1323_pad_type_0' of type const Adding op '1323_pad_0' of type const Converting op 1324 : size Adding op '1324_shape' of type shape Adding op 'const_0' of type const Converting op 1325 : size Adding op '1325_shape' of type shape Adding op 'const_1' of type const Converting op 1326 : size Adding op '1326_shape' of type shape Adding op 'const_2' of type const Converting op 1327 : listconstruct Adding op '1327' of type const Converting op 1328 : view Adding op 'cast_0' of type cast Adding op 'cast_0_dtype_0' of type const Adding op '1328' of type reshape Converting op 1329 : listconstruct Adding op '1329' of type const Converting op 1330 : permute Adding op '1330' of type transpose Converting op 1331 : contiguous Setting pytorch op: %1331 = contiguous[](%1330, %1306) to no-op. Converting op 1334 : listconstruct Adding op '1334' of type const Converting op 1335 : listconstruct Adding op '1335' of type const Converting op 1336 : listconstruct Adding op '1336' of type const Converting op 1337 : listconstruct Adding op '1337' of type const Converting op 1338 : _convolution Adding op '1338' of type conv Adding op '1338_pad_type_0' of type const Adding op '1338_pad_0' of type const Converting op 1339 : size Adding op '1339_shape' of type shape Adding op 'const_3' of type const Converting op 1340 : size Adding op '1340_shape' of type shape Adding op 'const_4' of type const Converting op 1341 : size Adding op '1341_shape' of type shape Adding op 'const_5' of type const Converting op 1342 : listconstruct Adding op '1342' of type const Converting op 1343 : view Adding op 'cast_1' of type cast Adding op 'cast_1_dtype_0' of type const Adding op '1343' of type reshape Converting op 1344 : listconstruct Adding op '1344' of type const Converting op 1345 : permute Adding op '1345' of type transpose Converting op 1346 : contiguous Setting pytorch op: %1346 = contiguous[](%1345, %1306) to no-op. Converting op 1349 : listconstruct Adding op '1349' of type const Converting op 1350 : listconstruct Adding op '1350' of type const Converting op 1351 : listconstruct Adding op '1351' of type const Converting op 1352 : listconstruct Adding op '1352' of type const Converting op 1353 : _convolution Adding op '1353' of type conv Adding op '1353_pad_type_0' of type const Adding op '1353_pad_0' of type const Converting op 1354 : size Adding op '1354_shape' of type shape Adding op 'const_6' of type const Converting op 1355 : size Adding op '1355_shape' of type shape Adding op 'const_7' of type const Converting op 1356 : size Adding op '1356_shape' of type shape Adding op 'const_8' of type const Converting op 1357 : listconstruct Adding op '1357' of type const Converting op 1358 : view Adding op 'cast_2' of type cast Adding op 'cast_2_dtype_0' of type const Adding op '1358' of type reshape Converting op 1359 : listconstruct Adding op '1359' of type const Converting op 1360 : permute Adding op '1360' of type transpose Converting op 1361 : contiguous Setting pytorch op: %1361 = contiguous[](%1360, %1306) to no-op. Converting op 1364 : listconstruct Adding op '1364' of type const Converting op 1365 : listconstruct Adding op '1365' of type const Converting op 1366 : listconstruct Adding op '1366' of type const Converting op 1367 : listconstruct Adding op '1367' of type const Converting op 1368 : _convolution Adding op '1368' of type conv Adding op '1368_pad_type_0' of type const Adding op '1368_pad_0' of type const Converting op 1369 : size Adding op '1369_shape' of type shape Adding op 'const_9' of type const Converting op 1370 : size Adding op '1370_shape' of type shape Adding op 'const_10' of type const Converting op 1371 : size Adding op '1371_shape' of type shape Adding op 'const_11' of type const Converting op 1372 : listconstruct Adding op '1372' of type const Converting op 1373 : view Adding op 'cast_3' of type cast Adding op 'cast_3_dtype_0' of type const Adding op '1373' of type reshape Converting op 1374 : listconstruct Adding op '1374' of type const Converting op 1375 : permute Adding op '1375' of type transpose Converting op 1376 : contiguous Setting pytorch op: %1376 = contiguous[](%1375, %1306) to no-op. Converting op 1377 : tupleconstruct Converting op 1378 : tupleunpack Converting PyTorch Frontend ==> MIL Ops: 100% 871/873 [00:00<00:00, 871.64 ops/s] Performing passes for torch frontend: "common::dead_code_elimination" Removing op "const_11" (type: const) Removing op "1371_shape" (type: shape) Removing op "const_10" (type: const) Removing op "1370_shape" (type: shape) Removing op "const_9" (type: const) Removing op "1369_shape" (type: shape) Removing op "1367" (type: const) Removing op "1365" (type: const) Removing op "const_8" (type: const) Removing op "1356_shape" (type: shape) Removing op "const_7" (type: const) Removing op "1355_shape" (type: shape) Removing op "const_6" (type: const) Removing op "1354_shape" (type: shape) Removing op "1352" (type: const) Removing op "1350" (type: const) Removing op "const_5" (type: const) Removing op "1341_shape" (type: shape) Removing op "const_4" (type: const) Removing op "1340_shape" (type: shape) Removing op "const_3" (type: const) Removing op "1339_shape" (type: shape) Removing op "1337" (type: const) Removing op "1335" (type: const) Removing op "const_2" (type: const) Removing op "1326_shape" (type: shape) Removing op "const_1" (type: const) Removing op "1325_shape" (type: shape) Removing op "const_0" (type: const) Removing op "1324_shape" (type: shape) Removing op "1322" (type: const) Removing op "1320" (type: const) Removing op "1308" (type: const) Removing op "1307" (type: const) Removing op "1306" (type: const) Removing op "1304" (type: const) Removing op "1303" (type: const) Removing op "1302" (type: const) Removing op "1301" (type: const) Removing op "1298" (type: const) Removing op "1296" (type: const) Removing op "1287" (type: const) Removing op "1285" (type: const) Removing op "1278" (type: const) Removing op "1276" (type: const) Removing op "1269" (type: const) Removing op "1267" (type: const) Removing op "1258" (type: const) Removing op "1256" (type: const) Removing op "1249" (type: const) Removing op "1247" (type: const) Removing op "1236" (type: const) Removing op "1234" (type: const) Removing op "1224" (type: const) Removing op "1223" (type: const) Removing op "1222" (type: const) Removing op "1216" (type: const) Removing op "1214" (type: const) Removing op "1209" (type: const) Removing op "1208" (type: const) Removing op "1207" (type: const) Removing op "1205" (type: const) Removing op "1202" (type: const) Removing op "1200" (type: const) Removing op "1191" (type: const) Removing op "1189" (type: const) Removing op "1182" (type: const) Removing op "1180" (type: const) Removing op "1173" (type: const) Removing op "1171" (type: const) Removing op "1162" (type: const) Removing op "1160" (type: const) Removing op "1153" (type: const) Removing op "1151" (type: const) Removing op "1140" (type: const) Removing op "1138" (type: const) Removing op "1128" (type: const) Removing op "1127" (type: const) Removing op "1126" (type: const) Removing op "1120" (type: const) Removing op "1118" (type: const) Removing op "1113" (type: const) Removing op "1112" (type: const) Removing op "1111" (type: const) Removing op "1109" (type: const) Removing op "1106" (type: const) Removing op "1104" (type: const) Removing op "1095" (type: const) Removing op "1093" (type: const) Removing op "1086" (type: const) Removing op "1084" (type: const) Removing op "1077" (type: const) Removing op "1075" (type: const) Removing op "1066" (type: const) Removing op "1064" (type: const) Removing op "1057" (type: const) Removing op "1055" (type: const) Removing op "1044" (type: const) Removing op "1042" (type: const) Removing op "1032" (type: const) Removing op "1031" (type: const) Removing op "1030" (type: const) Removing op "1024" (type: const) Removing op "1022" (type: const) Removing op "1017" (type: const) Removing op "1016" (type: const) Removing op "1015" (type: const) Removing op "1013" (type: const) Removing op "1010" (type: const) Removing op "1008" (type: const) Removing op "999" (type: const) Removing op "997" (type: const) Removing op "990" (type: const) Removing op "988" (type: const) Removing op "981" (type: const) Removing op "979" (type: const) Removing op "970" (type: const) Removing op "968" (type: const) Removing op "961" (type: const) Removing op "959" (type: const) Removing op "948" (type: const) Removing op "946" (type: const) Removing op "936" (type: const) Removing op "935" (type: const) Removing op "934" (type: const) Removing op "929" (type: const) Removing op "927" (type: const) Removing op "924" (type: const) Removing op "922" (type: const) Removing op "917" (type: const) Removing op "916" (type: const) Removing op "915" (type: const) Removing op "911" (type: const) Removing op "909" (type: const) Removing op "900" (type: const) Removing op "898" (type: const) Removing op "891" (type: const) Removing op "889" (type: const) Removing op "882" (type: const) Removing op "880" (type: const) Removing op "871" (type: const) Removing op "869" (type: const) Removing op "862" (type: const) Removing op "860" (type: const) Removing op "849" (type: const) Removing op "847" (type: const) Removing op "837" (type: const) Removing op "836" (type: const) Removing op "835" (type: const) Removing op "830" (type: const) Removing op "828" (type: const) Removing op "825" (type: const) Removing op "823" (type: const) Removing op "818" (type: const) Removing op "817" (type: const) Removing op "816" (type: const) Removing op "812" (type: const) Removing op "810" (type: const) Removing op "801" (type: const) Removing op "799" (type: const) Removing op "792" (type: const) Removing op "790" (type: const) Removing op "783" (type: const) Removing op "781" (type: const) Removing op "772" (type: const) Removing op "770" (type: const) Removing op "763" (type: const) Removing op "761" (type: const) Removing op "750" (type: const) Removing op "748" (type: const) Removing op "738" (type: const) Removing op "737" (type: const) Removing op "736" (type: const) Removing op "731" (type: const) Removing op "729" (type: const) Removing op "726" (type: const) Removing op "724" (type: const) Removing op "719" (type: const) Removing op "718" (type: const) Removing op "717" (type: const) Removing op "713" (type: const) Removing op "711" (type: const) Removing op "703" (type: const) Removing op "702" (type: const) Removing op "698" (type: const) Removing op "697" (type: const) Removing op "693" (type: const) Removing op "692" (type: const) Removing op "687" (type: const) Removing op "685" (type: const) Removing op "677" (type: const) Removing op "676" (type: const) Removing op "675" (type: const) Removing op "674" (type: const) Removing op "673" (type: const) Removing op "670" (type: const) Removing op "668" (type: const) Removing op "659" (type: const) Removing op "657" (type: const) Removing op "649" (type: const) Removing op "647" (type: const) Removing op "640" (type: const) Removing op "638" (type: const) Removing op "628" (type: const) Removing op "626" (type: const) Removing op "619" (type: const) Removing op "617" (type: const) Removing op "606" (type: const) Removing op "604" (type: const) Removing op "594" (type: const) Removing op "593" (type: const) Removing op "592" (type: const) Removing op "589" (type: const) Removing op "587" (type: const) Removing op "582" (type: const) Removing op "581" (type: const) Removing op "580" (type: const) Removing op "578" (type: const) Removing op "575" (type: const) Removing op "573" (type: const) Removing op "564" (type: const) Removing op "562" (type: const) Removing op "554" (type: const) Removing op "552" (type: const) Removing op "545" (type: const) Removing op "543" (type: const) Removing op "533" (type: const) Removing op "531" (type: const) Removing op "524" (type: const) Removing op "522" (type: const) Removing op "511" (type: const) Removing op "509" (type: const) Removing op "499" (type: const) Removing op "498" (type: const) Removing op "497" (type: const) Removing op "494" (type: const) Removing op "492" (type: const) Removing op "487" (type: const) Removing op "486" (type: const) Removing op "485" (type: const) Removing op "483" (type: const) Removing op "480" (type: const) Removing op "478" (type: const) Removing op "469" (type: const) Removing op "467" (type: const) Removing op "459" (type: const) Removing op "457" (type: const) Removing op "450" (type: const) Removing op "448" (type: const) Removing op "438" (type: const) Removing op "436" (type: const) Removing op "429" (type: const) Removing op "427" (type: const) Removing op "417" (type: const) Removing op "415" (type: const) Removing op "408" (type: const) Removing op "406" (type: const) Removing op "396" (type: const) Removing op "394" (type: const) Removing op "387" (type: const) Removing op "385" (type: const) Removing op "375" (type: const) Removing op "373" (type: const) Removing op "366" (type: const) Removing op "364" (type: const) Removing op "354" (type: const) Removing op "352" (type: const) Removing op "345" (type: const) Removing op "343" (type: const) Removing op "328" (type: const) Removing op "326" (type: const) Removing op "316" (type: const) Removing op "315" (type: const) Removing op "314" (type: const) Removing op "311" (type: const) Removing op "309" (type: const) Removing op "304" (type: const) Removing op "303" (type: const) Removing op "302" (type: const) Removing op "300" (type: const) Removing op "297" (type: const) Removing op "295" (type: const) Removing op "286" (type: const) Removing op "284" (type: const) Removing op "276" (type: const) Removing op "274" (type: const) Removing op "267" (type: const) Removing op "265" (type: const) Removing op "255" (type: const) Removing op "253" (type: const) Removing op "246" (type: const) Removing op "244" (type: const) Removing op "234" (type: const) Removing op "232" (type: const) Removing op "225" (type: const) Removing op "223" (type: const) Removing op "213" (type: const) Removing op "211" (type: const) Removing op "204" (type: const) Removing op "202" (type: const) Removing op "189" (type: const) Removing op "187" (type: const) Removing op "177" (type: const) Removing op "176" (type: const) Removing op "175" (type: const) Removing op "172" (type: const) Removing op "170" (type: const) Removing op "165" (type: const) Removing op "164" (type: const) Removing op "163" (type: const) Removing op "161" (type: const) Removing op "158" (type: const) Removing op "156" (type: const) Removing op "147" (type: const) Removing op "145" (type: const) Removing op "137" (type: const) Removing op "135" (type: const) Removing op "128" (type: const) Removing op "126" (type: const) Removing op "116" (type: const) Removing op "114" (type: const) Removing op "107" (type: const) Removing op "105" (type: const) Removing op "94" (type: const) Removing op "92" (type: const) Removing op "82" (type: const) Removing op "81" (type: const) Removing op "80" (type: const) Removing op "77" (type: const) Removing op "75" (type: const) Removing op "70" (type: const) Removing op "69" (type: const) Removing op "68" (type: const) Removing op "66" (type: const) Removing op "63" (type: const) Removing op "61" (type: const) Removing op "56" (type: const) Removing op "55" (type: const) Removing op "54" (type: const) Removing op "52" (type: const) Performing passes for torch frontend: "common::loop_invariant_elimination" Performing passes for torch frontend: "common::dead_code_elimination" Performing passes for torch frontend: "torch::torch_upsample_to_core_upsample" Performing passes for torch frontend: "torch::torch_tensor_assign_to_core" Running MIL Common passes: 0% 0/38 [00:00<?, ? passes/s]Performing pass: "common::update_output_dtypes" Performing pass: "common::cast_optimization" Performing pass: "common::const_elimination" Performing pass: "common::sanitize_input_output_names" /usr/local/lib/python3.7/dist-packages/coremltools/converters/mil/mil/passes/name_sanitization_utils.py:129: UserWarning: Output, '1330', of the source model, has been renamed to 'var_1330' in the Core ML model. warnings.warn(msg.format(var.name, new_name)) /usr/local/lib/python3.7/dist-packages/coremltools/converters/mil/mil/passes/name_sanitization_utils.py:129: UserWarning: Output, '1345', of the source model, has been renamed to 'var_1345' in the Core ML model. warnings.warn(msg.format(var.name, new_name)) /usr/local/lib/python3.7/dist-packages/coremltools/converters/mil/mil/passes/name_sanitization_utils.py:129: UserWarning: Output, '1360', of the source model, has been renamed to 'var_1360' in the Core ML model. warnings.warn(msg.format(var.name, new_name)) /usr/local/lib/python3.7/dist-packages/coremltools/converters/mil/mil/passes/name_sanitization_utils.py:129: UserWarning: Output, '1375', of the source model, has been renamed to 'var_1375' in the Core ML model. warnings.warn(msg.format(var.name, new_name)) Performing pass: "common::divide_to_multiply" Performing pass: "common::add_conv_transpose_output_shape" Performing pass: "common::const_elimination" Performing pass: "common::loop_invariant_elimination" Performing pass: "common::remove_symbolic_reshape" remove_symbolic_reshape: changed 0 reshapes. Performing pass: "common::noop_elimination" Performing pass: "common::fuse_matmul_weight_bias" Performing pass: "common::fuse_linear_bias" Performing pass: "common::fuse_gelu_tanh_approximation" Adding op 'pow' of type pow Adding op 'pow_y_0' of type const Adding op 'mul_1' of type mul Adding op 'mul_1_x_0' of type const Adding op 'add' of type add Adding op 'mul_2' of type mul Adding op 'mul_2_x_0' of type const Adding op 'tanh' of type tanh Adding op 'add_1' of type add Adding op 'add_1_x_0' of type const Adding op 'mul' of type mul Adding op 'mul_x_0' of type const Adding op 'mul_3' of type mul Adding op 'pow' of type pow Adding op 'pow_y_1' of type const Adding op 'mul_1' of type mul Adding op 'mul_1_x_1' of type const Adding op 'add' of type add Adding op 'mul_2' of type mul Adding op 'mul_2_x_1' of type const Adding op 'tanh' of type tanh Adding op 'add_1' of type add Adding op 'add_1_x_1' of type const Adding op 'mul' of type mul Adding op 'mul_x_1' of type const Adding op 'mul_3' of type mul Performing pass: "common::fuse_gelu_exact" Performing pass: "common::fuse_leaky_relu" Performing pass: "common::rank0_expand_dims_swap" Performing pass: "common::use_reflection_padding" Performing pass: "common::merge_consecutive_paddings" Performing pass: "common::pad_conv_connect" Running MIL Common passes: 50% 19/38 [00:00<00:00, 182.55 passes/s]Performing pass: "common::image_input_preprocess" Performing pass: "common::replace_stack_reshape" Performing pass: "common::reduce_transposes" Adding op 'identity_0' of type identity Adding op 'identity_1' of type identity Adding op 'identity_2' of type identity Adding op 'identity_3' of type identity Performing pass: "common::fuse_conv_scale" Performing pass: "common::fuse_conv_bias" Performing pass: "common::fuse_onehot_matmul_to_gather" Performing pass: "common::fuse_layernorm_or_instancenorm" Performing pass: "common::fuse_elementwise_to_batchnorm" Performing pass: "common::fuse_reduce_mean" Performing pass: "common::fuse_conv_batchnorm" Performing pass: "common::fuse_conv_scale" Performing pass: "common::fuse_conv_bias" Performing pass: "common::fuse_conv_batchnorm" Performing pass: "common::detect_concat_interleave" Performing pass: "common::concat_to_pixel_shuffle" Performing pass: "common::fuse_prelu" Adding op 'neg' of type mul Adding op 'neg_y_0' of type const Adding op 'relu1' of type relu Adding op 'alpha_mul' of type mul Adding op 'alpha_mul_y_0' of type const Adding op 'relu2' of type relu Adding op 'out_op' of type add Adding op 'transpose' of type transpose Adding op 'transpose_perm_0' of type const Adding op 'neg' of type mul Adding op 'neg_y_1' of type const Adding op 'relu1' of type relu Adding op 'alpha_mul' of type mul Adding op 'alpha_mul_y_1' of type const Adding op 'relu2' of type relu Adding op 'out_op' of type add Performing pass: "common::prelu_to_lrelu" Performing pass: "common::remove_redundant_ops" Performing pass: "common::dead_code_elimination" Removing op "cast_3_dtype_0" (type: const) Removing op "cast_2_dtype_0" (type: const) Removing op "cast_1_dtype_0" (type: const) Removing op "cast_0_dtype_0" (type: const) Running MIL Common passes: 100% 38/38 [00:00<00:00, 65.86 passes/s] Running MIL Clean up passes: 0% 0/11 [00:00<?, ? passes/s]Performing pass: "common::dead_code_elimination" Performing pass: "common::const_elimination" Performing pass: "common::cast_optimization" Performing pass: "common::const_elimination" Performing pass: "common::loop_invariant_elimination" Performing pass: "common::noop_elimination" Performing pass: "common::dedup_op_and_var_names" Performing pass: "common::reduce_transposes" Adding op 'identity_4' of type identity Adding op 'identity_5' of type identity Adding op 'identity_6' of type identity Adding op 'identity_7' of type identity Running MIL Clean up passes: 73% 8/11 [00:00<00:00, 25.60 passes/s]Performing pass: "common::remove_redundant_ops" Performing pass: "common::topological_reorder" Adding op 'transpose_0' of type transpose Adding op 'transpose_1' of type transpose Adding op 'transpose_2' of type transpose Adding op 'transpose_3' of type transpose Performing pass: "common::dead_code_elimination" Running MIL Clean up passes: 100% 11/11 [00:00<00:00, 30.06 passes/s] Performing passes for nn_backend: "nn_backend::commingle_loop_vars" Performing passes for nn_backend: "nn_backend::handle_return_inputs_as_outputs" Performing passes for nn_backend: "common::const_elimination" Performing passes for nn_backend: "common::dead_code_elimination" Performing passes for nn_backend: "nn_backend::handle_unused_inputs" Performing passes for nn_backend: "nn_backend::alert_return_type_cast" Translating MIL ==> NeuralNetwork Ops: 100% 978/978 [00:03<00:00, 275.24 ops/s] CoreML: export success ✅ 37.7s, saved as v418e-best.mlmodel (134.6 MB)

Export complete (47.7s) Results saved to /content/yolov5 Detect: python detect.py --weights v418e-best.mlmodel Validate: python val.py --weights v418e-best.mlmodel PyTorch Hub: model = torch.hub.load('ultralytics/yolov5', 'custom', 'v418e-best.mlmodel') Vislize: https://netron.app/

junmcenroe commented 1 year ago

Dear @Mercury-ML

I checked again, but I cannot reproduce this failure. Other possible failure might be the different image size while training. Did you get the your best.pt with following? python train.py --data counter.yaml --weights yolov5m.pt --img 1280

featureMapDimensions error will occur when found the different model size because featureMapDimensions is defined as followings. ** parser.add_argument('-size','--imgSize', type=int, dest="imgSize", default=640, help='trained imageSize 640, 960,.. by yolov5 model') opt = parser.parse_args()

inputSize

imgSize = opt.imgSize

strides = [8, 16, 32] featureMapDimensions = [imgSize // stride for stride in strides] **

Could you check your training again?

Mercury-ML commented 1 year ago

Thanks again for working on this!!!

Here's the exact command I used for training:

python train.py \
  --data custom_dataset_wandb.yaml \
  --epochs 18 \
  --img 1280 \
  --hyp /content/yolov5/data/hyps/hyp.v4-100k.yaml \
  --batch-size 12 \
  --project v3-big-data \
  --bbox_interval 1 \
  --save-period 1 \
  --weights yolov5m6.pt 

(was logging with W&B, highly recommended!) As you can see, the flag --img 1280 was set.

Would you recommend I retrain the model? I ran this in early September

junmcenroe commented 1 year ago

Dear @Mercury-ML

I probably found the reason. My converter is just for YOLOv5-P5 model, such as yolov5m.pt <YOLOv5-P5 models : 3 output layers P3, P4, P5 at strides 8, 16, 32>

But your trained model is using YOLOv5-P6 model, such as yolov5m6.pt <YOLOv5-P6 models: 4 output layers P3, P4, P5, P6 at strides 8, 16, 32, 64>

These models have quite different output layer and strides. So my converter cannot handle as of today. Even if YOLOv5-P5 model, I can increase input image size from 640 to 1280 and can increase accuracy for my use case, so I build this converter with some reference.

I am not sure I can update to handle P6 model or not. need to time to study. Sorry.

Mercury-ML commented 1 year ago

Thank you for looking into this further and figuring out the issue! I will keep an eye out for an update for p6. I have an older model I converted using: https://github.com/mshamash/yolov5 that I can work with for the time being (mshamash repo is out of data and I don't think it will convert any longer). Thanks again! The work you've done so far is brilliant!

junmcenroe commented 1 year ago

Dear @Mercury-ML

I uploaded the YOLOv5P6CoreMLConverter which can handle the P6 model 1280 image size. Result shows as followings One restriction is anchor data is only for P6 1280 image size. If you train P6 with 640 image size, different anchor data required. so we need to modify the scripts. mlmodel file size is about 140MB.
yolov5-P6-1280

Mercury-ML commented 1 year ago

Thank you! This project is 1280 only. I will run this later today and let you know!

Mercury-ML commented 1 year ago

Nice job! You did it!!! Everything seems to be in order: Combine CoreMl model with nms and export model CoreML export success, saved as /content/v4e18-NMS.mlmodel

I've tested the model out in Xcode and it's working well!

One question: is it possible to change the IOU and Confidence threshold values inside Xcode? I noticed there's a note saying can be overwritten when using the model (YOLOv5P6CoreMLConverter.py line 203), and on the predictions page it has "double" but no actual value I can change (I can change the description).

In the older models I was using, there's additional metadata (under general / metadata) and if you convert the model in Xcode from .mlmodel to .mlpackage you can then change the IOU and Confidence thresholds inside of Xcode. Here's a screenshot of the old model additional metadata.

additional-metadata

Thanks again for everything!

junmcenroe commented 1 year ago

Dear @Mercury-ML

The IOU and Confidence threshold values are not changed in Xcode. These value should be defined in convert process, accuracy during NMS. Currently defined in line 203. nms.iouThreshold = 0.6 nms.confidenceThreshold = 0.4 These value are slightly different from yolov5 original repo's detect.py Original is conf_thres=0.25, # confidence threshold iou_thres=0.45, # NMS IOU threshold

So converted model might be slightly tight to detect as higher iou_thres, and higher conf_thres. you can change this value as you want in line 204-205, and put this value in metadata in line 238, 240. Currently this metadata is shown in Prediction tab. After modifying these value, metadata info also changed in Xcode.

Mercury-ML commented 1 year ago

Perfect, I will adjust it on conversion. Thanks again!