Skip to content

Commit

Permalink
torchgpipe -> :mod:torchgpipe
Browse files Browse the repository at this point in the history
  • Loading branch information
sublee committed Nov 29, 2019
1 parent 7149596 commit 83893ed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/guide.rst
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ It could be hard to determine the optimal balance of a model. In particular, if
you are still designing a model, the model architecture may change over time.
In this case, we highly recommend :mod:`torchgpipe.balance` for automatic
balancing. This won't give you the optimal balance, but a good-enough balance.
Note that this is provided by `torchgpipe` package, and is not from the GPipe
Note that this is provided by :mod:`torchgpipe`, and is not from the GPipe
paper by Huang et al.

There are two balancing tools, :func:`~torchgpipe.balance.balance_by_time` and
Expand Down

0 comments on commit 83893ed

Please sign in to comment.