YOLOv8n auto train with 100 epochs starts when in try to use the EigenCAM method

[enricopierga]

Hi, I want to use the EigenCAM technique on my YOLOv8n model, preaddestred for a detection task.
Once i declare my YOLO .pt weights and I create my tensor image, I try to call the EigenCAM method to generate activation maps with my chosen target layers. Now here's the problem: a 100 epochs automatic training starts.
Even when i put my model in .eval() mode this train starts... can someone please help me?

Here it is my image in tensors

Here it's when my train starts

These are my chosen target layers

I followed step by step the tutorial with the pretrained YOLOv5 model (EigenCAM puppies) and all things works perfectly in there...
@jacobgil

[Andreavisi1]

same problem

[atultiwari]

I am also facing the same problem.