torchMTL:pytorch中用于多任务学习的轻量级模块

pytorch中用于多任务学习的轻量级模块。 torchmtl试图帮助您以最小的努力来组成模块化的多任务架构。 您所需要的只是字典列表，您可以在其中定义图层以及它们如何相互构建。 由此， torchmtl构造了一个元计算图，该图在创建的MTLModel每个正向传递中执行。 为了组合来自多层的输出，提供了简单的。 安装 可以通过pip安装torchmtl ： pip install torchmtl 快速入门（或在查找示例） 假设您要针对三个任务训练网络，如下所示。 要使用torchmtl构建这样的体系结构，只需定义以下列表 tasks = [ { 'name' : "Embed1" , 'layers' : Sequential ( * [ Linear ( 16 , 32 ), Linear ( 32 , 8 )]