Skip to content

BBoxBlock issue with PyTorch 1.8.x #39

@Stijnp

Description

@Stijnp

There's an issue when running nbs/course2020/vision/06_Object_Detection.ipynb using PyTorch 1.8.x.

Creating a DataBlock (with BBoxBlock and BBoxLblBlock as blocks) using the getters specified will cause some issues.

Running DataBlock.summary() will give the following error:
TypeError: no implementation found for 'torch.Tensor.getitem' on types that implement torch_function: [TensorMultiCategory, TensorBBox]

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions