↓ 1 callersFunctiontrain_postal_detection(cfg, model, criterion, data_loader, optimizer, epoch, max_norm, lr_scheduler, writer=None, model_ema=None)
utils/video_action_recognition.py:86
↓ 1 callersFunctionvalidate_postal_ucf_detection(cfg, model, criterion, postprocessors, data_loader, epoch, writer, prefix='')
utils/video_action_recognition.py:476
↓ 1 callersFunctionvalidate_postal_ucf_detection_ema(cfg, model, criterion, postprocessors, data_loader, epoch, writer, prefix='')
utils/video_action_recognition.py:712
Method__init__(self, directory, video_path, transforms, clip_len=8, crop_size=224, resize_size=256,
mode='t
datasets/basketball_jhmdb_frame.py:48
Method__init__(self,
in_channels,
out_channels,
kernel_size=(1, 1, 1),
models/I3D_Backbone.py:40
Method__init__(self, dim, window_size, num_heads, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
models/backbones/video_swin_transformer.py:118
Method__init__(self, dim, num_heads, window_size=(2,7,7), shift_size=(0,0,0),
mlp_ratio=4., qkv_bias=True,
models/backbones/video_swin_transformer.py:207
Method__init__(self, patch_size=(2,4,4), in_chans=3, embed_dim=96, norm_layer=None)
models/backbones/video_swin_transformer.py:436
Method__init__(self,
pretrained=None,
pretrained2d=True,
patch_size=(4,4,
models/backbones/video_swin_transformer.py:493
Method__init__(self, in_planes, planes, stride=1, temporal_stride=1,
down_sample=None, expansion=2, tempora
models/backbones/ir_CSN_152.py:34
Method__init__(self, in_planes, planes, stride=1, temporal_stride=1,
down_sample=None, expansion=2, tempora
models/backbones/ir_CSN_50.py:31
Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
activation="relu", normalize_before
models/transformer/transformer_layers.py:47
Method__init__(self, embed_dim, num_heads, dropout=0.0, bias=True, kdim=None, vdim=None)
models/transformer/transformer_layers.py:170
Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation='relu')
models/transformer/transformer_layers.py:406