WeiChengTseng/Pytorch-PCGrad
Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
PythonBSD-3-Clause
Issues
- 2
gradient data type error
#18 opened by ryusangwon - 3
Gradient visualization
#6 opened by chengjianhong - 10
CUDA out of memory
#2 opened by remyhuang - 4
reproducing your result
#13 opened by NoaGarnett - 0
Is there any RL implements?
#17 opened by zerui981105 - 1
多GPU运行时报错
#16 opened by yongdajiaozhixiang - 2
support for amp?
#15 opened by HanGuangXin - 1
reduction is always 'mean'
#14 opened by tkbadamdorj - 1
Question about using PCGrad with Uncertainity based weights as discussed in the paper for CIFAR dataset
#11 opened by madhukiranets - 3
- 0
my fault orz
#12 opened by kunkun0w0 - 0
Did anyone validate the PCGrad algorithm?
#10 opened by jiangdada1221 - 2
- 2
g_j.norm() is equal to zero?
#7 opened by chengjianhong - 3
About dataset
#8 opened by chengjianhong - 12
support distributed training
#5 opened by chengjianhong - 2
went wrong in function _project_conflicting
#1 opened by leonzgtee