文件名称：torchMTL:pytorch中用于多任务学习的轻量级模块
文件大小：112KB
文件格式：ZIP
更新时间：2024-05-30 12:57:00
machine-learning framework deep-learning python3 pytorch
pytorch中用于多任务学习的轻量级模块。 torchmtl试图帮助您以最小的努力来组成模块化的多任务架构。 您所需要的只是字典列表，您可以在其中定义图层以及它们如何相互构建。 由此， torchmtl构造了一个元计算图，该图在创建的MTLModel每个正向传递中执行。 为了组合来自多层的输出，提供了简单的。 安装 可以通过pip安装torchmtl ： pip install torchmtl 快速入门（或在查找示例） 假设您要针对三个任务训练网络，如下所示。 要使用torchmtl构建这样的体系结构，只需定义以下列表 tasks = [ { 'name' : "Embed1" , 'layers' : Sequential ( * [ Linear ( 16 , 32 ), Linear ( 32 , 8 )]
【文件预览】：
torchMTL-main
----torchmtl()
--------wrapping_layers.py(1KB)
--------__init__.py(144B)
--------model_builder.py(6KB)
----images()
--------torchmtl_graph.png(13KB)
--------example.png(14KB)
--------torchmtl_logo.png(11KB)
----examples()
--------simple_triple_regression.ipynb(72KB)
--------uncertainty_weighted_loss.ipynb(30KB)
----LICENSE(1KB)
----README.md(6KB)
----tests()
--------test_meta_comp_graph.py(5KB)
--------__init__.py(0B)
----.gitignore(2KB)
----pyproject.toml(523B)