TypeError: repvit_m0_9() got an unexpected keyword argument 'pretrained_cfg'

THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything

https://arxiv.org/abs/2307.09283

Apache License 2.0

799 stars 60 forks source link

Closed ZackGoing closed 7 months ago

ZackGoing commented 8 months ago

我运行train.sh，报错了，请问怎么解决这个问题，谢谢

jameslahm commented 7 months ago

您好，可以在repvit_m0_9()入参最后加入**kwargs，类似这样

def repvit_m0_9(..., **kwargs)