Local variable 'input' referenced before assignment

[laohanlin]

I want to run your code. But no matter how I modify it, timm keeps telling me local variable 'input' referenced before assignment. As is shown in the picture:

[YuanLi]

What's the timm version you used?
Please installl timm 3.4.0 by: pip install timm==0.3.4