Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
reinforcement-learning deep-learning deep-reinforcement-learning pytorch mnist rl reimplementation multi-task-learning cifar-100 multi-task-reinforcement-learning multi-task-rl pytorch-pcgrad gradient-surgery mulit-mnist
-
Updated
Jun 22, 2021 - Python