Add in a baseline for trainable interventions. It learns a binary mask over representation, and the mask is used to choose the dimension to do intervention.
Testing Done
A tutorial is included.
Checklist:
[x] My PR title strictly follows the format: [Your Priority] Your Title
Description
Add in a baseline for trainable interventions. It learns a binary mask over representation, and the mask is used to choose the dimension to do intervention.
Testing Done
A tutorial is included.
Checklist:
[Your Priority] Your Title