作者您好，新人入门，请求解答

mxin262 / SwinTextSpotter

Pytorch re-implementation of Paper: SwinTextSpotter: Scene Text Spotting via Better Synergy between Text Detection and Text Recognition (CVPR 2022)

275 stars 42 forks source link

Open Haoddan opened 11 months ago

Haoddan commented 11 months ago

在尝试training的过程中，torch.cuda.OutOfMemoryError: CUDA out of memory.

我应该在哪个文件里修改参数，以保证可以正常运行呢

希望回复，感谢

mxin262 commented 11 months ago

可以把batchsize减小试试