↓ 2 callersMethodget_loss(self, loss, outputs, targets, indices, num_boxes, **kwargs)
src/models/criterion.py:199
↓ 2 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
src/models/swin_backbone.py:36
↓ 1 callersMethod__init__(self, d_model=256, nhead=8, num_encoder_layers=6, num_decoder_layers=6,
dim_feedforward=1024
src/models/devis_ablation_transformer_wo_t_conn.py:18
↓ 1 callersMethod_inference_forward(self, targets, out, hs, memories_att_map, masks_att_map, mask_head_feats, spatial_shapes)
src/models/devis_segmentation.py:75
↓ 1 callersMethod_inference_forward(self, out, hs, memories_att_map, masks_att_map, mask_head_feats)
src/models/deformable_segmentation.py:220
↓ 1 callersMethod_training_forward(self, targets, out, hs, loss_lvl, memories_att_map, masks_att_map, mask_head_feats)
src/models/devis_segmentation.py:69
↓ 1 callersFunctionadapt_weights_devis(checkpoint: dict, model_state_dict: dict, lvl_res: int, focal_loss: bool, finetune_class_logits: bool, num_fr
src/util/weights_loading_utils.py:52