Index
B
|
C
|
D
|
F
|
G
|
I
|
N
|
P
|
S
|
T
|
V
B
balance (torchgpipe.GPipe attribute)
balance_by_size() (in module torchgpipe.balance)
balance_by_time() (in module torchgpipe.balance)
C
checkpoint (torchgpipe.GPipe attribute)
chunks (torchgpipe.GPipe attribute)
D
devices (torchgpipe.GPipe attribute)
F
forward() (torchgpipe.GPipe method)
G
GPipe (class in torchgpipe)
I
is_checkpointing() (in module torchgpipe)
is_recomputing() (in module torchgpipe)
isolate() (torchgpipe.skip.skippable.Skippable method)
N
Namespace (class in torchgpipe.skip)
P
pop() (in module torchgpipe.skip)
S
skippable() (in module torchgpipe.skip)
stash() (in module torchgpipe.skip)
T
torchgpipe (module)
torchgpipe.balance (module)
torchgpipe.skip (module)
V
verify_skippables() (in module torchgpipe.skip)
torchgpipe
GPipe for PyTorch
Navigation
Understanding GPipe
User Guide
API
Benchmarks
Changelog
Source Code
Original Paper
Related Topics
Documentation overview
Quick search