Tutorial: Inspecting Model Architectures#

The visualization module provides functionalities to visualize neural network models and calculate their structural properties.

Computational Graph#

draw_graph#

The draw_graph function generates a visual representation of a PyTorch model’s computational graph. This visualization includes modules, module hierarchies, torch functions, and tensor shapes captured during forward propagation.

Usage#

[34]:
from vml_sdk.apis import draw_model_graph
from IPython.display import SVG, display


graph = draw_model_graph('atss_r50_fpn_1x_coco', "detection" , filter_by="RMS", depth=2)

display(SVG(graph))


07/09 16:50:26 - mmengine - INFO - model_name: mask-rcnn_r50_fpn_albu-1x_coco
model_name: atss_r50_fpn_1x_coco
model_name: atss_r101_fpn_1x_coco
model_name: autoassign_r50-caffe_fpn_1x_coco
model_name: boxinst_r50_fpn_ms-90k_coco
model_name: boxinst_r101_fpn_ms-90k_coco
model_name: faster-rcnn_r50_fpn_carafe_1x_coco
model_name: mask-rcnn_r50_fpn_carafe_1x_coco
model_name: cascade-rcnn_r50-caffe_fpn_1x_coco
model_name: cascade-rcnn_r50_fpn_1x_coco
model_name: cascade-rcnn_r50_fpn_20e_coco
model_name: cascade-rcnn_r101-caffe_fpn_1x_coco
model_name: cascade-rcnn_r101_fpn_1x_coco
model_name: cascade-rcnn_r101_fpn_20e_coco
model_name: cascade-rcnn_x101-32x4d_fpn_1x_coco
model_name: cascade-rcnn_x101-32x4d_fpn_20e_coco
model_name: cascade-rcnn_x101-64x4d_fpn_1x_coco
model_name: cascade-rcnn_x101_64x4d_fpn_20e_coco
model_name: cascade-mask-rcnn_r50-caffe_fpn_1x_coco
model_name: cascade-mask-rcnn_r50_fpn_1x_coco
model_name: cascade-mask-rcnn_r50_fpn_20e_coco
model_name: cascade-mask-rcnn_r101-caffe_fpn_1x_coco
model_name: cascade-mask-rcnn_r101_fpn_1x_coco
model_name: cascade-mask-rcnn_r101_fpn_20e_coco
model_name: cascade-mask-rcnn_x101-32x4d_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d_fpn_20e_coco
model_name: cascade-mask-rcnn_x101-64x4d_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-64x4d_fpn_20e_coco
model_name: cascade-mask-rcnn_r50-caffe_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_r50_fpn_mstrain_3x_coco
model_name: cascade-mask-rcnn_r101-caffe_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_r101_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_x101-32x4d_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_x101-32x8d_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_x101-64x4d_fpn_ms-3x_coco
model_name: cascade-rpn_fast-rcnn_r50-caffe_fpn_1x_coco
model_name: cascade-rpn_faster-rcnn_r50-caffe_fpn_1x_coco
model_name: centernet_r18-dcnv2_8xb16-crop512-140e_coco
model_name: centernet_r18_8xb16-crop512-140e_coco
model_name: centernet-update_r50-caffe_fpn_ms-1x_coco
model_name: centripetalnet_hourglass104_16xb6-crop511-210e-mstest_coco
model_name: condinst_r50_fpn_ms-poly-90k_coco_instance
model_name: conditional-detr_r50_8xb2-50e_coco
model_name: cornernet_hourglass104_10xb5-crop511-210e-mstest_coco
model_name: cornernet_hourglass104_8xb6-210e-mstest_coco
model_name: cornernet_hourglass104_32xb3-210e-mstest_coco
model_name: mask-rcnn_convnext-t-p4-w7_fpn_amp-ms-crop-3x_coco
model_name: cascade-mask-rcnn_convnext-t-p4-w7_fpn_4conv1fc-giou_amp-ms-crop-3x_coco
model_name: cascade-mask-rcnn_convnext-s-p4-w7_fpn_4conv1fc-giou_amp-ms-crop-3x_coco
model_name: crowddet-rcnn_refine_r50_fpn_8xb2-30e_crowdhuman
model_name: crowddet-rcnn_r50_fpn_8xb2-30e_crowdhuman
model_name: dab-detr_r50_8xb2-50e_coco
model_name: faster-rcnn_r50_fpn_dconv_c3-c5_1x_coco
model_name: faster-rcnn_r50_fpn_dpool_1x_coco
model_name: faster-rcnn_r101-dconv-c3-c5_fpn_1x_coco
model_name: faster-rcnn_x101-32x4d-dconv-c3-c5_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_dconv_c3-c5_1x_coco
model_name: mask-rcnn_r50_fpn_fp16_dconv_c3-c5_1x_coco
model_name: mask-rcnn_r101-dconv-c3-c5_fpn_1x_coco
model_name: cascade-rcnn_r50_fpn_dconv_c3-c5_1x_coco
model_name: cascade-rcnn_r101-dconv-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_r50_fpn_dconv_c3-c5_1x_coco
model_name: cascade-mask-rcnn_r101-dconv-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-dconv-c3-c5_fpn_1x_coco
model_name: faster-rcnn_r50_fpn_mdconv_c3-c5_1x_coco
model_name: faster-rcnn_r50_fpn_mdconv_c3-c5_group4_1x_coco
model_name: faster-rcnn_r50_fpn_mdpool_1x_coco
model_name: mask-rcnn_r50_fpn_mdconv_c3-c5_1x_coco
model_name: mask-rcnn_r50_fpn_fp16_mdconv_c3-c5_1x_coco
model_name: ddod_r50_fpn_1x_coco
model_name: deformable-detr_r50_16xb2-50e_coco
model_name: deformable-detr-refine_r50_16xb2-50e_coco
model_name: deformable-detr-refine-twostage_r50_16xb2-50e_coco
model_name: cascade-rcnn_r50-rfp_1x_coco
model_name: cascade-rcnn_r50-sac_1x_coco
model_name: detectors_cascade-rcnn_r50_1x_coco
model_name: htc_r50-rfp_1x_coco
model_name: htc_r50-sac_1x_coco
model_name: detectors_htc-r50_1x_coco
model_name: detr_r50_8xb2-150e_coco
model_name: dino-4scale_r50_8xb2-12e_coco
model_name: dino-4scale_r50_8xb2-24e_coco
model_name: dino-4scale_r50_8xb2-36e_coco
model_name: dino-5scale_swin-l_8xb2-12e_coco
model_name: dino-5scale_swin-l_8xb2-36e_coco
model_name: dino-4scale_r50_improved_8xb2-12e_coco
model_name: dh-faster-rcnn_r50_fpn_1x_coco
model_name: atss_r50-caffe_fpn_dyhead_1x_coco
model_name: atss_r50_fpn_dyhead_1x_coco
model_name: atss_swin-l-p4-w12_fpn_dyhead_ms-2x_coco
model_name: dynamic-rcnn_r50_fpn_1x_coco
model_name: retinanet_effb3_fpn_8xb4-crop896-1x_coco
model_name: faster-rcnn_r50_fpn_attention_1111_1x_coco
model_name: faster-rcnn_r50_fpn_attention_0010_1x_coco
model_name: faster-rcnn_r50_fpn_attention_1111_dcn_1x_coco
model_name: faster-rcnn_r50_fpn_attention_0010_dcn_1x_coco
model_name: faster-rcnn_r50-caffe-c4_1x_coco
model_name: faster-rcnn_r50-caffe-c4_mstrain_1x_coco
model_name: faster-rcnn_r50-caffe-dc5_1x_coco
model_name: faster-rcnn_r50-caffe_fpn_1x_coco
model_name: faster-rcnn_r50_fpn_1x_coco
model_name: faster-rcnn_r50_fpn_fp16_1x_coco
model_name: faster-rcnn_r50_fpn_2x_coco
model_name: faster-rcnn_r101-caffe_fpn_1x_coco
model_name: faster-rcnn_r101_fpn_1x_coco
model_name: faster-rcnn_r101_fpn_2x_coco
model_name: faster-rcnn_x101-32x4d_fpn_1x_coco
model_name: faster-rcnn_x101-32x4d_fpn_2x_coco
model_name: faster-rcnn_x101-64x4d_fpn_1x_coco
model_name: faster-rcnn_x101-64x4d_fpn_2x_coco
model_name: faster-rcnn_r50_fpn_iou_1x_coco
model_name: faster-rcnn_r50_fpn_giou_1x_coco
model_name: faster-rcnn_r50_fpn_bounded_iou_1x_coco
model_name: faster-rcnn_r50-caffe-dc5_mstrain_1x_coco
model_name: faster-rcnn_r50-caffe-dc5_mstrain_3x_coco
model_name: faster-rcnn_r50-caffe_fpn_ms-2x_coco
model_name: faster-rcnn_r50-caffe_fpn_ms-3x_coco
model_name: faster-rcnn_r50_fpn_mstrain_3x_coco
model_name: faster-rcnn_r101-caffe_fpn_ms-3x_coco
model_name: faster-rcnn_r101_fpn_ms-3x_coco
model_name: faster-rcnn_x101-32x4d_fpn_ms-3x_coco
model_name: faster-rcnn_x101-32x8d_fpn_ms-3x_coco
model_name: faster-rcnn_x101-64x4d_fpn_ms-3x_coco
model_name: faster-rcnn_r50_fpn_tnr-pretrain_1x_coco
model_name: fcos_r50-caffe_fpn_gn-head_1x_coco
model_name: fcos_r50-caffe_fpn_gn-head-center-normbbox-centeronreg-giou_1x_coco
model_name: fcos_r50-dcn-caffe_fpn_gn-head-center-normbbox-centeronreg-giou_1x_coco
model_name: fcos_r101-caffe_fpn_gn-head-1x_coco
model_name: fcos_r50-caffe_fpn_gn-head_ms-640-800-2x_coco
model_name: fcos_r101-caffe_fpn_gn-head_ms-640-800-2x_coco
model_name: fcos_x101-64x4d_fpn_gn-head_ms-640-800-2x_coco
model_name: fovea_r50_fpn_4xb4-1x_coco
model_name: fovea_r50_fpn_4xb4-2x_coco
model_name: fovea_r50_fpn_gn-head-align_4xb4-2x_coco
model_name: fovea_r50_fpn_gn-head-align_ms-640-800-4xb4-2x_coco
model_name: fovea_r101_fpn_4xb4-1x_coco
model_name: fovea_r101_fpn_4xb4-2x_coco
model_name: fovea_r101_fpn_gn-head-align_4xb4-2x_coco
model_name: fovea_r101_fpn_gn-head-align_ms-640-800-4xb4-2x_coco
model_name: faster-rcnn_r50_fpg_crop640-50e_coco
model_name: faster-rcnn_r50_fpg-chn128_crop640-50e_coco
model_name: mask-rcnn_r50_fpg_crop640-50e_coco
model_name: mask-rcnn_r50_fpg-chn128_crop640-50e_coco
model_name: retinanet_r50_fpg_crop640_50e_coco
model_name: retinanet_r50_fpg-chn128_crop640_50e_coco
model_name: freeanchor_r50_fpn_1x_coco
model_name: freeanchor_r101_fpn_1x_coco
model_name: freeanchor_x101-32x4d_fpn_1x_coco
model_name: fsaf_r50_fpn_1x_coco
model_name: fsaf_r101_fpn_1x_coco
model_name: fsaf_x101-64x4d_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_r16_gcb_c3-c5_1x_coco
model_name: mask-rcnn_r50_fpn_r4_gcb_c3-c5_1x_coco
model_name: mask-rcnn_r101-gcb-r16-c3-c5_fpn_1x_coco
model_name: mask-rcnn_r101-gcb-r4-c3-c5_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_syncbn-backbone_1x_coco
model_name: mask-rcnn_r50_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco
model_name: mask-rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco
model_name: mask-rcnn_r101-syncbn_fpn_1x_coco
model_name: mask-rcnn_r101-syncbn-gcb-r16-c3-c5_fpn_1x_coco
model_name: mask-rcnn_r101-syncbn-gcb-r4-c3-c5_fpn_1x_coco
model_name: mask-rcnn_x101-32x4d-syncbn_fpn_1x_coco
model_name: mask-rcnn_x101-32x4d-syncbn-gcb-r16-c3-c5_fpn_1x_coco
model_name: mask-rcnn_x101-32x4d-syncbn-gcb-r4-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-syncbn_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-syncbn-r16-gcb-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-syncbn-r4-gcb-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-syncbn-dconv-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-syncbn-dconv-c3-c5-r16-gcb-c3-c5_fpn_1x_coco
model_name: cascade-mask-rcnn_x101-32x4d-syncbn-dconv-c3-c5-r4-gcb-c3-c5_fpn_1x_coco
model_name: gfl_r50_fpn_1x_coco
model_name: gfl_r50_fpn_ms-2x_coco
model_name: gfl_r101_fpn_ms-2x_coco
model_name: gfl_r101-dconv-c3-c5_fpn_ms-2x_coco
model_name: gfl_x101-32x4d_fpn_ms-2x_coco
model_name: gfl_x101-32x4d-dconv-c4-c5_fpn_ms-2x_coco
model_name: retinanet_r50_fpn_ghm-1x_coco
model_name: retinanet_r101_fpn_ghm-1x_coco
model_name: retinanet_x101-32x4d_fpn_ghm-1x_coco
model_name: retinanet_x101-64x4d_fpn_ghm-1x_coco
model_name: mask-rcnn_r50_fpn_gn-all_2x_coco
model_name: mask-rcnn_r50_fpn_gn-all_3x_coco
model_name: mask-rcnn_r101_fpn_gn-all_2x_coco
model_name: mask-rcnn_r101_fpn_gn-all_3x_coco
model_name: mask-rcnn_r50_fpn_gn-all_contrib_2x_coco
model_name: mask-rcnn_r50_fpn_gn-all_contrib_3x_coco
model_name: faster-rcnn_r50_fpn_gn_ws-all_1x_coco
model_name: faster-rcnn_r101_fpn_gn-ws-all_1x_coco
model_name: faster-rcnn_x50-32x4d_fpn_gn-ws-all_1x_coco
model_name: faster-rcnn_x101-32x4d_fpn_gn-ws-all_1x_coco
model_name: mask-rcnn_r50_fpn_gn_ws-all_2x_coco
model_name: mask-rcnn_r101_fpn_gn-ws-all_2x_coco
model_name: mask-rcnn_x50-32x4d_fpn_gn-ws-all_2x_coco
model_name: mask-rcnn_x101-32x4d_fpn_gn-ws-all_2x_coco
model_name: mask-rcnn_r50_fpn_gn_ws-all_20_23_24e_coco
model_name: mask-rcnn_r101_fpn_gn-ws-all_20-23-24e_coco
model_name: mask-rcnn_x50-32x4d_fpn_gn-ws-all_20-23-24e_coco
model_name: mask-rcnn_x101-32x4d_fpn_gn-ws-all_20-23-24e_coco
model_name: grid-rcnn_r50_fpn_gn-head_2x_coco
model_name: grid-rcnn_r101_fpn_gn-head_2x_coco
model_name: grid-rcnn_x101-32x4d_fpn_gn-head_2x_coco
model_name: grid-rcnn_x101-64x4d_fpn_gn-head_2x_coco
model_name: faster-rcnn_r50_fpn_groie_1x_coco
model_name: grid-rcnn_r50_fpn_gn-head-groie_1x_coco
model_name: mask-rcnn_r50_fpn_groie_1x_coco
model_name: mask-rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco
model_name: mask-rcnn_r101_fpn_syncbn-r4-gcb_c3-c5-groie_1x_coco
model_name: ga-rpn_r50-caffe_fpn_1x_coco
model_name: ga-rpn_r101-caffe_fpn_1x_coco
model_name: ga-rpn_x101-32x4d_fpn_1x_coco
model_name: ga-rpn_x101-64x4d_fpn_1x_coco
model_name: ga-faster-rcnn_r50-caffe_fpn_1x_coco
model_name: ga-faster-rcnn_r101-caffe_fpn_1x_coco
model_name: ga-faster-rcnn_x101-32x4d_fpn_1x_coco
model_name: ga-faster-rcnn_x101-64x4d_fpn_1x_coco
model_name: ga-retinanet_r50-caffe_fpn_1x_coco
model_name: ga-retinanet_r101-caffe_fpn_1x_coco
model_name: ga-retinanet_x101-32x4d_fpn_1x_coco
model_name: ga-retinanet_x101-64x4d_fpn_1x_coco
model_name: faster-rcnn_hrnetv2p-w18-1x_coco
model_name: faster-rcnn_hrnetv2p-w18-2x_coco
model_name: faster-rcnn_hrnetv2p-w32-1x_coco
model_name: faster-rcnn_hrnetv2p-w32_2x_coco
model_name: faster-rcnn_hrnetv2p-w40-1x_coco
model_name: faster-rcnn_hrnetv2p-w40_2x_coco
model_name: mask-rcnn_hrnetv2p-w18-1x_coco
model_name: mask-rcnn_hrnetv2p-w18-2x_coco
model_name: mask-rcnn_hrnetv2p-w32-1x_coco
model_name: mask-rcnn_hrnetv2p-w32-2x_coco
model_name: mask-rcnn_hrnetv2p-w40_1x_coco
model_name: mask-rcnn_hrnetv2p-w40-2x_coco
model_name: cascade-rcnn_hrnetv2p-w18-20e_coco
model_name: cascade-rcnn_hrnetv2p-w32-20e_coco
model_name: cascade-rcnn_hrnetv2p-w40-20e_coco
model_name: cascade-mask-rcnn_hrnetv2p-w18_20e_coco
model_name: cascade-mask-rcnn_hrnetv2p-w32_20e_coco
model_name: cascade-mask-rcnn_hrnetv2p-w40-20e_coco
model_name: htc_hrnetv2p-w18_20e_coco
model_name: htc_hrnetv2p-w32_20e_coco
model_name: htc_hrnetv2p-w40_20e_coco
model_name: fcos_hrnetv2p-w18-gn-head_4xb4-1x_coco
model_name: fcos_hrnetv2p-w18-gn-head_4xb4-2x_coco
model_name: fcos_hrnetv2p-w32-gn-head_4xb4-1x_coco
model_name: fcos_hrnetv2p-w32-gn-head_4xb4-2x_coco
model_name: fcos_hrnetv2p-w18-gn-head_ms-640-800-4xb4-2x_coco
model_name: fcos_hrnetv2p-w32-gn-head_ms-640-800-4xb4-2x_coco
model_name: fcos_hrnetv2p-w40-gn-head_ms-640-800-4xb4-2x_coco
model_name: htc_r50_fpn_1x_coco
model_name: htc_r50_fpn_20e_coco
model_name: htc_r101_fpn_20e_coco
model_name: htc_x101-32x4d_fpn_16xb1-20e_coco
model_name: htc_x101-64x4d_fpn_16xb1-20e_coco
model_name: htc_x101-64x4d-dconv-c3-c5_fpn_ms-400-1400-16xb1-20e_coco
model_name: mask-rcnn_r50_fpn_instaboost_4x_coco
model_name: mask-rcnn_r101_fpn_instaboost-4x_coco
model_name: mask-rcnn_x101-64x4d_fpn_instaboost-4x_coco
model_name: cascade-mask-rcnn_r50_fpn_instaboost_4x_coco
model_name: lad_r101-paa-r50_fpn_2xb8_coco_1x
model_name: lad_r50-paa-r101_fpn_2xb8_coco_1x
model_name: ld_r18-gflv1-r101_fpn_1x_coco
model_name: ld_r34-gflv1-r101_fpn_1x_coco
model_name: ld_r50-gflv1-r101_fpn_1x_coco
model_name: ld_r101-gflv1-r101-dcn_fpn_2x_coco
model_name: libra-faster-rcnn_r50_fpn_1x_coco
model_name: libra-faster-rcnn_r101_fpn_1x_coco
model_name: libra-faster-rcnn_x101-64x4d_fpn_1x_coco
model_name: libra-retinanet_r50_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_sample1e-3_ms-2x_lvis-v0.5
model_name: mask-rcnn_r101_fpn_sample1e-3_ms-2x_lvis-v0.5
model_name: mask-rcnn_x101-32x4d_fpn_sample1e-3_ms-2x_lvis-v0.5
model_name: mask-rcnn_x101-64x4d_fpn_sample1e-3_ms-2x_lvis-v0.5
model_name: mask-rcnn_r50_fpn_sample1e-3_ms-1x_lvis-v1
model_name: mask-rcnn_r101_fpn_sample1e-3_ms-1x_lvis-v1
model_name: mask-rcnn_x101-32x4d_fpn_sample1e-3_ms-1x_lvis-v1
model_name: mask-rcnn_x101-64x4d_fpn_sample1e-3_ms-1x_lvis-v1
model_name: mask2former_swin-s-p4-w7-224_8xb2-lsj-50e_coco-panoptic
model_name: mask2former_r101_8xb2-lsj-50e_coco
model_name: mask2former_r101_8xb2-lsj-50e_coco-panoptic
model_name: mask2former_r50_8xb2-lsj-50e_coco-panoptic
model_name: mask2former_swin-t-p4-w7-224_8xb2-lsj-50e_coco-panoptic
model_name: mask2former_r50_8xb2-lsj-50e_coco
model_name: mask2former_swin-l-p4-w12-384-in21k_16xb1-lsj-100e_coco-panoptic
model_name: mask2former_swin-b-p4-w12-384-in21k_8xb2-lsj-50e_coco-panoptic
model_name: mask2former_swin-b-p4-w12-384_8xb2-lsj-50e_coco-panoptic
model_name: mask2former_swin-t-p4-w7-224_8xb2-lsj-50e_coco
model_name: mask2former_swin-s-p4-w7-224_8xb2-lsj-50e_coco
model_name: mask-rcnn_r50-caffe_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_fp16_1x_coco
model_name: mask-rcnn_r50_fpn_2x_coco
model_name: mask-rcnn_r101-caffe_fpn_1x_coco
model_name: mask-rcnn_r101_fpn_1x_coco
model_name: mask-rcnn_r101_fpn_2x_coco
model_name: mask-rcnn_x101-32x4d_fpn_1x_coco
model_name: mask-rcnn_x101-32x4d_fpn_2x_coco
model_name: mask-rcnn_x101-64x4d_fpn_1x_coco
model_name: mask-rcnn_x101-64x4d_fpn_2x_coco
model_name: mask-rcnn_x101-32x8d_fpn_1x_coco
model_name: mask-rcnn_r50-caffe_fpn_ms-poly-2x_coco
model_name: mask-rcnn_r50-caffe_fpn_ms-poly-3x_coco
model_name: mask-rcnn_r50_fpn_mstrain-poly_3x_coco
model_name: mask-rcnn_r101_fpn_ms-poly-3x_coco
model_name: mask-rcnn_r101-caffe_fpn_ms-poly-3x_coco
model_name: mask-rcnn_x101-32x4d_fpn_ms-poly-3x_coco
model_name: mask-rcnn_x101-32x8d_fpn_ms-poly-1x_coco
model_name: mask-rcnn_x101-32x8d_fpn_ms-poly-3x_coco
model_name: mask-rcnn_x101-64x4d_fpn_ms-poly_3x_coco
model_name: maskformer_r50_ms-16xb1-75e_coco
model_name: maskformer_swin-l-p4-w12_64xb1-ms-300e_coco
model_name: ms-rcnn_r50-caffe_fpn_1x_coco
model_name: ms-rcnn_r50-caffe_fpn_2x_coco
model_name: ms-rcnn_r101-caffe_fpn_1x_coco
model_name: ms-rcnn_r101-caffe_fpn_2x_coco
model_name: ms-rcnn_x101-32x4d_fpn_1x_coco
model_name: ms-rcnn_x101-64x4d_fpn_1x_coco
model_name: ms-rcnn_x101-64x4d_fpn_2x_coco
model_name: nas-fcos_r50-caffe_fpn_nashead-gn-head_4xb4-1x_coco
model_name: nas-fcos_r50-caffe_fpn_fcoshead-gn-head_4xb4-1x_coco
model_name: retinanet_r50_fpn_crop640-50e_coco
model_name: retinanet_r50_nasfpn_crop640-50e_coco
model_name: faster-rcnn_r50_fpn_32x2_1x_openimages
model_name: retinanet_r50_fpn_32xb2-1x_openimages
model_name: ssd300_32xb8-36e_openimages
model_name: faster-rcnn_r50_fpn_32x2_1x_openimages_challenge
model_name: faster-rcnn_r50_fpn_32x2_cas_1x_openimages
model_name: faster-rcnn_r50_fpn_32x2_cas_1x_openimages_challenge
model_name: paa_r50_fpn_1x_coco
model_name: paa_r50_fpn_1.5x_coco
model_name: paa_r50_fpn_2x_coco
model_name: paa_r50_fpn_mstrain_3x_coco
model_name: paa_r101_fpn_1x_coco
model_name: paa_r101_fpn_2x_coco
model_name: paa_r101_fpn_mstrain_3x_coco
model_name: faster-rcnn_r50_pafpn_1x_coco
model_name: panoptic_fpn_r50_fpn_1x_coco
model_name: panoptic_fpn_r50_fpn_mstrain_3x_coco
model_name: panoptic_fpn_r101_fpn_1x_coco
model_name: panoptic_fpn_r101_fpn_mstrain_3x_coco
model_name: retinanet_pvt-t_fpn_1x_coco
model_name: retinanet_pvt-s_fpn_1x_coco
model_name: retinanet_pvt-m_fpn_1x_coco
model_name: retinanet_pvtv2-b0_fpn_1x_coco
model_name: retinanet_pvtv2-b1_fpn_1x_coco
model_name: retinanet_pvtv2-b2_fpn_1x_coco
model_name: retinanet_pvtv2-b3_fpn_1x_coco
model_name: retinanet_pvtv2-b4_fpn_1x_coco
model_name: retinanet_pvtv2-b5_fpn_1x_coco
model_name: pisa_faster_rcnn_r50_fpn_1x_coco
model_name: pisa_faster_rcnn_x101_32x4d_fpn_1x_coco
model_name: pisa_mask_rcnn_r50_fpn_1x_coco
model_name: pisa_retinanet_r50_fpn_1x_coco
model_name: pisa_retinanet_x101_32x4d_fpn_1x_coco
model_name: pisa_ssd300_coco
model_name: pisa_ssd512_coco
model_name: point_rend_r50_caffe_fpn_mstrain_1x_coco
model_name: point_rend_r50_caffe_fpn_mstrain_3x_coco
model_name: queryinst_r50_fpn_1x_coco
model_name: queryinst_r50_fpn_ms-480-800-3x_coco
model_name: queryinst_r50_fpn_300-proposals_crop-ms-480-800-3x_coco
model_name: queryinst_r101_fpn_ms-480-800-3x_coco
model_name: queryinst_r101_fpn_300-proposals_crop-ms-480-800-3x_coco
model_name: mask-rcnn_regnetx-3.2GF_fpn_1x_coco
model_name: mask-rcnn_regnetx-4GF_fpn_1x_coco
model_name: mask-rcnn_regnetx-6.4GF_fpn_1x_coco
model_name: mask-rcnn_regnetx-8GF_fpn_1x_coco
model_name: mask-rcnn_regnetx-12GF_fpn_1x_coco
model_name: mask-rcnn_regnetx-3.2GF-mdconv-c3-c5_fpn_1x_coco
model_name: faster-rcnn_regnetx-3.2GF_fpn_1x_coco
model_name: faster-rcnn_regnetx-3.2GF_fpn_2x_coco
model_name: retinanet_regnetx-800MF_fpn_1x_coco
model_name: retinanet_regnetx-1.6GF_fpn_1x_coco
model_name: retinanet_regnetx-3.2GF_fpn_1x_coco
model_name: faster-rcnn_regnetx-400MF_fpn_ms-3x_coco
model_name: faster-rcnn_regnetx-800MF_fpn_ms-3x_coco
model_name: faster-rcnn_regnetx-1.6GF_fpn_ms-3x_coco
model_name: faster-rcnn_regnetx-3.2GF_fpn_ms-3x_coco
model_name: faster-rcnn_regnetx-4GF_fpn_ms-3x_coco
model_name: mask-rcnn_regnetx-3.2GF_fpn_ms-3x_coco
model_name: mask-rcnn_regnetx-400MF_fpn_ms-poly-3x_coco
model_name: mask-rcnn_regnetx-800MF_fpn_ms-poly-3x_coco
model_name: mask-rcnn_regnetx-1.6GF_fpn_ms-poly-3x_coco
model_name: mask-rcnn_regnetx-3.2GF_fpn_ms-3x_coco
model_name: mask-rcnn_regnetx-4GF_fpn_ms-poly-3x_coco
model_name: cascade-mask-rcnn_regnetx-400MF_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_regnetx-800MF_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_regnetx-1.6GF_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_regnetx-3.2GF_fpn_ms-3x_coco
model_name: cascade-mask-rcnn_regnetx-4GF_fpn_ms-3x_coco
model_name: reppoints-bbox_r50_fpn-gn_head-gn-grid_1x_coco
model_name: reppoints-bbox_r50-center_fpn-gn_head-gn-grid_1x_coco
model_name: reppoints-moment_r50_fpn_1x_coco
model_name: reppoints-moment_r50_fpn-gn_head-gn_1x_coco
model_name: reppoints-moment_r50_fpn-gn_head-gn_2x_coco
model_name: reppoints-moment_r101_fpn-gn_head-gn_2x_coco
model_name: reppoints-moment_r101-dconv-c3-c5_fpn-gn_head-gn_2x_coco
model_name: reppoints-moment_x101-dconv-c3-c5_fpn-gn_head-gn_2x_coco
model_name: faster-rcnn_res2net-101_fpn_2x_coco
model_name: mask-rcnn_res2net-101_fpn_2x_coco
model_name: cascade-rcnn_res2net-101_fpn_20e_coco
model_name: cascade-mask-rcnn_res2net-101_fpn_20e_coco
model_name: htc_res2net-101_fpn_20e_coco
model_name: faster-rcnn_s50_fpn_syncbn-backbone+head_ms-range-1x_coco
model_name: faster-rcnn_s101_fpn_syncbn-backbone+head_ms-range-1x_coco
model_name: mask-rcnn_s50_fpn_syncbn-backbone+head_ms-1x_coco
model_name: mask-rcnn_s101_fpn_syncbn-backbone+head_ms-1x_coco
model_name: cascade-rcnn_s50_fpn_syncbn-backbone+head_ms-range-1x_coco
model_name: cascade-rcnn_s101_fpn_syncbn-backbone+head_ms-range-1x_coco
model_name: cascade-mask-rcnn_s50_fpn_syncbn-backbone+head_ms-1x_coco
model_name: cascade-mask-rcnn_s101_fpn_syncbn-backbone+head_ms-1x_coco
model_name: faster-rcnn_r50_fpn_rsb-pretrain_1x_coco
model_name: cascade-mask-rcnn_r50_fpn_rsb-pretrain_1x_coco
model_name: retinanet_r50-rsb-pre_fpn_1x_coco
model_name: mask-rcnn_r50_fpn_rsb-pretrain_1x_coco
model_name: retinanet_r18_fpn_1x_coco
model_name: retinanet_r18_fpn_1xb8-1x_coco
model_name: retinanet_r50-caffe_fpn_1x_coco
model_name: retinanet_r50_fpn_1x_coco
model_name: retinanet_r50_fpn_amp-1x_coco
model_name: retinanet_r50_fpn_2x_coco
model_name: retinanet_r50_fpn_ms-640-800-3x_coco
model_name: retinanet_r101-caffe_fpn_1x_coco
model_name: retinanet_r101-caffe_fpn_ms-3x_coco
model_name: retinanet_r101_fpn_1x_coco
model_name: retinanet_r101_fpn_2x_coco
model_name: retinanet_r101_fpn_ms-640-800-3x_coco
model_name: retinanet_x101-32x4d_fpn_1x_coco
model_name: retinanet_x101-32x4d_fpn_2x_coco
model_name: retinanet_x101-64x4d_fpn_1x_coco
model_name: retinanet_x101-64x4d_fpn_2x_coco
model_name: retinanet_x101-64x4d_fpn_ms-640-800-3x_coco
model_name: rpn_r50-caffe_fpn_1x_coco
model_name: rpn_r50_fpn_1x_coco
model_name: rpn_r50_fpn_2x_coco
model_name: rpn_r101-caffe_fpn_1x_coco
model_name: rpn_x101-32x4d_fpn_1x_coco
model_name: rpn_x101-32x4d_fpn_2x_coco
model_name: rpn_x101-64x4d_fpn_1x_coco
model_name: rpn_x101-64x4d_fpn_2x_coco
model_name: rtmdet_tiny_8xb32-300e_coco
model_name: rtmdet-t
model_name: rtmdet_s_8xb32-300e_coco
model_name: rtmdet-s
model_name: rtmdet_m_8xb32-300e_coco
model_name: rtmdet-m
model_name: rtmdet_l_8xb32-300e_coco
model_name: rtmdet-l
model_name: rtmdet_x_8xb32-300e_coco
model_name: rtmdet-x
model_name: rtmdet_x_p6_4xb8-300e_coco
model_name: rtmdet-x_p6
model_name: rtmdet-ins_tiny_8xb32-300e_coco
model_name: rtmdet-ins-t
model_name: rtmdet-ins_s_8xb32-300e_coco
model_name: rtmdet-ins-s
model_name: rtmdet-ins_m_8xb32-300e_coco
model_name: rtmdet-ins-m
model_name: rtmdet-ins_l_8xb32-300e_coco
model_name: rtmdet-ins-l
model_name: rtmdet-ins_x_8xb16-300e_coco
model_name: rtmdet-ins-x
model_name: sabl-faster-rcnn_r50_fpn_1x_coco
model_name: sabl-faster-rcnn_r101_fpn_1x_coco
model_name: sabl-cascade-rcnn_r50_fpn_1x_coco
model_name: sabl-cascade-rcnn_r101_fpn_1x_coco
model_name: sabl-retinanet_r50_fpn_1x_coco
model_name: sabl-retinanet_r50-gn_fpn_1x_coco
model_name: sabl-retinanet_r101_fpn_1x_coco
model_name: sabl-retinanet_r101-gn_fpn_1x_coco
model_name: sabl-retinanet_r101-gn_fpn_ms-640-800-2x_coco
model_name: sabl-retinanet_r101-gn_fpn_ms-480-960-2x_coco
model_name: scnet_r50_fpn_1x_coco
model_name: scnet_r50_fpn_20e_coco
model_name: scnet_r101_fpn_20e_coco
model_name: scnet_x101-64x4d_fpn_20e_coco
model_name: faster-rcnn_r50_fpn_gn-all_scratch_6x_coco
model_name: mask-rcnn_r50_fpn_gn-all_scratch_6x_coco
model_name: mask-rcnn_r50_fpn_random_seesaw_loss_mstrain_2x_lvis_v1
model_name: mask-rcnn_r50_fpn_random_seesaw_loss_normed_mask_mstrain_2x_lvis_v1
model_name: mask-rcnn_r101_fpn_seesaw-loss_random-ms-2x_lvis-v1
model_name: mask-rcnn_r101_fpn_seesaw-loss-normed-mask_random-ms-2x_lvis-v1
model_name: mask-rcnn_r50_fpn_sample1e-3_seesaw_loss_mstrain_2x_lvis_v1
model_name: mask-rcnn_r50_fpn_sample1e-3_seesaw_loss_normed_mask_mstrain_2x_lvis_v1
model_name: mask-rcnn_r101_fpn_seesaw-loss_sample1e-3-ms-2x_lvis-v1
model_name: mask-rcnn_r101_fpn_seesaw-loss-normed-mask_sample1e-3-ms-2x_lvis-v1
model_name: cascade-mask-rcnn_r101_fpn_seesaw-loss_random-ms-2x_lvis-v1
model_name: cascade-mask-rcnn_r101_fpn_seesaw-loss-normed-mask_random-ms-2x_lvis-v1
model_name: cascade-mask-rcnn_r101_fpn_seesaw-loss_sample1e-3-ms-2x_lvis-v1
model_name: cascade-mask-rcnn_r101_fpn_seesaw-loss-normed-mask_sample1e-3-ms-2x_lvis-v1
model_name: mask-rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_32x2_270k_coco
model_name: mask-rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_32x2_90k_coco
model_name: mask-rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_scp_32x2_270k_coco
model_name: mask-rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_scp_32x2_90k_coco
model_name: soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.01-coco.py
model_name: soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.02-coco.py
model_name: soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.05-coco.py
model_name: soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.1-coco.py
model_name: sparse-rcnn_r50_fpn_1x_coco
model_name: sparse-rcnn_r50_fpn_ms-480-800-3x_coco
model_name: sparse-rcnn_r50_fpn_300-proposals_crop-ms-480-800-3x_coco
model_name: sparse-rcnn_r101_fpn_ms-480-800-3x_coco
model_name: sparse-rcnn_r101_fpn_300-proposals_crop-ms-480-800-3x_coco
model_name: decoupled-solo_r50_fpn_1x_coco
model_name: decoupled-solo_r50_fpn_3x_coco
model_name: decoupled-solo-light_r50_fpn_3x_coco
model_name: solo_r50_fpn_3x_coco
model_name: solo_r50_fpn_1x_coco
model_name: solov2_r50_fpn_1x_coco
model_name: solov2_r50_fpn_ms-3x_coco
model_name: solov2_r101-dcn_fpn_ms-3x_coco
model_name: solov2_x101-dcn_fpn_ms-3x_coco
model_name: solov2-light_r18_fpn_ms-3x_coco
model_name: solov2-light_r50_fpn_ms-3x_coco
model_name: ssd300_coco
model_name: ssd512_coco
model_name: ssdlite_mobilenetv2-scratch_8xb24-600e_coco
model_name: mask-rcnn_r50-caffe_fpn_rpn-2conv_4conv1fc_syncbn-all_lsj-100e_coco
model_name: mask-rcnn_swin-s-p4-w7_fpn_amp-ms-crop-3x_coco
model_name: mask-rcnn_swin-t-p4-w7_fpn_ms-crop-3x_coco
model_name: mask-rcnn_swin-t-p4-w7_fpn_1x_coco
model_name: mask-rcnn_swin-t-p4-w7_fpn_amp-ms-crop-3x_coco
model_name: tridentnet_r50-caffe_1x_coco
model_name: tridentnet_r50-caffe_ms-1x_coco
model_name: tridentnet_r50-caffe_ms-3x_coco
model_name: tood_r101_fpn_ms-2x_coco
model_name: tood_x101-64x4d_fpn_ms-2x_coco
model_name: tood_r101-dconv-c3-c5_fpn_ms-2x_coco
model_name: tood_r50_fpn_anchor-based_1x_coco
model_name: tood_r50_fpn_1x_coco
model_name: tood_r50_fpn_ms-2x_coco
model_name: vfnet_r50_fpn_1x_coco
model_name: vfnet_r50_fpn_ms-2x_coco
model_name: vfnet_r50-mdconv-c3-c5_fpn_ms-2x_coco
model_name: vfnet_r101_fpn_1x_coco
model_name: vfnet_r101_fpn_ms-2x_coco
model_name: vfnet_r101-mdconv-c3-c5_fpn_ms-2x_coco
model_name: vfnet_x101-32x4d-mdconv-c3-c5_fpn_ms-2x_coco
model_name: vfnet_x101-64x4d-mdconv-c3-c5_fpn_ms-2x_coco
model_name: yolact_r50_1x8_coco
model_name: yolact_r50_8x8_coco
model_name: yolact_r101_1x8_coco
model_name: yolov3_d53_320_273e_coco
model_name: yolov3_d53_mstrain-416_273e_coco
model_name: yolov3_d53_mstrain-608_273e_coco
model_name: yolov3_d53_fp16_mstrain-608_273e_coco
model_name: yolov3_mobilenetv2_8xb24-320-300e_coco
model_name: yolov3_mobilenetv2_8xb24-ms-416-300e_coco
model_name: yolof_r50_c5_8x8_1x_coco
model_name: yolox_s_8x8_300e_coco
model_name: yolox_l_8x8_300e_coco
model_name: yolox_x_8x8_300e_coco
model_name: yolox_tiny_8x8_300e_coco
model_name: bytetrack_yolox_x_8xb4-amp-80e_crowdhuman-mot17halftrain_test-mot17halfval
model_name: bytetrack_yolox_x_8xb4-amp-80e_crowdhuman-mot17halftrain_test-mot17test
model_name: bytetrack_yolox_x_8xb4-amp-80e_crowdhuman-mot20train_test-mot20test
model_name: strongsort_yolox_x_8xb4-80e_crowdhuman-mot17halftrain_test-mot17halfval
model_name: strongsort_yolox_x_8xb4-80e_crowdhuman-mot20train_test-mot20test
model_name: ocsort_yolox_x_crowdhuman_mot17-private-half
model_name: sort_faster-rcnn_r50_fpn_8xb2-4e_mot17halftrain_test-mot17halfval
model_name: deepsort_faster-rcnn_r50_fpn_8xb2-4e_mot17halftrain_test-mot17halfval
model_name: qdtrack_faster-rcnn_r50_fpn_8xb2-4e_mot17halftrain_test-mot17halfval
model_name: mask2former_r50_8xb2-8e_youtubevis2021
model_name: mask2former_r101_8xb2-8e_youtubevis2021
model_name: mask2former_swin-l-p4-w12-384-in21k_8xb2-8e_youtubevis2021.py
model_name: masktrack-rcnn_mask-rcnn_r50_fpn_8xb1-12e_youtubevis2019
model_name: masktrack-rcnn_mask-rcnn_r101_fpn_8xb1-12e_youtubevis2019
model_name: masktrack-rcnn_mask-rcnn_x101_fpn_8xb1-12e_youtubevis2019
model_name: masktrack-rcnn_mask-rcnn_r50_fpn_8xb1-12e_youtubevis2021
model_name: masktrack-rcnn_mask-rcnn_r101_fpn_8xb1-12e_youtubevis2021
model_name: masktrack-rcnn_mask-rcnn_x101_fpn_8xb1-12e_youtubevis2021
model_name: glip_atss_swin-t_a_fpn_dyhead_pretrain_obj365
model_name: glip_atss_swin-t_b_fpn_dyhead_pretrain_obj365
model_name: glip_atss_swin-t_c_fpn_dyhead_pretrain_obj365-goldg
model_name: glip_atss_swin-t_fpn_dyhead_pretrain_obj365-goldg-cc3m-sub
model_name: glip_atss_swin-l_fpn_dyhead_pretrain_mixeddata
model_name: glip_atss_swin-t_a_fpn_dyhead_16xb2_ms-2x_funtune_coco
model_name: glip_atss_swin-t_b_fpn_dyhead_16xb2_ms-2x_funtune_coco
model_name: glip_atss_swin-t_c_fpn_dyhead_16xb2_ms-2x_funtune_coco
model_name: glip_atss_swin-t_fpn_dyhead_16xb2_ms-2x_funtune_coco
model_name: glip_atss_swin-l_fpn_dyhead_16xb2_ms-2x_funtune_coco
model_name: ddq-detr-4scale_r50_8xb2-12e_coco
model_name: ddq-detr-5scale_r50_8xb2-12e_coco
model_name: ddq-detr-4scale_swinl_8xb2-30e_coco
model_name: grounding_dino_swin-t_pretrain_obj365_goldg_cap4m
model_name: grounding_dino_swin-b_pretrain_mixeddata
model_name: grounding_dino_swin-t_finetune_16xb2_1x_coco
model_name: grounding_dino_swin-b_finetune_16xb2_1x_coco
model_name: grounding_dino_r50_scratch_8xb2_1x_coco

Loads checkpoint by http backend from path: https://download.openmmlab.com/mmdetection/v2.0/atss/atss_r50_fpn_1x_coco/atss_r50_fpn_1x_coco_20200209-985f7bd0.pth
/home/usama/Desktop/visual-ml/venv/lib/python3.10/site-packages/mmdet/models/dense_heads/anchor_head.py:108: UserWarning: DeprecationWarning: `num_anchors` is deprecated, for consistency or also use `num_base_priors` instead
  warnings.warn('DeprecationWarning: `num_anchors` is deprecated, '
/home/usama/Desktop/visual-ml/venv/lib/python3.10/site-packages/mmengine/visualization/visualizer.py:196: UserWarning: Failed to add <class 'mmengine.visualization.vis_backend.LocalVisBackend'>, please provide the `save_dir` argument.
  warnings.warn(f'Failed to add {vis_backend.__class__}, '
/home/usama/Desktop/visual-ml/venv/lib/python3.10/site-packages/mmdet/apis/inference.py:70: UserWarning: checkpoint is None, use COCO classes by default.
  warnings.warn('checkpoint is None, use COCO classes by default.')
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'torch.nn.modules.normalization.GroupNorm'> is not supported for memory computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for flops computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for madds computation.
  warnings.warn(
/home/usama/Desktop/visual-ml/metinor/metinor/profiler/handlers/__init__.py:15: UserWarning: Module <class 'mmcv.cnn.bricks.scale.Scale'> is not supported for memory computation.
  warnings.warn(
../../_images/nervai-optim_inspection_tutorial_1_4_2.svg

Parameters#

  • model_name: Model name from torchvision , mmdetection or mmpose.

  • task: classification , detection or pose

  • filter_by: Optional , default set to None

  • depth: Optional , default set to max

  • device: Optional , default set tot cpu

Returns#

  • ComputationGraph object that includes the visualization of the model.