captum icon indicating copy to clipboard operation
captum copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Add support for tuples of layers (instead of strictly list) in LayerLRP

Open alexandrelarouche opened this issue 1 year ago • 2 comments

It seems to me there is no reason why LayerLRP is restricted to lists only. It seems any ordered iterable should do the trick here.

  • Add checks for tuples instead of just lists of layers in LayerLRP
  • Add new type for hinting / documentation -> ModuleOrModuleListOrModuleTuple

Edit: Reference issue #1329

alexandrelarouche avatar Aug 15 '24 19:08 alexandrelarouche