https://github.com/YaNgZhAnG-V5/informationbottleneck/blob/aa728708eeedd43355168172b5f37377a27d0614/iba/models/bottlenecks/vision_input_iba.py#L80 As marked in the code, we need to consider the case where `reverse_lambda = True` in `VisionInputIBA`
https://github.com/YaNgZhAnG-V5/informationbottleneck/blob/aa728708eeedd43355168172b5f37377a27d0614/iba/models/bottlenecks/vision_input_iba.py#L80
As marked in the code, we need to consider the case where
reverse_lambda = TrueinVisionInputIBA