torchgpipe是GPipe的一个PyTorch中实现,它针对CUDA而不是TPU进行了优化

torchgpipe是GPipe的一个PyTorch中实现,它针对CUDA而不是TPU进行了优化