Skip to content

Instantly share code, notes, and snippets.

@yang-zhang
Last active July 4, 2020 12:51
Show Gist options
  • Save yang-zhang/cdf4dd828273040d456cf5254a55a205 to your computer and use it in GitHub Desktop.
Save yang-zhang/cdf4dd828273040d456cf5254a55a205 to your computer and use it in GitHub Desktop.
boilerplates
  • pin_memory=True
  • sigmoid used in cross_entropy_loss
  • set_seed before fold and train
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment