CUDA out of memory

[keep123ing]

租服务器复现，批量大小改成了1，但是一直显示gpu容量不够，换成3090和4090也不行，还能怎么办啊

[huanghuada]

解决了吗xd

[Hmbb0606]

可以在超参数中调小dims参数解决，例如dims=48，24。