o
    {i	                     @   s4   d dl mZ ddlmZ eddG dd dZdS )	    )	dataclass   )PositionEncodingT)kw_onlyc                   @   s  e Zd ZU dZeed< dZeed< dZeed< dZ	eed< dZ
eed< dZeed	< d
Zeed< dZeed< dZeed< 	 dZeed< dZeed< ejZeed< dZeed< dZeed< dZeed< dZeed< dZeed< dZeed< dZeed < dZeed!< d"Zeed#< d$Zeed%< d&Z eed'< d(Z!eed)< d*Z"eed+< d,Z#eed-< dZ$eed.< dZ%eed/< d0Z&e'e d0B ed1< d0Z(e'e d0B ed2< d3Z)eed4< d0Z*ed0B ed5< d0Z+e'e d0B ed6< d(Z,eed7< d0S )8DetectionHeadConfig[   num_classesTwith_box_refine	two_stagemixed_selectionlook_forward_twice   
k_one2manyg      ?lambda_one2manyi,  num_queries_one2onei  num_queries_one2manyreparamd   topkposition_embeddingr   num_feature_levels
dec_layersi   dim_feedforward   
hidden_dimg        dropout   nheadspre_norm	norm_typeaux_loss   proposal_feature_levels2   proposal_min_sizeglobal_rpe_decompdecoder_typeFdecoder_use_checkpointi   decoder_rpe_hidden_dimlineardecoder_rpe_typeadd_transformer_encodernum_encoder_layersNlayers_to_useblocks_to_trainr   n_windows_sqrtproposal_in_strideproposal_tgt_stridesbackbone_use_layernorm)-__name__
__module____qualname__r   int__annotations__r	   boolr
   r   r   r   r   floatr   r   r   r   r   SINEr   r   r   r   r   r   r   r   strr    r"   r$   r&   r'   r(   r*   r+   r,   r-   listr.   r/   r0   r1   r2    r=   r=   H/data/cameron/keygrip/volume_dino_tracks/dinov3/eval/detection/config.pyr      sH   
 r   N)dataclassesr   Zmodels.position_encodingr   r   r=   r=   r=   r>   <module>   s   