walkwithfastai.github.io icon indicating copy to clipboard operation
walkwithfastai.github.io copied to clipboard

Published 20 hours ago • verified publisher icon walkwithfastai

BBoxBlock issue with PyTorch 1.8.x

Open Stijnp opened this issue 3 years ago • 1 comments

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]

Stijnp avatar Jun 11 '21 20:06 Stijnp

I am facing similar problem. Previously bounding box model was working well - now it is returning all zeros

johnyquest7 avatar Jul 14 '21 01:07 johnyquest7