Hello, does it work for a detection network with branches?

kakaobrain / torchgpipe

A GPipe implementation in PyTorch

https://torchgpipe.readthedocs.io/

BSD 3-Clause "New" or "Revised" License

814 stars 98 forks source link

Closed yangpc615 closed 5 years ago

yangpc615 commented 5 years ago

Must be the network of the sequence structure?

model = cast(nn.Sequential, model) model = GPipe(model, balance, devices=devices, chunks=chunks)

sublee commented 5 years ago

torchgpipe supports only nn.Sequential modules. Users have a responsibility to design a module fitting into nn.Sequential.

yangpc615 commented 5 years ago

OK， Thank you.