GitHub / bindog / pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/bindog%2Fpytorch-model-parallel
Stars: 74
Forks: 20
Open issues: 2
License: None
Language: Python
Size: 85 KB
Dependencies parsed at: Pending
Created at: over 5 years ago
Updated at: over 1 year ago
Pushed at: almost 5 years ago
Last synced at: about 1 year ago
Topics: distributed-training, half-precision, model-parallel, pytorch, re-id
Loading...