Skip to content

drop_last = False not supported #388

@richardrl

Description

@richardrl
  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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions