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 beFalse
.Cross_stitch
is only supported by ResNet-based encoders.