search

V2AI / Det3D

World's first general purpose 3D object detection codebse.
https://arxiv.org/abs/1908.09492
Apache License 2.0
1.49k stars 298 forks source link

Not support pytorch 1.1 #53

Closed s-ryosky closed 4 years ago

s-ryosky commented 4 years ago

Instructions To Reproduce the Issue:

  1. what changes you made (git diff) or what code you wrote After the commit b567905d961bad4eef3fcfd2bc8253826a882536

  2. what exact command you run: Det3D/tools/train.py

  3. what you observed (including the full logs):

    AttributeError: 'Tensor' object has no attribute 'bool'

    is occurred in the following line. https://github.com/poodarchu/Det3D/blob/56402d4761a5b73acd23080f537599b0888cce07/det3d/models/bbox_heads/mg_head.py#L1038

  4. please also simplify the steps as much as possible so they do not require additional resources to Run on PyTorch 1.1

Expected behavior:

PyTorch 1.1 or higher is recommended in readme. But PyTorch==1.1 doesn't support to the 'bool' attribute.

poodarchu commented 4 years ago

dir_labels.byte()