eora-ai / torchok

Production-oriented Computer Vision models training pipeline for common tasks: classification, segmentation, detection and representationšŸ„¤
https://torchok.readthedocs.org
Apache License 2.0
54 stars 6 forks source link

Bug Fix checkpoint with onnx top k #154

Closed VladislavPatrushev closed 1 year ago