From 83893ed1e5158d37acb433b830b28073e49fa8dd Mon Sep 17 00:00:00 2001 From: Heungsub Hans Lee Date: Fri, 29 Nov 2019 18:56:01 +0900 Subject: [PATCH] `torchgpipe` -> :mod:`torchgpipe` --- docs/guide.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/guide.rst b/docs/guide.rst index db1183b..e2fe436 100644 --- a/docs/guide.rst +++ b/docs/guide.rst @@ -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