Udpate in 2025: we recently updated the code to support Python 3.11 and PyTorch 2.4. It also supports DDP for single-machine multiple-GPU training. (Please use torchrun --nproc_per_node=4 train.py ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results
Feedback