文件名称：Pytorch-PCGrad:Pytorch重新实现“多任务学习的梯度手术”
文件大小：13KB
文件格式：ZIP
更新时间：2024-05-24 08:44:10
reinforcement-learning deep-learning deep-reinforcement-learning pytorch mnist
PyTorch-PCGrad 该存储库提供了PyTorch 1.6.0中的重新实现代码。 设置 通过以下方式安装所需的软件包： pip install -r requirements.txt 用法 import torch import torch . nn as nn import torch . optim as optim from pcgrad import PCGrad # wrap your favorite optimizer optimizer = PCGrad ( optim . Adam ( net . parameters ())) losses = [...] # a list of per-task losses assert len ( losses ) == num_tasks optimizer . pc_backward ( losses ) # c
【文件预览】：
Pytorch-PCGrad-master
----utils.py(354B)
----data()
--------multi_mnist.py(13KB)
----.github()
--------ISSUE_TEMPLATE()
----pcgrad.py(6KB)
----LICENSE(1KB)
----depreciate()
--------pcgrad_ori.py(4KB)
----net()
--------lenet.py(1KB)
----requirements.txt(21B)
----.gitignore(2KB)
----main_multi_mnist.py(3KB)
----README.md(2KB)