Gradient class activation mapping for regression tasks

Hi, I’m trying to visualize and create heatmaps using saliency, grad cams, etc for my age prediction task. I found the blog post on visualization very useful but I see that the example is a classification task. Any suggestions on how I can use it for a regression task?

© Copyright 2013-2019 Analytics Vidhya