No module named 'nasnet_mobile'?

报错

python trans.py
Traceback (most recent call last):
  File "trans.py", line 6, in <module>
    from nasnet_mobile import nasnetamobile
ModuleNotFoundError: No module named 'nasnet_mobile'

代码

#coding: utf-8
import torch
#import torchvision
# 1.导入PyTorch模型定义

from nasnet_mobile import nasnetamobile
# 2.指定输入大小的shape
dummy_input = torch.randn(1, 3, 224, 224)
 
# 3. 构建PyTorch model
model = nasnetamobile(121,pretrained=False)
# 4. 载入模型参数
model.load_state_dict(torch.load('L:/Yolov5_DeepSort_Pytorch-master/yolov5/weights/best.pt', map_location='cpu'))
 
# 5.导出onnx模型文件
torch.onnx.export(model, dummy_input, "nasnet.onnx", verbose=True)

求

nasnet_mobile 模块不存在

参考http://www.10qianwan.com/articledetail/627516.html

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632

非常感谢您使用有问必答服务，为了后续更快速的帮您解决问题，现诚邀您参与有问必答体验反馈。您的建议将会运用到我们的产品优化中，希望能得到您的支持与协助！

速戳参与调研>>>https://t.csdnimg.cn/Kf0y