Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Issue with YOLOX-Tiny: Variable Input Size Not Functioning #4108

Open
eugene123tw opened this issue Nov 11, 2024 · 0 comments
Open

Issue with YOLOX-Tiny: Variable Input Size Not Functioning #4108

eugene123tw opened this issue Nov 11, 2024 · 0 comments

Comments

@eugene123tw
Copy link
Contributor

eugene123tw commented Nov 11, 2024

Describe the bug

Variable input size with YOLOX-tiny variant not working

Steps to Reproduce

otx train --config src/otx/recipe/detection/yolox_tiny.yaml
            --data.input_size 640
            --data_root tests/assets/car_tree_bug
            --engine.device gpu

Environment:

  • OS:
  • Framework version:
  • Python version:
  • OpenVINO version:
  • CUDA/cuDNN version:
  • GPU model and memory:
Exception has occurred: RuntimeError       (note: full exception trace is shown but execution is paused at: _run_module_as_main)
Sizes of tensors must match except in dimension 1. Expected size 124 but got size 123 for tensor number 1 in the list.
  File "/home/yuchunli/git/otx-2x/src/otx/algo/detection/necks/yolox_pafpn.py", line 158, in forward
    inner_out = self.top_down_blocks[len(self.in_channels) - 1 - idx](torch.cat([upsample_feat, feat_low], 1))
                                                                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant