↓ 2 callersFunctionsave_checkpoint_new(config, epoch, model, max_accuracy, optimizer, lr_scheduler, logger, name=None)
agent_transformer/utils.py:193
↓ 2 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
downstream/detection/mmdet/models/backbones/swin_bsl.py:37
↓ 2 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
downstream/detection/mmdet/models/backbones/agent_swin.py:41
↓ 2 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
downstream/segmentation/mmseg/models/backbones/agent_swin.py:41
↓ 2 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
agent_transformer/models/agent_swin.py:38
↓ 2 callersMethodwindow_partition Args: x: (B, H, W, C) Returns: windows: (num_windows*B, window_size, window_size, C)
downstream/detection/mmdet/models/backbones/swin.py:272
↓ 2 callersMethodwindow_partition Args: x: (B, H, W, C) Returns: windows: (num_windows*B, window_size, window_size, C)
downstream/segmentation/mmseg/models/backbones/swin.py:272
↓ 1 callersMethod__init__(self,
depth=53,
out_indices=(3, 4, 5),
frozen_stages=-1,
downstream/detection/mmdet/models/backbones/darknet.py:101
↓ 1 callersMethod__init__(self,
num_classes,
in_channels,
base_edge_list=(16, 32, 64
downstream/detection/mmdet/models/dense_heads/fovea_head.py:55
↓ 1 callersMethod__init__(self, *args, tempearture=20, power=1.0, eps=1e-6, **kwargs)
downstream/detection/mmdet/models/utils/normed_predictor.py:21
↓ 1 callersMethod__init__(self,
in_channels=(64, 256, 256),
out_channels=128,
conv_c
downstream/segmentation/mmseg/models/necks/ic_neck.py:103
↓ 1 callersMethod__init__(self, dilations=(1, 6, 12, 18), **kwargs)
downstream/segmentation/mmseg/models/decode_heads/aspp_head.py:65
↓ 1 callersMethod__init__(self, pool_scales, in_channels, channels, conv_cfg, norm_cfg,
act_cfg, align_corners, **kwar
downstream/segmentation/mmseg/models/decode_heads/psp_head.py:25