search

schyun9212 / maskrcnn-benchmark

Converting maskrcnn-benchmark model to TorchScript or ONNX
MIT License
2 stars 0 forks source link

Replace c++ implemented custom operators to PyTorch #4

Closed schyun9212 closed 4 years ago

schyun9212 commented 4 years ago

As custom operators like nms, roi_align_forward make trouble during tracing, replacing such operators to PyTorch may be needed to make tracing easy.