We use python 3.11/CUDA 12.4/torch 2.5.1 for implementation. We train our models on 8 NVIDIA A100 GPUs with 80G memory, please make sure that your VRAM is sufficient to avoid the potential OOM issues ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results
Feedback