/cross-stitch-pytorch

This project is a PyTorch algorithm implementation of Misra, I., Shrivastava, A., Gupta, A., & Hebert, M. (2016). Cross-stitch networks for multi-task learning. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (pp. 3994-4003).

MIT LicenseMIT

Stargazers