解决:torch.cuda.OutOfMemoryError: CUDA out of memory.

1. 场景:在改进yolov7过程中出现错误。

2. 解决:可以尝试将batch_size调小。

猜你喜欢

转载自blog.csdn.net/weixin_44813538/article/details/130619845