Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Aside from the optimizer and scheduler parameters, you'll need to ensure your [Trainer] command line arguments match the DeepSpeed configuration.