Skip to content

Commit

Permalink
Create TARGETS for captum/concept (pytorch#1257)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#1257

Split captum/concept to a separate TARGETS

Reviewed By: vivekmig

Differential Revision: D55222411

fbshipit-source-id: 961f3ad9abc70edced998e61c852791a0d2c56af
  • Loading branch information
yucu authored and facebook-github-bot committed Mar 22, 2024
1 parent 38ed993 commit f8dc1b7
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion captum/concept/_core/tcav.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,9 @@
from captum._utils.av import AV
from captum._utils.common import _format_tensor_into_tuples, _get_module_from_name
from captum._utils.typing import TargetType, TensorOrTupleOfTensorsGeneric
from captum.attr import LayerActivation, LayerAttribution, LayerGradientXActivation
from captum.attr._core.layer.layer_activation import LayerActivation
from captum.attr._core.layer.layer_gradient_x_activation import LayerGradientXActivation
from captum.attr._utils.attribution import LayerAttribution
from captum.concept._core.cav import CAV
from captum.concept._core.concept import Concept, ConceptInterpreter
from captum.concept._utils.classifier import Classifier, DefaultClassifier
Expand Down

0 comments on commit f8dc1b7

Please sign in to comment.