-
- 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