edith-wq

I also tried to write the training code, but when I run the training code with A100 40G, I get CUDA oom. What is the GPU memory you have or...