model-parallel

There are 0 repository under model-parallel topic.

cavalleria / cavaface
face recognition training project(pytorch)
face-recognition pytorch network loss arcnegface arcface resnest attention-irse apex randaugment mixup dataparallel model-parallel resnet-irse curricularface cosface amsoftmax circleloss ghostnet efficientnet
Language:Python 453
PaddlePaddle / PLSC
Paddle Large Scale Classification Tools，supports ArcFace, CosFace, PartialFC, Data Parallel + Model Parallel. Model includes ResNet, ViT, Swin, DeiT, CaiT, FaceViT, MoCo, MAE, ConvMAE, CAE.
face-recognition arcface cosface partial-fc data-parallel model-parallel large-scale paddlepaddle paddle distributed-training hight-speed deit resnet vit facevit cait moco-v3 mae convmae swin-transformer
Language:Python 144
bindog / pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
model-parallel distributed-training half-precision re-id pytorch
Language:Python 76
4paradigm / OpenEmbedding
OpenEmbedding is an open source framework for Tensorflow distributed training acceleration.
tensorflow-training distributed-training tensorflow embedding-layers model-parallel parameter-server
Language:C++ 28
knagrecha / hydra
Execution framework for multi-task model parallelism. Enables the training of arbitrarily large models with a single GPU, with linear speedups for multi-gpu multi-task execution.
deep-learning pytorch large-scale-machine-learning distributed-computing model-parallel task-parallel machine-learning gpu systems-engineering machine-learning-library
Language:Python 20

model-parallel

cavalleria / cavaface

PaddlePaddle / PLSC

bindog / pytorch-model-parallel

4paradigm / OpenEmbedding

knagrecha / hydra