Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/flash text lightning trainer.py

Created February 24, 2021 09:19
Show Gist options
  • Save aribornstein/c99feee37b06e211acdb509eae12bf59 to your computer and use it in GitHub Desktop.
Save aribornstein/c99feee37b06e211acdb509eae12bf59 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
flash text lightning trainer.py
trainer = flash.Trainer(gpus=8)
trainer = flash.Trainer(gpus=8, num_nodes=32)
trainer = flash.Trainer(tpu_cores=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment