Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
A model trained with ZeRO-2 saves the pytorch_model.bin weights in fp16.