change anchor_image_size None for compatibility (#31640)

* change anchor_image_size None for compatibility

* make fix-copies
This commit is contained in:
Sangbum Daniel Choi
2024-06-27 20:36:55 +09:00
committed by GitHub
parent 3a028101e9
commit be50a0338b
2 changed files with 4 additions and 4 deletions

View File

@@ -118,8 +118,8 @@ class RTDetrConfig(PretrainedConfig):
Scale or magnitude of noise to be added to the bounding boxes.
learn_initial_query (`bool`, *optional*, defaults to `False`):
Indicates whether the initial query embeddings for the decoder should be learned during training
anchor_image_size (`Tuple[int, int]`, *optional*, defaults to `[640, 640]`):
Height and width of the input image used during evaluation to generate the bounding box anchors.
anchor_image_size (`Tuple[int, int]`, *optional*):
Height and width of the input image used during evaluation to generate the bounding box anchors. If None, automatic generate anchor is applied.
disable_custom_kernels (`bool`, *optional*, defaults to `True`):
Whether to disable custom kernels.
with_box_refine (`bool`, *optional*, defaults to `True`):
@@ -218,7 +218,7 @@ class RTDetrConfig(PretrainedConfig):
label_noise_ratio=0.5,
box_noise_scale=1.0,
learn_initial_query=False,
anchor_image_size=[640, 640],
anchor_image_size=None,
disable_custom_kernels=True,
with_box_refine=True,
is_encoder_decoder=True,

View File

@@ -91,7 +91,7 @@ class RTDetrModelTester:
label_noise_ratio=0.5,
box_noise_scale=1.0,
learn_initial_query=False,
anchor_image_size=[64, 64],
anchor_image_size=None,
image_size=64,
disable_custom_kernels=True,
with_box_refine=True,