Skip to content

Commit

Permalink
Added discussion of PyTorch.
Browse files Browse the repository at this point in the history
  • Loading branch information
emeryberger committed May 28, 2021
1 parent 8e4685a commit 6636d95
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -239,6 +239,10 @@ manually download the `PKGBUILD` and run `makepkg -cirs` to build. Note that thi

**A:** Pass in the `--noreload` option (see https://github.com/plasma-umass/scalene/issues/178).

**Q: How do I use Scalene with PyTorch?**

**A:** Scalene works with PyTorch version 1.5.1. There's a bug in newer versions of PyTorch (https://github.com/pytorch/pytorch/issues/57185) that interferes with Scalene (discussion here: https://github.com/plasma-umass/scalene/issues/110).

# Technical Information

For technical details on Scalene, please see the following paper: [Scalene: Scripting-Language Aware Profiling for Python](https://github.com/plasma-umass/scalene/raw/master/docs/scalene-paper.pdf) ([arXiv link](https://arxiv.org/abs/2006.03879)).
Expand Down

0 comments on commit 6636d95

Please sign in to comment.