Sandeep G
Sandeep G
> @jfzhang95 I run inference.py with pretrained model: > Traceback (most recent call last): > File "inference.py", line 78, in > main() > File "inference.py", line 32, in main >...
> @umbertogriffo thanks a lot for your reply and for the clarification. I tried to use your code with a few modifications for multi-label classification. After looking at the code...
@ccl-private check this one to convert torch to onnx. https://github.com/onnx/models/blob/master/vision/object_detection_segmentation/retinanet/dependencies/retinanet-export.py
I changed generator = torch.Generator() to generator =torch.Generator(device='cuda') in torch\utils\data\sampler.py in line 115. It worked for me.
@sja9527 refer this code https://github.com/Sandeep418/parseq_onnx/blob/main/onnx_inference.cpp