Skip to content

Instantly share code, notes, and snippets.

@szagoruyko

szagoruyko/PyTorch_tasks.md

Last active March 10, 2017 12:00
Show Gist options
  • Save szagoruyko/3cf31e6cb452c431a8e0b3a3b14f45b4 to your computer and use it in GitHub Desktop.
Save szagoruyko/3cf31e6cb452c431a8e0b3a3b14f45b4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
PyTorch_tasks.md
  • BiasAdd function (with CUDNN and not)
  • PTX execution in pytorch (cutorch-rtc equivalent)
  • cublas<C>dgmm wrappers for torch and cutorch
  • zeros_like, new_like, ones_like
  • docs for torch.nn.functional
  • F.pad: constant, reflect, replicate
  • gradcheck
  • model.train(False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment