↓ 2 callersFunctionbuild_custom_train_loader(
dataset, *, mapper, sampler,
total_batch_size=16,
num_workers=0,
num_datase
grit/data/custom_dataset_dataloader.py:70
↓ 1 callersMethodbox_reg_loss(
self, proposal_boxes, gt_boxes, pred_deltas, gt_classes,
num_classes=-1)
grit/modeling/roi_heads/grit_fast_rcnn.py:80
↓ 1 callersFunctionbuild_dataset_batch_data_loader(
dataset_bs, dataset, sampler, total_batch_size, num_datasets, num_workers=0
)
grit/data/custom_dataset_dataloader.py:96
Method__init__(self, bottom_up=None, top_block=None, out_channels=None, strides=None, vit_out_dim=None)
grit/modeling/backbone/vit.py:367