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

776 stars 61 forks source link

Open cporrasn opened 2 months ago

cporrasn commented 2 months ago

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