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

drop_last = False not supported #388

Open
richardrl opened this issue Aug 19, 2024 · 0 comments
Open

drop_last = False not supported #388

richardrl opened this issue Aug 19, 2024 · 0 comments

Comments

@richardrl
Copy link

  File "/data/scratch-oc40/pulkitag/rli14/micromamba/ffcv_oct40/lib/python3.9/threading.py", line 980, in _bootstrap_inner
    self.run()
  File "/data/pulkitag/models/rli14/ffcv/ffcv/loader/epoch_iterator.py", line 84, in run
    result = self.run_pipeline(b_ix, ixes, slot, events[slot])
  File "/data/pulkitag/models/rli14/ffcv/ffcv/loader/epoch_iterator.py", line 146, in run_pipeline
    results = stage_code(**args)
  File "", line 2, in stage_code_1
  File "/data/pulkitag/models/rli14/ffcv/ffcv/transforms/ops.py", line 153, in convert
    return inp.view(self.target_dtype)
RuntimeError: shape '[2, 26, 84]' is invalid for input of size 2184

I think the code does not support the variable size final batch constructed by drop_last = False.

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