The stage division is not specified in the code.

[XrXie]

In vit.py, the stage division is not cleared, as well as the definitions of "flops_1stage, flops_2stage, flops_3stage, and flops_4stage". Looking forward to your reply! Thanks!!

[Cydia2018]

pruning location is a hyperparameter, modified directly in main.py, and passed into vit.py

PRUNING_LOC = [4,7,10]

[sxbkdby8]

Same Problem, but I simply ignored the undefined variable, because it seemed to be simply used for statistics.