Initializes the model. Args: backbone: torch module of the backbone to be used. See backbone.py transformer: torch module of the transformer architecture. See transformer.py num_classes: number of object classes num_queries: number of object q
(
self,
backbone,
transformer,
num_classes,
num_queries,
aux_loss=False,
contrastive_hdim=64,
contrastive_loss=False,
contrastive_align_loss=False,
qa_dataset: Optional[str] = None,
split_qa_heads=True,
predict_final=False,
yourefit=False,
pose=True
)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected