文件名称:AuxiLearn:内隐差异辅助学习的正式实施[ICLR 2021]
文件大小:1.02MB
文件格式:ZIP
更新时间:2024-05-24 12:53:05
computer-vision semantic-segmentation optimization-algorithms multitask-learning auxiliary-learning
AuxiLearn-通过隐式微分的辅助学习 该存储库包含源代码,以支持Aviv Navon *,Idan Achituve *,Haggai Maron,Gal Chechik †和Ethan Fetaya † (ICLR 2021)撰写的论文《 。 链接 安装 克隆仓库并安装: git clone https://github.com/AvivNavon/AuxiLearn.git cd AuxiLearn pip install . 用法 给定一个双层优化问题,其中高层参数(即辅助参数)仅隐式影响高层目标,您可以使用auxilearn通过隐式微分来计算高层梯度。 您将需要使用的主要代码组件是auxilearn.optim.MetaOptimizer 。 它是PyTorch优化器的包装,后者通过隐式差异更新其参数。 代码示例 我们假设两种型号, primary_model和auxil
【文件预览】:
AuxiLearn-master
----setup.py(1KB)
----.gitignore(92B)
----requirements.txt(152B)
----resources()
--------framework.png(104KB)
--------nyu_losses_and_gradients_tight.png(907KB)
----experiments()
--------__init__.py(0B)
--------utils.py(1KB)
--------oxford_pet()
--------weight_methods.py(8KB)
--------nyuv2()
----LICENSE(1KB)
----README.md(4KB)
----auxilearn()
--------__init__.py(0B)
--------hypernet.py(15KB)
--------implicit_diff.py(2KB)
--------optim.py(2KB)