mmclassification/tools/deployment
Ma Zerun c5374854aa
[Fix] Fix num_classes bug in pytorch2onnx.py (#458)
* Fix num_classes bug in pytorch2onnx.py

* Fix `num_classes=-1`.
2021-09-25 09:25:58 +08:00
..
mmcls2torchserve.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
mmcls_handler.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
onnx2tensorrt.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
pytorch2onnx.py [Fix] Fix num_classes bug in pytorch2onnx.py (#458) 2021-09-25 09:25:58 +08:00
pytorch2torchscript.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
test.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00