when i test ,RuntimeError: CUDA out of memory. Tried to allocate 10.62 MiB (GPU 0; 5.77 GiB total capacity; 4.84 GiB already allocated; 2.69 MiB free; 109.45 MiB cached

[ZZH2950228]