Skip to content

Instantly share code, notes, and snippets.

@williamFalcon
Created July 29, 2019 17:49
Show Gist options
  • Save williamFalcon/f73a0832eb14e6e12d494ff31a90f9a1 to your computer and use it in GitHub Desktop.
Save williamFalcon/f73a0832eb14e6e12d494ff31a90f9a1 to your computer and use it in GitHub Desktop.
trainer = Trainer(amp_level='O2', use_amp=False)
trainer.fit(model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment