Open Papenkov opened 2 years ago
I want to use "export.py" file to convert detector to onnx but I got a mistake [KeyError: 'model' ] in torch.load(w, map_location=map_location)['model'].float().fuse().eval()". This code in function attempt_load in "experimental.py" file.
It maybe in "torch.load" hasn't key " model",but why?
Hello, mate @elyha7 . I'm trying to convert face detector to onnx via your "export.py" file and have some issue: after convertion output sizes don't fit with output sizes from loaded model ("model = YoloDetector( ... )")