Skip to content

Instantly share code, notes, and snippets.

@williamFalcon
Created November 15, 2019 16:03
Show Gist options
  • Save williamFalcon/260a4024115bafadd8561ab0b4acc7e0 to your computer and use it in GitHub Desktop.
Save williamFalcon/260a4024115bafadd8561ab0b4acc7e0 to your computer and use it in GitHub Desktop.
def configure_optimizers(self):
return torch.optim.Adam(self.transformer.parameters(), self.hparams.lr)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment