kakaobrain/torchgpipe · AI Repo StatsKAkakaobrain/torchgpipe
A GPipe implementation in PyTorch
Training & fine-tuningPython BSD-3-Clause 99created 2019-05-10 #checkpointing#deep-learning#gpipe#model-parallelism#parallelism#pipeline-parallelism#pytorch