代码复现出现结构问题

Question

复现MetaFormer出现一些问题
项目：https://github.com/sail-sg/poolformer
报错如下：KeyError: "RetinaNet: 'poolformer_s12_feat is not in the models registry'"
类似的问题在网上也找了很久，没有实质性进展。

Answer 1

这个错误提示通常表明你的模型名称或者其中的某个层名称没有被正确注册。在 PyTorch 中，模型和层的注册通常在 torchvision.models 中进行。
你可以尝试以下步骤来解决这个问题：

1. 确保你正在使用的模型名称是正确的，没有拼写错误，并且该模型已经在 torchvision.models 中注册。你可以在 PyTorch 官方文档 中查看所有已注册的模型。
2. 如果你的模型是自定义的，你需要确保在你的模型代码中，使用 torchvision.models.register() 函数将你的模型注册到 torchvision.models 中。例如：

   import torchvision.models as models
   from . import my_model
   
   models.register(my_model)
   

3. 如果问题仍然存在，你可以尝试在你的代码中导入 torchvision.models 模块，并打印出所有的注册模型名称，看看是否有你正在使用的模型名称。例如：

   import torchvision.models as models
   print(models.__all__)