Separate attention-map capture from gradient hooks - #1232

Separate attention-map capture from gradient hooks#1232

LIU-kaiyu wants to merge 1 commit into

sunlabuiuc:mastersunlabuiuc/PyHealth:masterfrom

LIU-kaiyu:liu-kaiyu/1166-attention-captureLIU-kaiyu/PyHealth:liu-kaiyu/1166-attention-captureCopy head branch name to clipboard