it's 2022 now? Do we have a pytorch verion?

tianheyu927 / PCGrad

Code for "Gradient Surgery for Multi-Task Learning"

MIT License

302 stars 34 forks source link

it's 2022 now? Do we have a pytorch verion? #14

Closed Sean-asc closed 2 years ago

xgxg1314 commented 2 years ago

hello, i know a pytorch verion. pcgrad_PytorchVersion

Sean-asc commented 2 years ago

thank you for mention that! I've used some unofficial version in my code, facing the problem of lack of GPU memerory, which the gradient parameters are two big for my network.