URL details: torchgpipe.readthedocs.io
URL title:
torchgpipe — torchgpipe 0.0.7 documentation
URL paragraphs:
A GPipe implementation in PyTorch . from torchgpipe import GPipe model = nn . Sequential ( a , b , c , d ) model = GPipe ( model , balance = [ 1 , 1 , 1 , 1 ], chunks = 8 ) for input in data_loader : output = model ( input ) What is GPipe? ¶ GPipe is a sc
URL last crawled:
2024-07-26
URL speed:
0.065 MB/s,
downloaded in 0.200 seconds
We found no external links pointing to this url.