Fix the judgement of torchvision.__version__

fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

Apache License 2.0

3.23k stars 520 forks source link

Open chenjoya opened 3 years ago

chenjoya commented 3 years ago

Hi, thank you for your excellent work!

I found when torchvision.version >= 0.10.0, the original code would be False, which is unexpected.

I think we should only consider the intermediate version number of that.