Best Annotation tool with Pytorch model (Active Learning)

I am looking on feedback on Image annotation tools that support Pytorch model in the loop.
My task is segmenting images with polygons masks. I am currently using CVAT but have not been able to connect my UNET to CVAT.
What tools are people using to annotate images?
Any best practices I should know?
Thanks!