Siria Yerlany Sadeddin Mejia

Did someone have an issue with yolor-ssss-dwt.pt export to ONNX? I can export all of the models, but not this one. I got the error "Couldn't export Python operator AFB2D",...

``` # Input img = torch.randn((opt.batch_size, 3, *opt.img_size)) # image size(1,3,320,192) iDetection # Load PyTorch model attempt_download(opt.weights) model = torch.load(opt.weights, map_location=torch.device('cpu'))['model'].float() from onnxsim import simplify # Update model for k,...

Hi! can you tell me how did you test it so I can also try it. I tried for example ```import sys #python test.py --img-size 1536 --weights ./yolov4-p7.pt sys.path.append('./') #...

I don't know why but it wasn't working using Cuda... I am using CPU device and started working fine.

Use python 3.6 and an older version of pytorch. conda install pytorch==1.2.0 torchvision==0.4.0 -c pytorch

try https://github.com/Yuxiang1995/ICDAR2021_MFD/blob/main/demo/inference_demo.ipynb it worked for me