torchsig.models.spectrogram_models.detr.modules.create_detr

torchsig.models.spectrogram_models.detr.modules.create_detr(backbone: str = 'efficientnet_b0', transformer: str = 'xcit-nano', num_classes: int = 61, num_objects: int = 50, hidden_dim: int = 256, drop_rate_backbone: float = 0.2, drop_path_rate_backbone: float = 0.2, drop_path_rate_transformer: float = 0.1, ds_rate_transformer: int = 2, ds_method_transformer: str = 'chunker') → Module

Function used to build a DETR network

Parameters:

TODO

Returns:

torch.nn.Module