DeepSpeed
1ec34e54 - Update megatron GPT2Model

Commit
2 years ago
Update megatron GPT2Model
Author
Parents
  • tests
    • model/Megatron_GPT2
      • File
        ds_gpt2_test.sh
    • unit
      • File
        megatron_model.py