Torch 1.6 compatibility

alifahrri commented 4 years ago

Type of changes

Please check the type of change your PR introduces: - [x] Bugfix - [ ] Feature - [ ] Code style update (formatting, renaming) - [ ] Refactoring (no functional changes, no api changes) - [ ] Build related changes - [ ] Documentation content changes - [ ] Other (please describe): ## What is the current behavior?

close #17

Related issues :

What is the new behavior?

now works with torch 1.6.0

Checklist

[ ] Unit tests pass locally with my changes
[ ] I have added tests that prove my fix is effective or that my feature works
[ ] I have added necessary documentation (if appropriate)
[ ] Any dependent changes have been merged and published in downstream modules
[x] I have filled the [Unreleased] Changelog

alphinside commented 4 years ago

error at test_postprocess_export.py

E           onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid Feed Input Name:iou_threshold

FAILED tests/test_postprocess_export.py::PostprocessTest::test_yolo_postprocess_onnx - onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNX...

triwahyuu commented 4 years ago

all good, but while testing it I found that efficientnet original model is failed to be exported, I guess it needs another issue for that. probably something to do with the activation function I changed.

alifahrri commented 4 years ago

hmm ok

triwahyuu commented 4 years ago

I guess it's okay for this to be merged

nodefluxio / vortex