"src/lib/vscode:/vscode.git/clone" did not exist on "2f23325c90bc0d2416c93352b4f4f084e097dc80"
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): ...@@ -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): ...@@ -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,
......
...@@ -91,7 +91,7 @@ class RTDetrModelTester: ...@@ -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,
......
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