chr5tphr / zennit

Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.
Other
190 stars 32 forks source link

Docs: Use modified_model in Composites example #64

Closed rodrigobdz closed 2 years ago

rodrigobdz commented 2 years ago

In the with statement, the variable modified_model was not used to compute the output and the original model was used instead.

chr5tphr commented 2 years ago

Hey Rodrigo,

thank you for the fix! Just as a side note, both model and modified_model describe the same object, since composites (and also attributors) temporarily change models in-place for the duration of the context. The context variable is used for clarity.

rodrigobdz commented 2 years ago

@chr5tphr Glad to help! I see, thanks for the clarification.