84 citations · 116 across the 2 of their papers we have counts for
2 papers
cs.DC2020★ 32 cited
torchgpipe: On-the-fly Pipeline Parallelism for Training Giant Models
Chiheon Kim, Heungsub Lee, Myungryong Jeong +5
We design and implement a ready-to-use library in PyTorch for performing micro-batch pipeline parallelism with checkpointing proposed by GPipe (Huang et al., 2019). In particular,…
cs.LG2019★ 84 cited
Scalable Neural Architecture Search for 3D Medical Image Segmentation
Sungwoong Kim, Ildoo Kim, Sungbin Lim +5
In this paper, a neural architecture search (NAS) framework is proposed for 3D medical image segmentation, to automatically optimize a neural architecture from a large design space…