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 ...
When you purchase through links on our site, we may earn an affiliate commission. Here’s how it works.
Some results have been hidden because they may be inaccessible to you
Show inaccessible results