nodefluxio / vortex

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

Add Shape inference #84

Closed alifahrri closed 3 years ago

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? no shape inference - close #39 Related issues : ## What is the new behavior?

Checklist

alifahrri commented 4 years ago

Note:

========================================================================================= short test summary info ========================================================================================== FAILED tests/test_model.py::test_model[detection-darknet7] - RuntimeError: Sizes of tensors must match except in dimension 1. Got 40 and 20 in dimension 2 (The offending index is 1) FAILED tests/test_model.py::test_model[detection-efficientnet_b0] - RuntimeError: Given groups=1, weight of size [256, 192, 1, 1], expected input[1, 368, 40, 40] to have 192 channels, but got 368 chann... FAILED tests/test_model.py::test_model[detection-mobilenet_v2] - RuntimeError: Given groups=1, weight of size [256, 176, 1, 1], expected input[1, 352, 40, 40] to have 176 channels, but got 352 channels...

triwahyuu commented 4 years ago

I think you forgot to add sympy in the requirements

triwahyuu commented 4 years ago

I have fixed the failed tests in #85

alifahrri commented 4 years ago

I think you forgot to add sympy in the requirements

I think you forgot to add sympy in the requirements

fixed

alifahrri commented 4 years ago

can you elaborate what the scripts/model_explorer.py does? I tried to use it but I don't understand how to do anything with it.

model_explorer is similar to dataset visualizer but intended for trying model inference interactively

triwahyuu commented 4 years ago

how can I use it? after I load the model, I dont know what to do

triwahyuu commented 4 years ago

ahh, I got it now, all good. LGTM.