Error in inference

[xpixelgroup]

Error in inference

[0x3f3f3f3fun]

The reason may be cuda out of memory. DiffBIR requires at least 10G gpu memory for inference at a resolution of 512 now.

[ZhouYFeng]

One V100 32G used for the inference. But still failed.

Seems like the data dimensions do not match the actual data dimensions when the nn.Linear is defined, and need to be checked and modified.

The cuda version of mine is 11.4. I am not sure if it is okay for the env requirement provieded in DiffBIR.

[0x3f3f3f3fun]

Since the program automatically handles data shape, I don't think it's a data shape issue. However, if you still have doubts, you can test it with some 512x512 images.
After searching on Google, I found that most people solved this problem by updating their environment:

• ultralytics/yolov5#10046 (comment)
• https://stackoverflow.com/questions/68571902/how-to-fix-runtimeerror-cuda-error-cublas-status-invalid-value-when-calling-cub

So, you can try reinstalling your environment, especially PyTorch and CUDA Toolkit, strictly following the instructions in the README.md.