SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

change anchor_image_size None for compatibility (#31640)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/models/rt_detr/configuration_rt_detr.py
View File

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

2
tests/models/rt_detr/test_modeling_rt_detr.py
View File

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