-
- Downloads
PyTorch 1.3 Support (#426)
* summary_graph.py:
* Change ONNX op.uniqueName() to op.debugName()
* Removed scope-naming workaround which isn't needed in PyTorch 1.3
* Tests:
* Naming of trace entries changed in 1.3. Fixed SummaryGraph unit
test that checked that
* Adjusted expected values in full_flow_tests
* Adjusted tolerance in test_sim_bn_fold
* Filter some new warnings
Showing
- README.md 5 additions, 2 deletionsREADME.md
- distiller/summary_graph.py 8 additions, 25 deletionsdistiller/summary_graph.py
- requirements.txt 2 additions, 2 deletionsrequirements.txt
- tests/full_flow_tests.py 9 additions, 9 deletionstests/full_flow_tests.py
- tests/test_learning_rate.py 1 addition, 0 deletionstests/test_learning_rate.py
- tests/test_post_train_quant.py 3 additions, 0 deletionstests/test_post_train_quant.py
- tests/test_sim_bn_fold.py 1 addition, 1 deletiontests/test_sim_bn_fold.py
- tests/test_summarygraph.py 8 additions, 17 deletionstests/test_summarygraph.py
Loading
Please register or sign in to comment