About CUDA out of Memory And Multiple GPU calls

[Santasu28]

Hi！
When i run your code ， i use a computer with 4*3090， but i meet the problem with CUDA out of memory when training reach about 33%。I see you assign gpu0 ， so i can‘t use other gpu, only one gpu works, how can i use all the GPU, how to modify the code？Thank you for your help