The SampleTracker (used to track the result of adversarial attacks) crashes the execution whether the input samples are different from images. I would refactor this, introducing a generic SampleTracker for all input, and then a dedicated ImageSampleTracker that can be visualized.