UVTR icon indicating copy to clipboard operation
UVTR copied to clipboard

Cuda out of memory when batch size was set to 1

Open RhythmOfTheRain-Byte opened this issue 1 year ago • 2 comments

When using RTX4090 to train the multimodal model, the batch size was set to 1. While the code still reports an error of "cuda out of memory". How can I solve this problem. Thank you!

RhythmOfTheRain-Byte avatar May 25 '23 13:05 RhythmOfTheRain-Byte