Pythontorchgpipe是GPipe的一个PyTorch中实现它针对CUDA而不是TPU进行了优化
torchgpipe是GPipe的一个PyTorch中实现，它针对CUDA而不是TPU进行了优化
下载地址
用户评论
