search

nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[BUG] Compatibility issue with Pytorch 1.6.0 and torchvision 0.7.0 #17

Closed alphinside closed 4 years ago

alphinside commented 4 years ago

Describe the bug Error when export, possible due to removing unused graph input on nms layer

Expected behavior Should be proceed normally as in Pytorch 1.4.0