is it available to get gradient when training??

[SPA-junghokim]

What is the feature?

I want to apply PCGrad [1] on my model when training in mmdetection3d code. but it has to need to get gradient in each module and it is required to customize gradient. is it available to get gradient when training??

thx

[1] Yu, Tianhe, et al. "Gradient surgery for multi-task learning." Advances in Neural Information Processing Systems 33 (2020): 5824-5836. https://arxiv.org/pdf/2001.06782.pdf

Any other context?

No response