Unverified Commit be50a033 authored by Sangbum Daniel Choi's avatar Sangbum Daniel Choi Committed by GitHub
Browse files

change anchor_image_size None for compatibility (#31640)

* change anchor_image_size None for compatibility

* make fix-copies
parent 3a028101
......@@ -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,
......
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment