2021年7月24日
加载模型进行推理
摘要： 打算使用训练好的模型提取MS1M的人脸图像的特征进行聚类，记录一下。 模型加载 将Pytorch模型转换成ONNX模型的主要函数为torch.onnx.export() 对pytorch模型(nn.Module)对象调用named_parameters()可以得到对应的参数名(key)和参数值(va 阅读全文
posted @ 2021-07-24 17:04 渐渐的笔记本 阅读(1516) 评论(0) 推荐(1)