My Python class instance is breaking after loading a pytorch weights file
I had an AutoAnnotation script for yolo v7, which is part of a annotation tool. When I call an api, it take the dataset, weights path and all required details, and publish it via rabbitmq to a gpu server, In gpu server, there is a call back instance, which is loading the model to do auto annotation.
I am loading the model using YOLOv7 –>models–>experimental–>attempt_load function, where Iam successfully loading the function, but