Closed enpassanty closed 3 years ago
Thanks for the issue! need to update the DeepSpeed trainer config, but this is the preferred approach:
! pip install git+https://github.com/PytorchLightning/lightning-transformers.git@master --upgrade --quiet
! python train.py task=nlp/language_modeling dataset=nlp/language_modeling/wikitext trainer/plugins=deepspeed
traceback: