search

iscyy / yoloair

🔥🔥🔥 专注于YOLOv5，YOLOv7、YOLOv8、YOLOv9改进模型，Support to improve backbone, neck, head, loss, IoU, NMS and other modules🚀
https://github.com/iscyy/yoloair
GNU General Public License v3.0
2.47k stars 428 forks source link

trainig completed but getting error at last epoch , yolov7 repvgg -at fuse_conv_and_bn #143

Closed akashAD98 closed 1 year ago

akashAD98 commented 1 year ago

im able to complete training,but at last epoch I'm getting below error, also for doing inference I'm again getting error, how can we fix this error

   fusedconv.weight.copy_(torch.mm(w_bn, w_conv).view(fusedconv.weight.shape))
RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
image