UVTR
UVTR copied to clipboard
Cuda out of memory when batch size was set to 1
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!