Use this library for CNN networks like Unet

tunib-ai / parallelformers

Parallelformers: An Efficient Model Parallelization Toolkit for Deployment

https://tunib-ai.github.io/parallelformers

Apache License 2.0

779 stars 61 forks source link

Open cporrasn opened 5 months ago

cporrasn commented 5 months ago

Is it possible to use this library for CNN networks implemented with pytorch? Can you show me an example?