missing iformer_small

[sullivantee]

when running valid.py it shows this error
Traceback (most recent call last):
File "validate.py", line 371, in
main()
File "validate.py", line 358, in main
validate(args)
File "validate.py", line 156, in validate
model = create_model(
File "C:\Users\CHU001\anaconda3\envs\pytorch\lib\site-packages\timm\models\factory.py", line 71, in create_model
raise RuntimeError('Unknown model (%s)' % model_name)
RuntimeError: Unknown model (iformer_small)

the code i ran:
python validate.py ./imagenet --model iformer_small --checkpoint checkpoint/iformer_small/model_best.pth.tar

i found out it grabs the model "iformer_small" from the directory of anaconda3\envs\pytorch\lib\site-packages\timm\models
but there seems to be no such model as "iformer_small"

[shangshang0912]

您的邮件已经收到。我会及时做出回复。

[aulaywang]

我也有同样的问题，当我在运行finetune或者validation时显示RuntimeError: Unknown model (iformer_small)

[shangshang0912]

您的邮件已经收到。我会及时做出回复。

[ArptPlank]

我也是遇到了这个问题，我运行finetune时显示没有这个模型RuntimeError: Unknown model (iformer_small)

[shangshang0912]

您的邮件已经收到。我会及时做出回复。

[TCK2001]

@sullivantee @aulaywang @ArptPlank

• Step 1
  Add this line from models.inception_transformer import * at the top of validate.py.
• Step 2
  Uncomment the code below and replace it with the following code:

create model

model = create_model(
    args.model,
    pretrained=args.pretrained,
    num_classes=args.num_classes,
    in_chans=3,
    global_pool=args.gp,
    scriptable=args.torchscript,
    mem_index=args.mem_idx,
    K=args.K)

replace to ....
model = iformer_small(pretrained="./checkpoint/iformer_small/model_best.pth")

[shangshang0912]

您的邮件已经收到。我会及时做出回复。