update

[Datou98]

Hello, pytorch-unet has been updated to version 3.0 and supports multi-classification. Can you update the code of tensorrt-unet? thank you very much

[YuzhouPeng]

Hello, pytorch-unet has been updated to version 3.0 and supports multi-classification. Can you update the code of tensorrt-unet? thank you very much

Unfortunately I don't have any available time to update code now, u can try to fork my code and edit the layer if possible :)

[YuzhouPeng]

Hello, pytorch-unet has been updated to version 3.0 and supports multi-classification. Can you update the code of tensorrt-unet? thank you very much

https://github.com/wang-xinyu/tensorrtx/tree/master/unet maybe you can check newest 8.x tensorrt version unet and have a try