nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[BUG] Compatibility issue with Pytorch 1.6.0 and torchvision 0.7.0

alphinside opened this issue · comments

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