摘要: 测试模型用之前文章训练的Alexnet模型。 首先将pth文件转为onnx文件: import torch import torch.nn as nn # 自定义AlexNet模型 class AlexNet(nn.Module): def __init__(self): super(AlexNet 阅读全文
posted @ 2023-10-19 22:56 Dsp Tian 阅读(86) 评论(0) 推荐(0) 编辑
摘要: 用pip install xxx安装库时有时候会遇到下面提示: To fix this you could try to: 1. loosen the range of package versions you've specified 2. remove package versions to a 阅读全文
posted @ 2023-10-19 22:12 Dsp Tian 阅读(1928) 评论(0) 推荐(0) 编辑
摘要: 运行时忽略掉警告: import warnings warnings.filterwarnings("ignore") 阅读全文
posted @ 2023-10-19 20:59 Dsp Tian 阅读(22) 评论(0) 推荐(0) 编辑