Question about visualization

[Grand-ou]

Hello.
At first, thank you for your code and paper.I really appreciate your work.
I have a problem. When visualizing training log by our own dataset, we got some question.

An error occur when i tried to call forward function of model.Error is showed as below.

And I notice that when running the code below, it shows that our model is call visionTransformer, however the output in github show that model named TSCAM, can you tell me what make it different?

I'm hoping for your answer
Thank you!

[vasgaowei]

Hi, thanks for your attention.

Maybe the "model" in your script is an instance of the class VisionTransformer. And class VisionTransformer only needs two arguments: one is self, another is input x. So I guess when declaring an instance of a TSCAM, you should use model = TSCAM(...).