摘要:
测试模型用之前文章训练的Alexnet模型。 首先将pth文件转为onnx文件: import torch import torch.nn as nn # 自定义AlexNet模型 class AlexNet(nn.Module): def __init__(self): super(AlexNet 阅读全文
摘要:
用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 阅读全文
摘要:
运行时忽略掉警告: import warnings warnings.filterwarnings("ignore") 阅读全文