LibMTL.architecture.Cross_stitch

class Cross_stitch(task_name, encoder_class, decoders, rep_grad, multi_input, device, **kwargs)[source]

Bases: LibMTL.architecture.abstract_arch.AbsArchitecture

Cross-stitch Networks (Cross_stitch).

This method is proposed in Cross-stitch Networks for Multi-task Learning (CVPR 2016) and implemented by us.

Warning

  • Cross_stitch does not work with multiple inputs MTL problem, i.e., multi_input must be False.

  • Cross_stitch is only supported by ResNet-based encoders.