将训练好的模型保存到网页中并提供下载链接
时间: 2023-09-11 18:08:06 浏览: 191
网页 模版 下载
要将训练好的模型保存到网页中并提供下载链接,可以使用 Flask 框架。
首先,需要将模型保存为文件。可以使用 PyTorch 中的 `torch.save()` 方法将模型保存为文件,并使用 Flask 提供的 `send_file()` 方法将文件发送给用户。
以下是示例代码:
```python
from flask import Flask, send_file
import torch
app = Flask(__name__)
# 加载模型
model = torch.load('model.pth')
# 定义下载链接
@app.route('/download')
def download():
# 将模型保存为文件
torch.save(model, 'model_download.pth')
# 发送文件给用户
return send_file('model_download.pth', as_attachment=True)
if __name__ == '__main__':
app.run()
```
在浏览器中访问 `http://localhost:5000/download` 即可下载保存的模型文件。
阅读全文