iamfaith / pytorch-model-parallel Goto Github PK
View Code? Open in Web Editor NEWThis project forked from bindog/pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch